async def register_phone_write(message: types.Message, state: FSMContext): """Ввод номера телефона.""" await message.delete() user = DataBaseFunc.get_user(message.from_user.id) data = await state.get_data() message_id = data['callback_message_id'] phone = "".join(ch for ch in message.text if ch.isdigit()) if (phone[0] == '8'): phone = '7' + phone[1:] errors = get_text(user, 'write_phone_errors') keyboard = UserGeneratorKeyboard.register_button(user) # keyboard = types.InlineKeyboardMarkup() # keyboard.add(types.InlineKeyboardButton(get_text_but(user, 'register_write_back'), callback_data="register_write_back")) if len(phone) == 0: await bot.edit_message_text(text=errors['empty'], chat_id=message.chat.id, message_id=message_id, reply_markup=keyboard) await UserStateRegister.main_menu.set() return check_user = DataBaseFunc.get_user_for_phone(phone) if (check_user != None): await bot.edit_message_text(text=errors['is_register'], chat_id=message.chat.id, message_id=message_id, reply_markup=keyboard) await UserStateRegister.main_menu.set() return contact = DataBaseFunc.get_contact(phone=phone) if (contact == None): await bot.edit_message_text(text=errors['not_found'], chat_id=message.chat.id, message_id=message_id, reply_markup=keyboard) await UserStateRegister.main_menu.set() return user.phone = phone user.mail = contact.mail user.is_register = True user.course_id = contact.course_id contact.is_register = True DataBaseFunc.commit() DataBaseFunc.add_course_in_user(user, DataBaseFunc.get_course(user.course_id)) await bot.edit_message_text(text=get_text(user, 'start'), chat_id=message.chat.id, message_id=message_id, reply_markup=await UserGeneratorKeyboard.start_button(user)) await UserStateMainMenu.main_menu.set()
async def register_mail_write(message: types.Message, state: FSMContext): """Ввод номера телефона.""" await message.delete() user = DataBaseFunc.get_user(message.from_user.id) data = await state.get_data() message_id = data['callback_message_id'] mail = message.text mail = mail.lower() errors = get_text(user, 'write_mail_errors') keyboard = UserGeneratorKeyboard.register_button(user) # keyboard = types.InlineKeyboardMarkup() # keyboard.add(types.InlineKeyboardButton(get_text_but(user, 'register_write_back'), callback_data="register_write_back")) if ("@" in mail) == False: await bot.edit_message_text(text=errors['empty'], chat_id=message.chat.id, message_id=message_id, reply_markup=keyboard) await UserStateRegister.main_menu.set() return check_user = DataBaseFunc.get_user_for_mail(mail) if (check_user != None): await bot.edit_message_text(text=errors['is_register'], chat_id=message.chat.id, message_id=message_id, reply_markup=keyboard) await UserStateRegister.main_menu.set() return contact = DataBaseFunc.get_contact(mail=mail) if (contact == None): await bot.edit_message_text(text=errors['not_found'], chat_id=message.chat.id, message_id=message_id, reply_markup=keyboard) await UserStateRegister.main_menu.set() return user.mail = mail user.phone = contact.phone user.is_register = True contact.is_register = True user.course_id = contact.course_id DataBaseFunc.commit() DataBaseFunc.add_course_in_user(user, DataBaseFunc.get_course(user.course_id)) await bot.edit_message_text(text=get_text(user, 'start'), chat_id=message.chat.id, message_id=message_id, reply_markup=await UserGeneratorKeyboard.start_button(user)) await UserStateMainMenu.main_menu.set()