initial

daemons/bot.py

@@ -0,0 +1,13 @@
+import os
+
+import telebot
+from telebot.types import Message
+
+
+bot = telebot.TeleBot(os.getenv("TELEGRAM_TOKEN"))
+
+
+@bot.message_handler()
+def do_action(message: Message):
+    from helpers.answer import answer
+    answer.process(message)

daemons/fetch.py

@@ -0,0 +1,83 @@
+import datetime
+import zoneinfo
+from time import sleep
+
+from helpers.mongo import mongo
+from helpers.ruz import ruz
+
+
+def fetch_schedule_for_user(user_hse_id: int):
+    zone = zoneinfo.ZoneInfo("Europe/Moscow")
+    today = datetime.datetime.now(zone)
+    next_day = today + datetime.timedelta(days=7)
+    schedule = ruz.get_schedule(user_hse_id, today, next_day)
+    if schedule is None:
+        return False
+    saved_ids = []
+    for element in schedule:
+        year, month, day = element['date'].split('.')
+        begin_hour, begin_minute = element['beginLesson'].split(':')
+        end_hour, end_minute = element['endLesson'].split(':')
+        lesson = mongo.lessons_collection.find_one({
+            "discipline": element['discipline'],
+            "auditorium": element['auditorium'],
+            "hse_user_id": user_hse_id,
+            "begin": datetime.datetime(
+                year=int(year),
+                month=int(month),
+                day=int(day),
+                hour=int(begin_hour),
+                minute=int(begin_minute)
+            )
+        })
+        if lesson is None:
+            result = mongo.lessons_collection.insert_one({
+                "discipline": element['discipline'],
+                "auditorium": element['auditorium'],
+                "hse_user_id": user_hse_id,
+                "begin": datetime.datetime(
+                    year=int(year),
+                    month=int(month),
+                    day=int(day),
+                    hour=int(begin_hour),
+                    minute=int(begin_minute)
+                ),
+                "end": datetime.datetime(
+                    year=int(year),
+                    month=int(month),
+                    day=int(day),
+                    hour=int(end_hour),
+                    minute=int(end_minute)
+                ),
+                "building": element['building'],
+                "lecturer": element['lecturer'],
+                "notified": False
+            })
+            saved_ids.append(result.inserted_id)
+        else:
+            saved_ids.append(lesson['_id'])
+    mongo.lessons_collection.delete_many({"hse_user_id": user_hse_id, "_id": {"$nin": saved_ids}})
+    return True
+
+
+def process():
+    for user in mongo.users_collection.find({}):
+        fetch_schedule_for_user(user['hse_id'])
+
+
+def delete_old():
+    zone = zoneinfo.ZoneInfo("Europe/Moscow")
+    today = datetime.datetime.now(zone)
+    mongo.lessons_collection.delete_many({"end": {"$lte": today - datetime.timedelta(days=1)}})
+
+
+def fetch():
+    while True:
+        print("fetch start")
+        begin = datetime.datetime.now()
+        process()
+        end = datetime.datetime.now()
+        print('fetch finished')
+        print("time elapsed", (end - begin).total_seconds())
+        delete_old()
+        sleep(60 * 60)

daemons/notify.py

@@ -0,0 +1,38 @@
+import datetime
+import zoneinfo
+from time import sleep
+
+from daemons.bot import bot
+from helpers.mongo import mongo
+
+
+def process():
+    for user in mongo.users_collection.find({"notify_minutes": {"$ne": None}}):
+        zone = zoneinfo.ZoneInfo("Europe/Moscow")
+        now = datetime.datetime.now(zone)
+        for lesson in mongo.lessons_collection.find({
+            "hse_user_id": user["hse_id"],
+            "begin": {"$lte": now + datetime.timedelta(minutes=5)},
+            "notified": False
+        }):
+            ans = ""
+            ans += "Аудитория: " + lesson["building"] + ", " + lesson["auditorium"] + "\n"
+            ans += "Начало: " + lesson["begin"].strftime("%H:%M") + "\n"
+            ans += "Конец: " + lesson["end"].strftime("%H:%M") + "\n"
+            ans += "Преподаватель: " + lesson["lecturer"] + "\n"
+            bot.send_message(
+                user["chat_id"],
+                "Уведомляю о занятиях!\n" + ans
+            )
+            mongo.lessons_collection.update_one({"_id": lesson['_id']}, {"$set": {"notified": True}})
+
+
+def notify():
+    while True:
+        print("notify start")
+        begin = datetime.datetime.now()
+        process()
+        end = datetime.datetime.now()
+        print('notify finished')
+        print("time elapsed", (end - begin).total_seconds())
+        sleep(60 * 2)