Добавлено оповещение администраторов

bot/config.py

@@ -11,3 +11,4 @@ DB_HOST = os.getenv('DB_HOST')
 DB_NAME = os.getenv('DB_NAME')
 DB_USER = os.getenv('DB_USER')
 DB_PASS = os.getenv('DB_PASS')
+CHAT_ID = os.getenv('CHAT_ID')

handlers/admin/__init__.py

@@ -1,3 +1,4 @@
 import handlers.admin.admin_menu
 import handlers.admin.manage_users
 import handlers.admin.manage_events
+import handlers.admin.notifications

handlers/admin/notifications.py

@@ -1 +1,17 @@
 # TODO: Рассылка для администраторов
+from bot.dispatcher import bot
+import bot.config as cnf
+
+async def new_event():
+    await bot.send_message(cnf.CHAT_ID, "Новая заявка мероприятия")
+
+
+async def new_user():
+    await bot.send_message(cnf.CHAT_ID, "Новая заявка на регистрацию")
+
+
+async def new_bug(data):
+    msg = f"❗Баг❗\n" \
+          f"Баг: {data['bug']}\n" \
+          f"Отправитель: {data['from_user']}\n"
+    await bot.send_message(cnf.CHAT_ID, msg)

handlers/start.py

@@ -1,10 +1,11 @@
-import asyncio
-
-import aiogram
 from aiogram import types, Dispatcher
 from bot.keyboards import main_kb, register_kb,   check_register_kb
 from bot import database
 from bot import sql
+from bot.dispatcher import bot
+from aiogram.dispatcher.storage import FSMContext
+from handlers.user.states import SendBugState
+from handlers.admin.notifications import new_bug
 
 
 # @dp.message_handler(commands=['start'])
@@ -14,8 +15,8 @@ async def start_cmd(message: types.Message):
     if not db.sql_fetchone(sql.check_id(message.from_user.id)):
         await message.answer(f"🤖Вас приветствует лакей ТТИТ🤖\n\n"
                              "Для доступа к функциям нужно пройти простую регистрацию\n", reply_markup=register_kb)
-    elif not db.sql_fetchone(sql.check_approved(message.from_user.id)):
-        await message.answer(f"Ваша заявка находится на рассмотрернии", reply_markup=check_register_kb)
+    elif db.sql_fetchone(sql.check_approved(message.from_user.id)) == 0:
+        await message.answer(f"Ваша заявка находится на рассмотрении", reply_markup=check_register_kb)
     else:
         await message.answer(f"🤖Вас приветствует лакей ТТИТ🤖\n"
                              f"\n"
@@ -26,7 +27,8 @@ async def start_cmd(message: types.Message):
                              f"Мои события\n"
                              f"Все события\n\n"
                              f"Если есть пожелания или замечания\n"
-                             f"Можете обратиться к @FeldwebelWillman",
+                             f"Можете обратиться к @FeldwebelWillman\n"
+                             f"Или воспользовтаься обратной связью /bug",
                              reply_markup=main_kb)
 
 
@@ -37,6 +39,21 @@ async def stop_cmd(message: types.Message):
         await message.answer("Я слушаюсь только создателя")
 
 
+async def send_report(message: types.Message):
+    await SendBugState.send_bug.set()
+    await message.answer("Опишите проблему")
+
+
+async def get_report(message: types.Message, state: FSMContext):
+    await state.update_data(bug=message.text)
+    await state.update_data(from_user=message.from_user.username)
+    data = await state.get_data()
+    await state.finish()
+    await new_bug(data)
+
+
 def main_register(dp: Dispatcher):
     dp.register_message_handler(start_cmd, commands=['start', 'help'])
     dp.register_message_handler(stop_cmd, commands=['stop'])
+    dp.register_message_handler(send_report, commands=['bug'])
+    dp.register_message_handler(get_report, state=SendBugState.send_bug)

handlers/user/make_events.py

@@ -5,11 +5,7 @@ from bot.functions import make_date, time_validator, normalize_time, to_quotes,
 from handlers.user.states import BookingState
 from aiogram.dispatcher.storage import FSMContext
 from bot import messages
-
-
-# TODO: Убрать вывод данных состояния
-# Добавить визуализацию занятого времени
-# Переписать реактирование сообщений
+from handlers.admin.notifications import new_event
 
 
 async def make_event(message: types.message):
@@ -90,8 +86,8 @@ async def send_event(message: types.Message, state: FSMContext):
         data = await state.get_data()
         await message.answer("Заявка принята", reply_markup=main_kb)
         await state.finish()
-        print(data)
         db.sql_query_send(sql.sql_send_event(data))
+        await new_event()
 
 
 def events_register(dp: Dispatcher):

handlers/user/registration.py

@@ -4,20 +4,20 @@ from bot.functions import validate_fio, validate_phone, reject_latin, reject_cmd
 from aiogram.dispatcher.storage import FSMContext
 from bot.keyboards import reset_register_kb, register_kb, main_kb, check_register_kb
 from bot import database, sql
-
+from handlers.admin.notifications import new_user
 
 async def registration(message: types.Message):
     db = database.Database()
     await message.delete()
     if db.sql_fetchone(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)
+        await message.answer("Ваша заявка находится на рассмотрернии", reply_markup=check_register_kb)
     elif db.sql_fetchone(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"Дkя регистрации необходимо будет указать\n"
+                             f"Номер телефона и "
+                             f"Фамилию Имя Отчество")
 
         await message.answer(f"Введите номер телефона\n"
                              f"Возможные форматы:\n\n"
@@ -33,9 +33,11 @@ async def check_reg_status(message: types.Message):
     db = database.Database()
     await message.delete()
     if db.sql_fetchone(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)
+        await message.answer("Ваша заявка находится на рассмотрернии", reply_markup=check_register_kb)
     elif db.sql_fetchone(f'select tg_id from user_table where tg_id = {message.from_user.id} and approved = 1'):
         await message.answer("Вы зарегистрированны", reply_markup=main_kb)
+    else:
+        await message.answer("Вы не зарегистрированны", reply_markup=register_kb)
 
 
 async def get_number(message: types.Message, state: FSMContext):
@@ -66,8 +68,9 @@ async def get_fio(message: types.Message, state: FSMContext):
         await message.answer(f"Спасибо за регистрацию\n"
                              f"Вы сможете воспользоваться функциями после одобрения\n", reply_markup=check_register_kb)
 
-        db.sql_query_send(sql.sql_send_register(reg_data))
+        db.sql_query_send(sql.sql_send(reg_data))
         await state.finish()
+        await new_user()
 
 
 async def reset_register(message: types.Message, state: FSMContext):

handlers/user/show_events.py

@@ -43,6 +43,7 @@ async def select_range(call: types.CallbackQuery):
         else:
             await call.message.answer(beauty_all_events(sorted(data, key=lambda d: d['e_date'])))
 
+
     if call.data == "week":
         db = database.Database()
         time = date_range("week").split(" ")

handlers/user/states.py

@@ -12,3 +12,7 @@ class BookingState(StatesGroup):
     description = State()
 
 
+class SendBugState(StatesGroup):
+    send_bug = State()
+
+