Add antiflood and lint

handlers/admin/admin_user_editor.py

@@ -7,9 +7,15 @@ from aiogram.fsm.context import FSMContext
 from aiogram.fsm.state import State, StatesGroup
 from aiogram.types import CallbackQuery, InlineKeyboardButton, InlineKeyboardMarkup
 from aiogram.utils.keyboard import InlineKeyboardBuilder
-
 from config import TOTAL_GB
-from database import delete_user_data, get_client_id_by_email, get_servers_from_db, restore_trial, update_key_expiry
+
+from database import (
+    delete_user_data,
+    get_client_id_by_email,
+    get_servers_from_db,
+    restore_trial,
+    update_key_expiry,
+)
 from filters.admin import IsAdminFilter
 from handlers.keys.key_utils import (
     delete_key_from_cluster,
@@ -56,7 +62,7 @@ async def prompt_username(callback_query: CallbackQuery, state: FSMContext):
 async def handle_username_input(
     message: types.Message, state: FSMContext, session: Any
 ):
-    username = message.text.strip().lstrip('@').replace('https://t.me/', '')
+    username = message.text.strip().lstrip("@").replace("https://t.me/", "")
     user_record = await session.fetchrow(
         "SELECT tg_id FROM users WHERE username = $1", username
     )
@@ -113,12 +119,19 @@ async def handle_username_input(
             callback_data=f"restore_trial_{tg_id}",
         )
     )
-    builder.row(InlineKeyboardButton(text="❌ Удалить клиента", callback_data=f"confirm_delete_user_{tg_id}"))
-    builder.row(InlineKeyboardButton(text="🔄 Обновить клиента", callback_data=f"user_info|{tg_id}"))
     builder.row(
         InlineKeyboardButton(
-            text="✉️ Отправить сообщение",
-            callback_data=f"send_message_{tg_id}"
+            text="❌ Удалить клиента", callback_data=f"confirm_delete_user_{tg_id}"
+        )
+    )
+    builder.row(
+        InlineKeyboardButton(
+            text="🔄 Обновить клиента", callback_data=f"user_info|{tg_id}"
+        )
+    )
+    builder.row(
+        InlineKeyboardButton(
+            text="✉️ Отправить сообщение", callback_data=f"send_message_{tg_id}"
         )
     )
     builder.row(InlineKeyboardButton(text="🔙 Назад", callback_data="user_editor"))
@@ -201,8 +214,16 @@ async def handle_tg_id_input(message: types.Message, state: FSMContext, session:
             callback_data=f"change_balance_{tg_id}",
         )
     )
