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

again and again

Browse Source
This commit is contained in:
Administrator 2022-10-25 15:42:58 +03:00
parent 2e0f66c5af
commit b879e5e19f
2 changed files with 18 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
helpers/answer.py
View File

@ -43,10 +43,14 @@ class Answer(BaseAnswer):
def handle_state_wait_for_student_or_teacher(self, message: Message, user: User): def handle_state_wait_for_student_or_teacher(self, message: Message, user: User):
if message.text == "Студент": if message.text == "Студент":
bot.send_message(user.chat_id, "Принято! Теперь отправь мне свое ФИО.") bot.send_message(user.chat_id, "Принято! Теперь отправь мне свое ФИО.", reply_markup=again_keyboard())
mongo.users_collection.update_one(
{"chat_id": user.chat_id},
{"$set": {"is_teacher": False, "state": "wait_for_name"}}
)
self.set_state(user, "wait_for_name") self.set_state(user, "wait_for_name")
elif message.text == "Преподаватель": elif message.text == "Преподаватель":
bot.send_message(user.chat_id, "Принято! Теперь отправь мне свое ФИО.") bot.send_message(user.chat_id, "Принято! Теперь отправь мне свое ФИО.", reply_markup=again_keyboard())
mongo.users_collection.update_one( mongo.users_collection.update_one(
{"chat_id": user.chat_id}, {"chat_id": user.chat_id},
{"$set": {"is_teacher": True, "state": "wait_for_name"}} {"$set": {"is_teacher": True, "state": "wait_for_name"}}
@ -59,8 +63,10 @@ class Answer(BaseAnswer):
if message.text == "Начать заново": if message.text == "Начать заново":
bot.send_message( bot.send_message(
message.chat.id, message.chat.id,
"Привет! Я буду помогать тебе выживать в вышке!\nДля начала пришли мне свое ФИО.", "Привет! Я буду помогать тебе выживать в вышке!\nДля начала сообщи мне, ты преподаватель или студент?",
reply_markup=student_or_teacher_keyboard()
) )
self.set_state(user, "wait_for_student_or_teacher")
return return
user.name = message.text user.name = message.text
data = ruz.find_person(user) data = ruz.find_person(user)
@ -91,6 +97,14 @@ class Answer(BaseAnswer):
self.set_state(user, "wait_for_group") self.set_state(user, "wait_for_group")
def handle_state_wait_for_group(self, message: Message, user: User): def handle_state_wait_for_group(self, message: Message, user: User):
if message.text == "Начать заново":
bot.send_message(
message.chat.id,
"Привет! Я буду помогать тебе выживать в вышке!\nДля начала сообщи мне, ты преподаватель или студент?",
reply_markup=student_or_teacher_keyboard()
)
self.set_state(user, "wait_for_student_or_teacher")
return
group = message.text group = message.text
data = ruz.find_person(user) data = ruz.find_person(user)
if data is None: if data is None:

1
helpers/keyboards.py
View File

@ -44,6 +44,7 @@ def groups_keyboard(data):
kb = telebot.types.ReplyKeyboardMarkup(True, False) kb = telebot.types.ReplyKeyboardMarkup(True, False)
for entity in data: for entity in data:
kb.row(entity['description']) kb.row(entity['description'])
kb.row("Начать заново")
return kb return kb