Booking/handlers/user/make_events.py

from aiogram import types, Dispatcher
from bot import database, sql
from bot.keyboards import register_kb, make_calendar, events_kb, cancel_booking, main_kb
from bot.functions import make_date, time_validator, split_time, to_quotes
from handlers.user.states import BookingState
from aiogram.dispatcher.storage import FSMContext


async def make_event(message: types.message):
    db = database.Database()
    if not db.sql_fetchone(f"select tg_id from user_table where tg_id ={message.from_user.id}") or \
            not db.sql_fetchone(f"select approved from user_table where tg_id={message.from_user.id}"):
        await message.delete()
        await message.answer("Команды станут доступны после регистрации", reply_markup=register_kb)
    else:
        if message.text == "🎯 Запланировать мероприятие":
            await message.delete()
            # TODO: Добавить переход на следующий месяц
            await message.answer(f"выберите дату чтобы увидеть список мероприятий\n\n"
                                 f"Так же календарь мероприятий можно посмотреть в "
                                 f"<a href=moodle.tomtit-tomsk.ru>Moodle</a>\n\n"
                                 f"Сегодняшняя дата <b>{make_date()}</b>", reply_markup=make_calendar())


async def select_date(call: types.CallbackQuery, state: FSMContext):
    db = database.Database()
    print(call.data)
    date = call.data.split("_")[1]

    booked = db.sql_fetchall(f"select events_table.e_start, events_table.e_end from events_table WHERE e_date = {to_quotes(date)}")
    await BookingState.start.set()
    await state.update_data(date=to_quotes(date))
    await state.update_data(owner=call.from_user.id)
    if len(booked) == 0:
        await call.message.edit_text("На этот день мероприятий не заплпнированно", reply_markup=events_kb())
    else:
        await call.message.edit_text(sorted(booked, key=lambda t: t['e_start'], reverse=True), reply_markup=events_kb())


async def edit_date(call: types.CallbackQuery, state: FSMContext):

    await call.message.edit_text(f"выберите дату чтобы увидеть список мероприятий\n\n"
                                 f"Так же календарь мероприятий можно посмотреть в "
                                 f"<a href=moodle.tomtit-tomsk.ru>Moodle</a>\n\n"
                                 f"Сегодняшняя дата <b>{make_date()}</b>", reply_markup=make_calendar())
    await state.finish()


async def booking_date(call: types.CallbackQuery):
    await call.message.edit_text("Введите диапазон времени\n"
                                 "Возможные форматы\n\n"
                                 "13.00 15.30\n"
                                 "13.00-15.30\n"
                                 "13:00 15:30\n"
                                 "13.00-15.30\n", reply_markup=cancel_booking())
    # TODO: Проверка на занятость
    await BookingState.time.set()


async def get_date(message: types.Message, state: FSMContext):
    if time_validator(message.text):
        db = database.Database()
        time = split_time(message.text)
        await state.update_data(t_start=time[0])
        await state.update_data(t_end=time[1])
        sql_data = db.sql_fetchall(sql.sql_time_range(time))
        print(len(sql_data))
        await message.answer("Введите краткое описание мероприятия", reply_markup=cancel_booking())
        await BookingState.description.set()
    else:
        await message.answer("Неверный формат времени")

    # TODO: огран по длине текста


async def send_event(message: types.Message, state: FSMContext):
    db = database.Database()
    await state.update_data(description=message.text)
    await state.update_data(approved=0)
    data = await state.get_data()
    await message.answer("Заявка принята", reply_markup=main_kb)
    await state.finish()
    await message.answer(data)
    db.sql_query_send(sql.sql_send_event(data))


def events_register(dp: Dispatcher):
    dp.register_message_handler(make_event, text="🎯 Запланировать мероприятие")
    dp.register_callback_query_handler(select_date, text_startswith='date_')
    dp.register_callback_query_handler(edit_date, text=['change', 'cancel_booking'], state=[BookingState.start,
                                                                                            BookingState.time,
                                                                                            BookingState.description])
    dp.register_callback_query_handler(booking_date, text='booking', state=BookingState.start)
    dp.register_message_handler(get_date, state=BookingState.time)
    dp.register_message_handler(send_event, state=BookingState.description)