upd command clean log

3 files changed, 17 insertions, 10 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()
diff --git a/main.py b/main.py
index 20fbfd8..655fbeb 100644
--- a/main.py
+++ b/main.py
@@ -35,13 +35,14 @@ def sendLog(msg):
         lines = f.readlines()
         maxLines = len(lines) - 1
         if maxLines < 0:
-            bot.sendMessage("Log kosong")
+            bot.sendMessage("Log kosong", log=False)
+            return
         startLine = max(len(lines) - numLines, 0)
         lines = lines[startLine:]
-        bot.sendMessage(f"Menampilkan log. Mulai dari line: {startLine}")
+        bot.sendMessage(f"Menampilkan log. Mulai dari line: {startLine}", log=False)
         for line in lines:
-            bot.sendMessage(line)
-        bot.sendMessage(f"Selesai menampilkan log. Line terakhir: {maxLines}")
+            bot.sendMessage(line, log=False)
+        bot.sendMessage(f"Selesai menampilkan log. Line terakhir: {maxLines}", log=False)
 
 
 def processMessages(messages):
@@ -54,6 +55,9 @@ def processMessages(messages):
             bot.sendMessage(msg)
         elif 'log' in msg:
             sendLog(msg)
+        elif msg == 'clean':
+            open(os.getenv('LOGFILE'), 'w').close()
+            bot.sendMessage("Log dihapus", log=False)
 
 
 @repeat(every().day.at("09:00"))
diff --git a/util.py b/util.py
index 6df64cd..37f52c0 100644
--- a/util.py
+++ b/util.py
@@ -17,6 +17,8 @@ def setupLogger():
         level=logging.DEBUG,
         format='%(asctime)s - %(levelname)s - %(message)s'
     )
+    logging.getLogger("requests").setLevel(logging.WARNING)
+    logging.getLogger("urllib3").setLevel(logging.WARNING)
     return logger