Skip to content

Commit

Permalink
logging added
Browse files Browse the repository at this point in the history
  • Loading branch information
Dilshodbek Khojametov committed Sep 1, 2022
1 parent d8db1ac commit 369f727
Show file tree
Hide file tree
Showing 4 changed files with 18 additions and 7 deletions.
2 changes: 2 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -4,3 +4,5 @@
.env
.mypy_cache
.idea/

logfile.txt
13 changes: 6 additions & 7 deletions src/handler.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
from abc import ABC
import logging

from src.logger import logger
from src.messengers import WhatsappMessenger, TelegramMessenger
from src.payload import Payload

Expand All @@ -27,15 +26,15 @@ def handle(self, payload: Payload) -> Payload:
push = PushNotification()
telegram_messages = []
for message in messages:
type = payload.get_type(message)
if type == TELEGRAM:
payload_type = payload.get_type(message)
if payload_type == TELEGRAM:
telegram_messages.append(message)
elif type == WHATSAPP:
elif payload_type == WHATSAPP:
whatsapp.send(message)
elif type == PUSH:
elif payload_type == PUSH:
push.send(message)
else:
logging.log(logging.ERROR, "Unknown message type")
logger.error("Unknown message type")
if telegram_messages:
telegram.send(telegram_messages)
return payload
5 changes: 5 additions & 0 deletions src/logger.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
import logging

logger = logging.getLogger()
handler = logging.FileHandler("logfile.txt")
logger.addHandler(handler)
5 changes: 5 additions & 0 deletions src/messengers.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,13 @@
from src.logger import logger


class TelegramMessenger:
def send(self, message: str) -> str:
logger.info(f"Telegram Message: {message}")
pass


class WhatsappMessenger:
def send(self, message: str) -> str:
logger.info(f"Whatsapp Message: {message}")
pass

0 comments on commit 369f727

Please sign in to comment.