upd command clean log

1 files changed, 7 insertions, 6 deletions

diff --git a/bot.py b/bot.py
index c66d6a0..5666987 100644
--- a/bot.py
+++ b/bot.py
@@ -1,7 +1,7 @@
 import os
 import sqlite3
-import urllib.error
 from urllib import parse
+
 import requests
 from dotenv import load_dotenv
 
@@ -20,14 +20,15 @@ class Bot:
         self.hasNewMessages = False
         self.messages = list()
 
-    def sendMessage(self, message):
+    def sendMessage(self, message, log=True):
         message = parse.quote_plus(message)
         url = f'https://api.telegram.org/bot{self.token}/sendMessage?chat_id={self.chatId}&parse_mode=html&text={message}'
         response = requests.get(url)
-        if response.status_code == 200:
-            logger.info(f'Message sent: {response.text}')
-        else:
-            logger.error(f'Error sending message: {response.text}')
+        if log:
+            if response.status_code == 200:
+                logger.info(f'Message sent: {response.text}')
+            else:
+                logger.error(f'Error sending message: {response.text}')
 
     def _dbUpdate(self, data):
         conn = getDbConn()