start project, add all command, add get schedule, add set group, add auto update and add auto send notification

2024-06-09 12:54:13 +07:00
commit 8544dda455
10 changed files with 1903 additions and 0 deletions
@@ -0,0 +1,87 @@
+import re
+import datetime
+from aiogram.types import Message, ReplyKeyboardMarkup, KeyboardButton
+from aiogram import Bot
+from get import get_timetable_for_day, fetch_timetable_data
+from DB import UseDB
+
+
+def get_group_id(user_id):
+    db = UseDB("users")
+    if db.find_document({"user_id": user_id}):
+        return db.find_document({"user_id": user_id})[0]["group_id"]
+    else:
+        return None
+
+
+async def on_notification(msg: Message):
+    time_on_notification = msg.text.split()[1]
+    pattern = r'^([01]?[0-9]|2[0-3]):([0-5]?[0-9])$'
+    if re.match(pattern, time_on_notification):
+        time_obj = datetime.datetime.strptime(time_on_notification, "%H:%M")
+        db = UseDB("users")
+        db.update_document(
+            {"user_id": msg.from_user.id},
+            {"notification": time_obj}
+        )
+        await msg.answer(f"Отлично, время теперь установленно на: {time_obj}")
+    else:
+        await msg.answer(
+            "Формат времени не правильный. "
+            "Правильный формат команды: /on_notification часы:минуты."
+        )
+
+
+async def set_group(msg: Message):
+    try:
+        group_id = int(msg.text.split()[1])
+    except ValueError:
+        return await msg.answer(
+            "Указанная вами группа не найдена. "
+            "Пожалуйста, проверьте корректность номера группы."
+        )
+    response = fetch_timetable_data(group_id)
+    if "404" in str(response) or "500" in str(response):
+        await msg.answer(
+            "Указанная вами группа не найдена. "
+            "Пожалуйста, проверьте корректность номера группы."
+        )
+    else:
+        db = UseDB("users")
+        if db.find_document({"user_id": msg.from_user.id}):
+            db.update_document(
+                {"user_id": msg.from_user.id},
+                {"group_id": group_id}
+            )
+        else:
+            db.insert_document(
+                {
+                    "user_id": msg.from_user.id,
+                    "group_id": group_id,
+                    "notification": "off"
+                }
+            )
+        kb = [
+            [KeyboardButton(text="сегодня")],
+            [KeyboardButton(text="завтра")],
+            [KeyboardButton(text="послезавтра")]
+        ]
+        keyboard = ReplyKeyboardMarkup(keyboard=kb, resize_keyboard=True)
+        await msg.answer("Я сохранил группу для вас.", reply_markup=keyboard)
+
+
+async def timetable_for_day(msg: Message, shift: int = 0):
+    group_id = get_group_id(msg.from_user.id)
+    if group_id is None:
+        return await msg.answer(
+            "Кажется, вашего пользователя нет в нашей базе данных."
+            "Чтобы начать работу, пожалуйста, укажите вашу группу с "
+            "помощью команды: /set_group номер_группы."
+        )
+    for text in get_timetable_for_day(group_id, shift):
+        await msg.answer(text)
+
+
+async def timetable_day(user_id: int, group_id: int, bot: Bot):
+    for text in get_timetable_for_day(group_id, 1):
+        await bot.send_message(chat_id=user_id, text=text)