Refactor client management functions in client.py to use a dataclass for client configuration. Update add_client, extend_client_key, and delete_client functions to improve parameter handling and logging. Enhance error handling and documentation for better clarity and maintainability.

client.py

@@ -1,76 +1,105 @@
+from dataclasses import dataclass
+from typing import Optional, Dict, Any, List
 import py3xui
 
 from config import LIMIT_IP, SUPERNODE
 from logger import logger
 
 
-async def add_client(
-    xui,
-    client_id: str,
-    email: str,
-    tg_id: str,
-    limit_ip: int,
-    total_gb: int,
-    expiry_time: int,
-    enable: bool,
-    flow: str,
-    inbound_id: int,
-    sub_id,
-):
+@dataclass
+class ClientConfig:
+    """Конфигурация клиента для добавления/обновления."""
+    client_id: str
+    email: str
+    tg_id: str
+    limit_ip: int
+    total_gb: int
+    expiry_time: int
+    enable: bool
+    flow: str
+    inbound_id: int
+    sub_id: str
+
+
+async def add_client(xui: py3xui.API, config: ClientConfig) -> Dict[str, Any]:
     """
-    Adds a client to the server via 3x-ui.
+    Добавляет клиента на сервер через 3x-ui.
+
+    Args:
+        xui: Экземпляр API клиента
+        config: Конфигурация клиента
+
+    Returns:
+        Dict[str, Any]: Результат операции в формате
+            {'status': 'success'|'failed'|'duplicate', 'error': str, 'email': str}
     """
     try:
         await xui.login()
 
         client = py3xui.Client(
-            id=client_id,
-            email=email.lower(),
-            limit_ip=limit_ip,
-            total_gb=total_gb,
-            expiry_time=expiry_time,
-            enable=enable,
-            tg_id=tg_id,
-            sub_id=sub_id,
-            flow=flow,
+            id=config.client_id,
+            email=config.email.lower(),
+            limit_ip=config.limit_ip,
+            total_gb=config.total_gb,
+            expiry_time=config.expiry_time,
+            enable=config.enable,
+            tg_id=config.tg_id,
+            sub_id=config.sub_id,
+            flow=config.flow,
         )
 
-        response = await xui.client.add(inbound_id, [client])
-
-        logger.info(f"Клиент {email} успешно добавлен с ID {client_id}.")
+        response = await xui.client.add(config.inbound_id, [client])
+        logger.info(f"Клиент {config.email} успешно добавлен с ID {config.client_id}")
 
         return response if response else {"status": "failed"}
 
     except Exception as e:
         error_message = str(e)
-
         if "Duplicate email" in error_message:
-            logger.warning(f"Дублированный email: {email}. Пропуск. Сообщение: {error_message}")
-            return {"status": "duplicate", "email": email}
+            logger.warning(f"Дублированный email: {config.email}. Пропуск. Сообщение: {error_message}")
+            return {"status": "duplicate", "email": config.email}
 
-        logger.error(f"Ошибка при добавлении клиента {email}: {error_message}")
+        logger.error(f"Ошибка при добавлении клиента {config.email}: {error_message}")
         return {"status": "failed", "error": error_message}
 
 
 async def extend_client_key(
-    xui, inbound_id, email: str, new_expiry_time: int, client_id: str, total_gb: int, sub_id=str
-):
+    xui: py3xui.API,
+    inbound_id: int,
+    email: str,
+    new_expiry_time: int,
+    client_id: str,
+    total_gb: int,
+    sub_id: str
+) -> Optional[bool]:
     """
-    Функция для обновления срока действия ключа клиента по email.
+    Обновляет срок действия ключа клиента.
+
+    Args:
+        xui: Экземпляр API клиента
+        inbound_id: ID входящего соединения
+        email: Email клиента
+        new_expiry_time: Новое время истечения
+        client_id: ID клиента
+        total_gb: Общий объем трафика
+        sub_id: ID подписки
+
+    Returns:
+        Optional[bool]: True если успешно, False если ошибка, None если клиент не найден
     """
-    await xui.login()
     try:
+        await xui.login()
         client = await xui.client.get_by_email(email)
 
         if not client:
-            logger.warning(f"Клиент с email {email} не найден.")
-            return
+            logger.warning(f"Клиент с email {email} не найден")
+            return None
 
         if not client.id:
-            logger.warning(f"Ошибка: клиент {email} не имеет действительного ID.")
-            return
+            logger.warning(f"Ошибка: клиент {email} не имеет действительного ID")
+            return None
 
-        logger.info(f"Обновление ключа клиента {client.email} с ID {client.id} до нового времени: {new_expiry_time}")
+        logger.info(f"Обновление ключа клиента {email} с ID {client.id} до {new_expiry_time}")
 
         client.id = client_id
         client.expiry_time = new_expiry_time
@@ -83,39 +112,48 @@ async def extend_client_key(
 
         await xui.client.update(client.id, client)
         await xui.client.reset_stats(inbound_id, email)
-        logger.info(f"Ключ клиента {client.email} успешно продлён до {new_expiry_time}.")
+        logger.info(f"Ключ клиента {email} успешно продлён до {new_expiry_time}")
+        return True
 
     except Exception as e:
         logger.error(f"Ошибка при обновлении клиента с email {email}: {e}")
+        return False
 
 
 async def delete_client(
-    xui,
+    xui: py3xui.API,
     inbound_id: int,
     email: str,
     client_id: str,
 ) -> bool:
     """
-    Функция для удаления клиента с сервера 3x-ui.
-    Возвращает True при успешном удалении, иначе False.
+    Удаляет клиента с сервера 3x-ui.
+
+    Args:
+        xui: Экземпляр API клиента
+        inbound_id: ID входящего соединения
+        email: Email клиента
+        client_id: ID клиента
+
+    Returns:
+        bool: True если удаление успешно, False в противном случае
     """
-    await xui.login()
     try:
+        await xui.login()
+
         if SUPERNODE:
             await xui.client.delete(inbound_id, client_id)
-            logger.info(f"Клиент с ID {client_id} был удален успешно (SUPERNODE).")
+            logger.info(f"Клиент с ID {client_id} был удален успешно (SUPERNODE)")
             return True
 
         client = await xui.client.get_by_email(email)
-
         if not client:
-            logger.warning(f"Клиент с email {email} и ID {client_id} не найден.")
+            logger.warning(f"Клиент с email {email} и ID {client_id} не найден")
             return False
 
         client.id = client_id
-
         await xui.client.delete(inbound_id, client.id)
-        logger.info(f"Клиент с ID {client_id} был удален успешно.")
+        logger.info(f"Клиент с ID {client_id} был удален успешно")
         return True
 
     except Exception as e: