redesign/bug fixes/admin trial

client.py

@@ -1,6 +1,8 @@
 import py3xui
 from loguru import logger
 
+from config import TOTAL_GB
+
 
 async def add_client(
     xui,
@@ -64,9 +66,14 @@ async def extend_client_key(xui, email: str, new_expiry_time: int, client_id: st
         logger.info(
             f"Обновление ключа клиента {client.email} с ID {client.id} до нового времени: {new_expiry_time}"
         )
+        
         client.id = client_id
         client.expiry_time = new_expiry_time
-
+        
+        if TOTAL_GB > 0:
+            client.total_gb = TOTAL_GB
+            logger.info(f"Установлен объем трафика для клиента {client.email}: {TOTAL_GB} ГБ")
+        
         await xui.client.update(client.id, client)
         logger.info(
             f"Ключ клиента {client.email} успешно продлён до {new_expiry_time}."
@@ -76,6 +83,7 @@ async def extend_client_key(xui, email: str, new_expiry_time: int, client_id: st
         logger.error(f"Ошибка при обновлении клиента с email {email}: {e}")
 
 
+
 async def delete_client(
     xui,
     email: str,