-    builder.row(InlineKeyboardButton(text="❌ Удалить клиента", callback_data=f"confirm_delete_user_{tg_id}"))
-    builder.row(InlineKeyboardButton(text="🔄 Обновить клиента", callback_data=f"user_info|{tg_id}"))
+    builder.row(
+        InlineKeyboardButton(
+            text="❌ Удалить клиента", callback_data=f"confirm_delete_user_{tg_id}"
+        )
+    )
+    builder.row(
+        InlineKeyboardButton(
+            text="🔄 Обновить клиента", callback_data=f"user_info|{tg_id}"
+        )
+    )
     builder.row(
         InlineKeyboardButton(
             text="🔄 Восстановить пробник",
@@ -211,8 +232,7 @@ async def handle_tg_id_input(message: types.Message, state: FSMContext, session:
     )
     builder.row(
         InlineKeyboardButton(
-            text="✉️ Отправить сообщение",
-            callback_data=f"send_message_{tg_id}"
+            text="✉️ Отправить сообщение", callback_data=f"send_message_{tg_id}"
         )
     )
 
@@ -645,14 +665,29 @@ async def handle_user_info(
             InlineKeyboardButton(text=f"🔑 {email}", callback_data=f"edit_key_{email}")
         )
 
-    builder.row(InlineKeyboardButton(text="📝 Изменить баланс", callback_data=f"change_balance_{tg_id}"))
-    builder.row(InlineKeyboardButton(text="🔄 Восстановить пробник", callback_data=f"restore_trial_{tg_id}"))
-    builder.row(InlineKeyboardButton(text="❌ Удалить клиента", callback_data=f"confirm_delete_user_{tg_id}"))
-    builder.row(InlineKeyboardButton(text="🔄 Обновить клиента", callback_data=f"user_info|{tg_id}"))
     builder.row(
         InlineKeyboardButton(
-            text="✉️ Отправить сообщение",
-            callback_data=f"send_message_{tg_id}"
+            text="📝 Изменить баланс", callback_data=f"change_balance_{tg_id}"
+        )
+    )
+    builder.row(
+        InlineKeyboardButton(
+            text="🔄 Восстановить пробник", callback_data=f"restore_trial_{tg_id}"
+        )
+    )
+    builder.row(
+        InlineKeyboardButton(
+            text="❌ Удалить клиента", callback_data=f"confirm_delete_user_{tg_id}"
+        )
+    )
+    builder.row(
+        InlineKeyboardButton(
+            text="🔄 Обновить клиента", callback_data=f"user_info|{tg_id}"
+        )
+    )
+    builder.row(
+        InlineKeyboardButton(
+            text="✉️ Отправить сообщение", callback_data=f"send_message_{tg_id}"
         )
     )
     builder.row(InlineKeyboardButton(text="🔙 Назад", callback_data="user_editor"))
@@ -671,19 +706,26 @@ async def handle_user_info(
 
 
 @router.callback_query(F.data.startswith("confirm_delete_user_"), IsAdminFilter())
-async def confirm_delete_user(callback_query: types.CallbackQuery, state: FSMContext, session: Any):
+async def confirm_delete_user(
+    callback_query: types.CallbackQuery, state: FSMContext, session: Any
+):
     tg_id = int(callback_query.data.split("_")[3])
 
     confirmation_markup = InlineKeyboardMarkup(
         row_width=2,
         inline_keyboard=[
-            [InlineKeyboardButton(text="✅ Подтвердить", callback_data=f"delete_user_{tg_id}")],
+            [
+                InlineKeyboardButton(
+                    text="✅ Подтвердить", callback_data=f"delete_user_{tg_id}"
+                )
+            ],
             [InlineKeyboardButton(text="❌ Отменить", callback_data="user_editor")],
         ],
     )
 
     await callback_query.message.answer(
-        f"Вы уверены, что хотите удалить пользователя с ID {tg_id}?", reply_markup=confirmation_markup
+        f"Вы уверены, что хотите удалить пользователя с ID {tg_id}?",
+        reply_markup=confirmation_markup,
     )
 
 
@@ -691,7 +733,9 @@ async def confirm_delete_user(callback_query: types.CallbackQuery, state: FSMCon
 async def delete_user(callback_query: types.CallbackQuery, session: Any):
     tg_id = int(callback_query.data.split("_")[2])
 
-    key_records = await session.fetch("SELECT email, client_id FROM keys WHERE tg_id = $1", tg_id)
+    key_records = await session.fetch(
+        "SELECT email, client_id FROM keys WHERE tg_id = $1", tg_id
+    )
 
     async def delete_keys_from_servers():
         try:
@@ -702,7 +746,9 @@ async def delete_user(callback_query: types.CallbackQuery, session: Any):
                     tasks.append(delete_key_from_cluster(cluster_id, email, client_id))
             await asyncio.gather(*tasks)
         except Exception as e:
-            logger.error(f"Ошибка при удалении ключей с серверов для пользователя {tg_id}: {e}")
+            logger.error(
+                f"Ошибка при удалении ключей с серверов для пользователя {tg_id}: {e}"
+            )
 
     await delete_keys_from_servers()
 
@@ -712,9 +758,13 @@ async def delete_user(callback_query: types.CallbackQuery, session: Any):
         back_button = InlineKeyboardButton(text="🔙 Назад", callback_data="user_editor")
         keyboard = InlineKeyboardMarkup(inline_keyboard=[[back_button]])
 
-        await callback_query.message.answer(f"🗑️ Пользователь с ID {tg_id} был удален.", reply_markup=keyboard)
+        await callback_query.message.answer(
+            f"🗑️ Пользователь с ID {tg_id} был удален.", reply_markup=keyboard
+        )
     except Exception as e:
-        logger.error(f"Ошибка при удалении данных из базы данных для пользователя {tg_id}: {e}")
+        logger.error(
+            f"Ошибка при удалении данных из базы данных для пользователя {tg_id}: {e}"
+        )
         await callback_query.message.answer(
             f"❌ Произошла ошибка при удалении пользователя с ID {tg_id}. Попробуйте снова."
         )