fix quality code

2026-02-01 23:03:23 +03:00
parent 731e68a597
commit f8962225ee
106 changed files with 8456 additions and 5810 deletions
--- a/helper_bot/handlers/admin/services.py
+++ b/helper_bot/handlers/admin/services.py
@@ -1,11 +1,16 @@
 from datetime import datetime
 from typing import List, Optional

-from helper_bot.handlers.admin.exceptions import (InvalidInputError,
-                                                  UserAlreadyBannedError)
-from helper_bot.utils.helper_func import (add_days_to_date,
-                                          get_banned_users_buttons,
-                                          get_banned_users_list)
+from helper_bot.handlers.admin.exceptions import (
+    InvalidInputError,
+    UserAlreadyBannedError,
+)
+from helper_bot.utils.helper_func import (
+    add_days_to_date,
+    get_banned_users_buttons,
+    get_banned_users_list,
+)
+
 # Local imports - metrics
 from helper_bot.utils.metrics import track_errors, track_time
 from logs.custom_logger import logger
@@ -13,6 +18,7 @@ from logs.custom_logger import logger

 class User:
    """Модель пользователя"""
+
    def __init__(self, user_id: int, username: str, full_name: str):
        self.user_id = user_id
        self.username = username
@@ -21,7 +27,10 @@ class User:

 class BannedUser:
    """Модель заблокированного пользователя"""
-    def __init__(self, user_id: int, username: str, reason: str, unban_date: Optional[datetime]):
+
+    def __init__(
+        self, user_id: int, username: str, reason: str, unban_date: Optional[datetime]
+    ):
        self.user_id = user_id
        self.username = username
        self.reason = reason
@@ -30,10 +39,10 @@ class BannedUser:

 class AdminService:
    """Сервис для административных операций"""
-    
+
    def __init__(self, bot_db):
        self.bot_db = bot_db
-    
+
    @track_time("get_last_users", "admin_service")
    @track_errors("admin_service", "get_last_users")
    async def get_last_users(self) -> List[User]:
@@ -41,17 +50,13 @@ class AdminService:
        try:
            users_data = await self.bot_db.get_last_users(30)
            return [
-                User(
-                    user_id=user[1],
-                    username='Неизвестно',
-                    full_name=user[0]
-                )
+                User(user_id=user[1], username="Неизвестно", full_name=user[0])
                for user in users_data
            ]
        except Exception as e:
            logger.error(f"Ошибка при получении списка последних пользователей: {e}")
            raise
-    
+
    @track_time("get_banned_users", "admin_service")
    @track_errors("admin_service", "get_banned_users")
    async def get_banned_users(self) -> List[BannedUser]:
@@ -65,18 +70,22 @@ class AdminService:
                username = await self.bot_db.get_username(user_id)
                full_name = await self.bot_db.get_full_name_by_id(user_id)
                user_name = username or full_name or f"User_{user_id}"
-                
-                banned_users.append(BannedUser(
-                    user_id=user_id,
-                    username=user_name,
-                    reason=reason,
-                    unban_date=unban_date
-                ))
+
+                banned_users.append(
+                    BannedUser(
+                        user_id=user_id,
+                        username=user_name,
+                        reason=reason,
+                        unban_date=unban_date,
+                    )
+                )
            return banned_users
        except Exception as e:
-            logger.error(f"Ошибка при получении списка заблокированных пользователей: {e}")
+            logger.error(
+                f"Ошибка при получении списка заблокированных пользователей: {e}"
+            )
            raise
-    
+
    @track_time("get_user_by_username", "admin_service")
    @track_errors("admin_service", "get_user_by_username")
    async def get_user_by_username(self, username: str) -> Optional[User]:
@@ -85,17 +94,15 @@ class AdminService:
            user_id = await self.bot_db.get_user_id_by_username(username)
            if not user_id:
                return None
-            
+
            full_name = await self.bot_db.get_full_name_by_id(user_id)
            return User(
-                user_id=user_id,
-                username=username,
-                full_name=full_name or 'Неизвестно'
+                user_id=user_id, username=username, full_name=full_name or "Неизвестно"
            )
        except Exception as e:
            logger.error(f"Ошибка при поиске пользователя по username {username}: {e}")
            raise
