Booking/handlers/registration.py

from aiogram import types, Dispatcher
# from dispatcher import bot
from states import RegisterStates
from functions import validate_fio, validate_phone, reject_latin, reject_cmd
from aiogram.dispatcher.storage import FSMContext
from keyboards import reset_register_kb, register_kb, main_kb, check_register_kb
from database import sql_check_user, sql_query_send


async def registration(message: types.Message):
    await message.delete()
    if sql_check_user(f'select tg_id from user_table where tg_id = {message.from_user.id} and approved = 0'):
        await message.answer("Ваша заявка рассматривается", reply_markup=check_register_kb)
    elif sql_check_user(f'select tg_id from user_table where tg_id = {message.from_user.id} and approved = 1'):
        msg = await message.answer("Вы зарегистрированны", reply_markup=main_kb)
        await msg.delete()
    else:
        await message.answer(f"Дkя регистрации необходимо указать\n"
                             f"Номер телефона\n"
                             f"Фамилия Имя Отчество")

        await message.answer(f"Введите номер телефона\n"
                             f"Возможные форматы:\n\n"
                             f"<b>+79995554433</b>\n"
                             f"<b>9997771122</b>\n"
                             f"<b>89995554433</b>\n"
                             f"<b>8-999-888-11-22</b>\n"
                             f"<b>+7-999-888-11-22</b>", reply_markup=reset_register_kb)

        await RegisterStates.phone.set()


async def get_number(message: types.Message, state: FSMContext):
    if validate_phone(message.text):
        await state.update_data(number=message.text)
        await message.answer(f"Теперь укажите ФИО\n"
                             f"Например: Иванов Иван Иванович", reply_markup=reset_register_kb)
        await RegisterStates.FIO.set()
    else:
        await message.delete()
        await message.answer(f"Указан некорректный номер телефона", reply_markup=reset_register_kb)


async def get_fio(message: types.Message, state: FSMContext):
    if reject_cmd(message.text):
        await message.delete()
        await message.answer("Нельзя использовать команды", reply_markup=reset_register_kb)
    elif reject_latin(message.text):
        await message.delete()
        await message.answer("Нельзя использовать латиницу и символы", reply_markup=reset_register_kb)
    elif validate_fio(message.text):
        await message.answer("Необходимо указать полное ФИО", reply_markup=reset_register_kb)
    else:
        await state.update_data(FIO=message.text)
        await state.update_data(id=message.from_user.id)
        reg_data = await state.get_data()
        await message.answer(f"Спасибо за регистрацию\n"
                             f"Вы сможете воспользоваться функциями после одобрения\n", reply_markup=check_register_kb)

        sql_query_send(f"INSERT INTO user_table"
                       f"(tg_id,name,phone) VALUES "
                       f"({reg_data['id']},"
                       f"'{reg_data['FIO']}'"
                       f",{reg_data['number']})")
        await state.finish()


async def reset_register(message: types.Message, state: FSMContext):
    await state.finish()
    await message.answer("Регистрация отменена", reply_markup=register_kb)


async def check_reg_status(message: types.Message):
    await message.delete()
    if sql_check_user(f'select tg_id from user_table where tg_id = {message.from_user.id} and approved = 0'):
        await message.answer("Ваша заявка рассматривается", reply_markup=check_register_kb)
    elif sql_check_user(f'select tg_id from user_table where tg_id = {message.from_user.id} and approved = 1'):
        await message.answer("Вы зарегистрированны", reply_markup=main_kb)


def register_handlers(dp: Dispatcher):
    # хендлеры регистрации
    dp.register_message_handler(registration, text="Зарегистрироваться")
    dp.register_message_handler(check_reg_status, text="Проверить статус заявки")
    dp.register_message_handler(reset_register, text='Отменить регистрацию', state=[RegisterStates.phone,
                                                                                    RegisterStates.FIO])
    dp.register_message_handler(get_number, state=RegisterStates.phone)
    dp.register_message_handler(get_fio, state=RegisterStates.FIO)