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.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"Отлично, теперь я могу подсказывать тебе расписание. В ближайшее время у тебя нет пар",