Browse Source

Добавлена защита

Vildan 2 years ago
parent
commit
e19c6febbd
2 changed files with 15 additions and 3 deletions
  1. 5 1
      handlers/admin/manage_users.py
  2. 10 2
      handlers/user/events.py

+ 5 - 1
handlers/admin/manage_users.py

@@ -1,11 +1,15 @@
 from bot import database
 from aiogram import types, Dispatcher
 from bot.functions import beauty_reg_request
-from bot.keyboards import user_manage_kb
+from bot.keyboards import user_manage_kb,register_kb
 
 
 async def list_users(message: types.Message):
     Db = database.Database()
+    if not Db.sql_simple_check(sql=f"select tg_id from user_table where tg_id ={message.from_user.id}") or \
+            not Db.sql_simple_check(sql=f"select approved from user_table where tg_id={message.from_user.id}"):
+        await message.delete()
+        await message.answer("Команды станут доступны после регистрации", reply_markup=register_kb)
     if not Db.sql_parse_users("select id,name,phone from user_table where approved = '0'"):
         await message.answer('Заявки на регистрацию отсутствуют')
     else:

+ 10 - 2
handlers/user/events.py

@@ -25,11 +25,19 @@ async def select_date(call: types.CallbackQuery):
 
 
 async def my_events(message: types.Message):
-    await message.answer("")
+    Db = database.Database()
+    if not Db.sql_simple_check(sql=f"select tg_id from user_table where tg_id ={message.from_user.id}") or \
+            not Db.sql_simple_check(sql=f"select approved from user_table where tg_id={message.from_user.id}"):
+        await message.delete()
+        await message.answer("Команды станут доступны после регистрации", reply_markup=register_kb)
 
 
 async def all_events(message: types.Message):
-    await message.answer("")
+    Db = database.Database()
+    if not Db.sql_simple_check(sql=f"select tg_id from user_table where tg_id ={message.from_user.id}") or \
+            not Db.sql_simple_check(sql=f"select approved from user_table where tg_id={message.from_user.id}"):
+        await message.delete()
+        await message.answer("Команды станут доступны после регистрации", reply_markup=register_kb)
 
 
 def events_register(dp: Dispatcher):