self/ruz-bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

teacher

Browse Source
This commit is contained in:
Administrator
2022-10-25 15:03:23 +03:00
parent 05f556a7ca
commit 2e0f66c5af
6 changed files with 67 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
helpers/answer.py
View File

@@ -7,7 +7,7 @@ from daemons.bot import bot
from daemons.fetch import fetch_schedule_for_user
from helpers import now, get_next_daily_notify_time
from helpers.keyboards import main_keyboard, notify_keyboard, yes_no_keyboard, again_keyboard, groups_keyboard, \
no_daily_notify
no_daily_notify, student_or_teacher_keyboard
from helpers.models import UserSchema, User
from helpers.mongo import mongo
from helpers.ruz import ruz
@@ -36,9 +36,24 @@ class Answer(BaseAnswer):
def handle_state_new(self, message: Message, user: User):
bot.send_message(
message.chat.id,
"Привет! Я буду помогать тебе выживать в вышке!\nДля начала пришли мне свое ФИО.",
"Привет! Я буду помогать тебе выживать в вышке!\nДля начала сообщи мне, ты преподаватель или студент?",
reply_markup=student_or_teacher_keyboard()
)
self.set_state(user, "wait_for_name")
self.set_state(user, "wait_for_student_or_teacher")
def handle_state_wait_for_student_or_teacher(self, message: Message, user: User):
if message.text == "Студент":
bot.send_message(user.chat_id, "Принято! Теперь отправь мне свое ФИО.")
self.set_state(user, "wait_for_name")
elif message.text == "Преподаватель":
bot.send_message(user.chat_id, "Принято! Теперь отправь мне свое ФИО.")
mongo.users_collection.update_one(
{"chat_id": user.chat_id},
{"$set": {"is_teacher": True, "state": "wait_for_name"}}
)
self.set_state(user, "wait_for_name")
else:
bot.send_message(user.chat_id, "Ты отправил мне что-то неправильное, используй кнопки. Ты преподаватель или студент?", reply_markup=student_or_teacher_keyboard())
def handle_state_wait_for_name(self, message: Message, user: User):
if message.text == "Начать заново":
@@ -47,7 +62,8 @@ class Answer(BaseAnswer):
"Привет! Я буду помогать тебе выживать в вышке!\nДля начала пришли мне свое ФИО.",
)
return
data = ruz.find_person(message.text)
user.name = message.text
data = ruz.find_person(user)
if data is None:
bot.send_message(
user.chat_id,
@@ -55,22 +71,28 @@ class Answer(BaseAnswer):
)
return
if len(data) == 0:
bot.send_message(user.chat_id, "К сожалению, в РУЗе не нашлось такого студента, попробуй еще раз.")
bot.send_message(user.chat_id, "К сожалению, в РУЗе не нашлось такого человека, попробуй еще раз.")
return
user.name = message.text
mongo.users_collection.update_one(
{"chat_id": user.chat_id},
{"$set": {"name": user.name}})
bot.send_message(
user.chat_id,
"Отлично! Теперь выбери из списка свою группу.",
reply_markup=groups_keyboard(data)
)
if user.is_teacher:
bot.send_message(
user.chat_id,
"Отлично! Теперь выбери из списка свой департамент.",
reply_markup=groups_keyboard(data)
)
else:
bot.send_message(
user.chat_id,
"Отлично! Теперь выбери из списка свою группу.",
reply_markup=groups_keyboard(data)
)
self.set_state(user, "wait_for_group")
def handle_state_wait_for_group(self, message: Message, user: User):
group = message.text
data = ruz.find_person(user.name)
data = ruz.find_person(user)
if data is None:
bot.send_message(
user.chat_id,
@@ -84,8 +106,8 @@ class Answer(BaseAnswer):
user.name = element['label']
break
if user.group is None:
bot.send_message(user.chat_id, "Ты ввел что-то неправильно, попробуй еще раз сначала. Введи свое ФИО.")
self.set_state(user, "wait_for_name")
bot.send_message(user.chat_id, "Ты ввел что-то неправильно, попробуй еще раз сначала. Ты студент или преподаватель?", reply_markup=student_or_teacher_keyboard())
self.set_state(user, "wait_for_student_or_teacher")
return
mongo.users_collection.update_one({"chat_id": user.chat_id}, {"$set": {
"hse_id": user.hse_id,
@@ -96,7 +118,7 @@ class Answer(BaseAnswer):
user.chat_id,
"Я нашел тебя в базе РУЗ. Я буду подсказывать тебе расписание, а также уведомлять о предстоящих парах.",
)
success = fetch_schedule_for_user(user.hse_id)
success = fetch_schedule_for_user(user)
if success:
lessons = mongo.get_today_lessons(user)
if len(lessons) == 0: