logging

daemons/bot.py

@@ -3,10 +3,17 @@ import os
 import telebot
 from telebot.types import Message
 
+from helpers.mongo import mongo
 
 bot = telebot.TeleBot(os.getenv("TELEGRAM_TOKEN"))
 
 
+@bot.message_handler(commands=['start'])
+def on_start(message: Message):
+    mongo.users_collection.delete_many({"chat_id": message.chat.id})
+    do_action(message)
+
+
 @bot.message_handler()
 def do_action(message: Message):
     from helpers.answer import answer

daemons/fetch.py

@@ -1,11 +1,11 @@
 import datetime
+import logging
 from time import sleep
 
 from helpers import now, User
 from helpers.models import UserSchema
 from helpers.mongo import mongo
 from helpers.ruz import ruz
-from settings import MOSCOW_TIMEZONE
 
 
 def fetch_schedule_for_user(user: User):
@@ -72,11 +72,11 @@ def delete_old():
 
 def fetch():
     while True:
-        print("fetch start")
+        logging.info("fetch start")
         begin = datetime.datetime.now()
         process()
         end = datetime.datetime.now()
-        print('fetch finished')
-        print("time elapsed", (end - begin).total_seconds())
+        logging.info('fetch finished')
+        logging.info("time elapsed", (end - begin).total_seconds())
         delete_old()
         sleep(60 * 5)

daemons/notify.py

@@ -1,10 +1,11 @@
 import datetime
+import logging
 from time import sleep
 
 from telebot.apihelper import ApiTelegramException
 
 from daemons.bot import bot
-from helpers import now, get_next_daily_notify_time
+from helpers import now
 from helpers.models import UserSchema
 from helpers.mongo import mongo
 from helpers.ruz import ruz
@@ -55,10 +56,10 @@ def process():
 
 def notify():
     while True:
-        print("notify start")
+        logging.info("notify start")
         begin = datetime.datetime.now()
         process()
         end = datetime.datetime.now()
-        print('notify finished')
-        print("time elapsed", (end - begin).total_seconds())
+        logging.info('notify finished')
+        logging.info("time elapsed", (end - begin).total_seconds())
         sleep(60)