-    
+
    @track_time("get_user_by_id", "admin_service")
    @track_errors("admin_service", "get_user_by_id")
    async def get_user_by_id(self, user_id: int) -> Optional[User]:
@@ -104,39 +111,50 @@ class AdminService:
            user_info = await self.bot_db.get_user_by_id(user_id)
            if not user_info:
                return None
-            
+
            return User(
                user_id=user_id,
-                username=user_info.username or 'Неизвестно',
-                full_name=user_info.full_name or 'Неизвестно'
+                username=user_info.username or "Неизвестно",
+                full_name=user_info.full_name or "Неизвестно",
            )
        except Exception as e:
            logger.error(f"Ошибка при поиске пользователя по ID {user_id}: {e}")
            raise
-    
+
    @track_time("ban_user", "admin_service")
    @track_errors("admin_service", "ban_user")
-    async def ban_user(self, user_id: int, username: str, reason: str, ban_days: Optional[int], ban_author_id: int) -> None:
+    async def ban_user(
+        self,
+        user_id: int,
+        username: str,
+        reason: str,
+        ban_days: Optional[int],
+        ban_author_id: int,
+    ) -> None:
        """Заблокировать пользователя"""
        try:
            # Проверяем, не заблокирован ли уже пользователь
            if await self.bot_db.check_user_in_blacklist(user_id):
                raise UserAlreadyBannedError(f"Пользователь {user_id} уже заблокирован")
-            
+
            # Рассчитываем дату разблокировки
            date_to_unban = None
            if ban_days is not None:
                date_to_unban = add_days_to_date(ban_days)
-            
+
            # Сохраняем в БД (username больше не передается, так как не используется в новой схеме)
-            await self.bot_db.set_user_blacklist(user_id, None, reason, date_to_unban, ban_author=ban_author_id)
-            
-            logger.info(f"Пользователь {user_id} ({username}) заблокирован. Причина: {reason}, срок: {ban_days} дней")
-            
+            await self.bot_db.set_user_blacklist(
+                user_id, None, reason, date_to_unban, ban_author=ban_author_id
+            )
+
+            logger.info(
+                f"Пользователь {user_id} ({username}) заблокирован. Причина: {reason}, срок: {ban_days} дней"
+            )
+
        except Exception as e:
            logger.error(f"Ошибка при блокировке пользователя {user_id}: {e}")
            raise
-    
+
    @track_time("unban_user", "admin_service")
    @track_errors("admin_service", "unban_user")
    async def unban_user(self, user_id: int) -> None:
@@ -147,7 +165,7 @@ class AdminService:
        except Exception as e:
            logger.error(f"Ошибка при разблокировке пользователя {user_id}: {e}")
            raise
-    
+
    @track_time("validate_user_input", "admin_service")
    @track_errors("admin_service", "validate_user_input")
    async def validate_user_input(self, input_text: str) -> int:
@@ -155,11 +173,13 @@ class AdminService:
        try:
            user_id = int(input_text.strip())
            if user_id <= 0:
-                raise InvalidInputError("ID пользователя должен быть положительным числом")
+                raise InvalidInputError(
+                    "ID пользователя должен быть положительным числом"
+                )
            return user_id
        except ValueError:
            raise InvalidInputError("ID пользователя должен быть числом")
-    
+
    @track_time("get_banned_users_for_display", "admin_service")
    @track_errors("admin_service", "get_banned_users_for_display")
    async def get_banned_users_for_display(self, page: int = 0) -> tuple[str, list]:
@@ -170,5 +190,7 @@ class AdminService:
            buttons_list = await get_banned_users_buttons(self.bot_db)
            return message_text, buttons_list
        except Exception as e:
-            logger.error(f"Ошибка при получении данных заблокированных пользователей: {e}")
+            logger.error(
+                f"Ошибка при получении данных заблокированных пользователей: {e}"
+            )
            raise