diff --git a/helpers/alice.py b/helpers/alice.py index 4eb4be1..cd88fac 100644 --- a/helpers/alice.py +++ b/helpers/alice.py @@ -11,9 +11,10 @@ class Processor: self.user_id = None self.message = data['request']['original_utterance'].lower() - def get_lesson_for_user(self, chat_id: int): - ... - + def get_lesson_for_user(self, hse_id: int): + for lesson in mongo.lessons_collection.find({"hse_user_id": hse_id}).sort({"begin": 1}): + return lesson + return None def process(self) -> dict: user = None @@ -27,14 +28,14 @@ class Processor: "text": "Привет! Я буду тебе подсказывать расписание занятий из РУЗа. Чтобы подключить меня к своему расписанию, зайди в бота, нажми на кнопку \"Подключение Алисы\" и назови мне код из сообщения." } else: - lesson = self.get_lesson_for_user(user['chat_id']) + lesson = self.get_lesson_for_user(user['hse_id']) if lesson is None: return { "text": f"В ближайшее время у тебя нет пар", "end_session": True } return { - "text": f"Твое ближайшее занятие {lesson['begin']}: {lesson['discipline']}", + "text": f"Твое ближайшее занятие {str(lesson['begin'])}: {lesson['discipline']}", "end_session": True } else: @@ -47,7 +48,7 @@ class Processor: else: mongo.users_collection.update_one({"hse_id": int(hse_id)}, {"$set": {"yandex_id": self.user_id}}) bot.send_message(user['chat_id'], "Алиса успешно подключена!") - lesson = self.get_lesson_for_user(user['chat_id']) + lesson = self.get_lesson_for_user(user['hse_id']) if lesson is None: return { "text": f"Отлично, теперь я могу подсказывать тебе расписание. В ближайшее время у тебя нет пар",