get lesson

This commit is contained in:
Administrator 2023-04-08 14:25:16 +03:00
parent 209a355e37
commit 9820cc5b91

View File

@ -11,9 +11,10 @@ class Processor:
self.user_id = None self.user_id = None
self.message = data['request']['original_utterance'].lower() 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: def process(self) -> dict:
user = None user = None
@ -27,14 +28,14 @@ class Processor:
"text": "Привет! Я буду тебе подсказывать расписание занятий из РУЗа. Чтобы подключить меня к своему расписанию, зайди в бота, нажми на кнопку \"Подключение Алисы\" и назови мне код из сообщения." "text": "Привет! Я буду тебе подсказывать расписание занятий из РУЗа. Чтобы подключить меня к своему расписанию, зайди в бота, нажми на кнопку \"Подключение Алисы\" и назови мне код из сообщения."
} }
else: else:
lesson = self.get_lesson_for_user(user['chat_id']) lesson = self.get_lesson_for_user(user['hse_id'])
if lesson is None: if lesson is None:
return { return {
"text": f"В ближайшее время у тебя нет пар", "text": f"В ближайшее время у тебя нет пар",
"end_session": True "end_session": True
} }
return { return {
"text": f"Твое ближайшее занятие {lesson['begin']}: {lesson['discipline']}", "text": f"Твое ближайшее занятие {str(lesson['begin'])}: {lesson['discipline']}",
"end_session": True "end_session": True
} }
else: else:
@ -47,7 +48,7 @@ class Processor:
else: else:
mongo.users_collection.update_one({"hse_id": int(hse_id)}, {"$set": {"yandex_id": self.user_id}}) mongo.users_collection.update_one({"hse_id": int(hse_id)}, {"$set": {"yandex_id": self.user_id}})
bot.send_message(user['chat_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: if lesson is None:
return { return {
"text": f"Отлично, теперь я могу подсказывать тебе расписание. В ближайшее время у тебя нет пар", "text": f"Отлично, теперь я могу подсказывать тебе расписание. В ближайшее время у тебя нет пар",