From 4ebe7399ba73e7df52533abe76d85d7d23d8b847 Mon Sep 17 00:00:00 2001
From: algizn97 <algiz_n@mail.ru>
Date: Thu, 21 Aug 2025 13:36:11 +0500
Subject: [PATCH] Update function

---
 app/services/Bybit/functions/balance.py | 25 ++++++++++++-------------
 1 file changed, 12 insertions(+), 13 deletions(-)

diff --git a/app/services/Bybit/functions/balance.py b/app/services/Bybit/functions/balance.py
index db3d74d..5785325 100644
--- a/app/services/Bybit/functions/balance.py
+++ b/app/services/Bybit/functions/balance.py
@@ -2,7 +2,8 @@
 
 from pybit.unified_trading import HTTP
 
-client = HTTP()
+import logging
+logging.basicConfig(level=logging.DEBUG)
 
 async def get_balance(tg_id, message):
     api_key = await rq.get_bybit_api_key(tg_id)
@@ -15,19 +16,17 @@ async def get_balance(tg_id, message):
 
     if api_key == 'None' or secret_key == 'None':
         await message.answer('⚠️ Подключите платформу для торговли')
-        return 0      
+        return 0
 
     try:
-        check_user = client.get_wallet_balance()
-
-        if check_user:
-            try:
-                balance = client.get_wallet_balance(accountType='UNIFIED', coin='USDT')['result']['list'][0]['coin'][0]['walletBalance']
-
-                return balance
-            except Exception as e:
-                await message.answer('⚠️ Ошибка при получении баланса пользователя') 
-                return 0
+        response = client.get_wallet_balance(accountType='UNIFIED')
+        if response['retCode'] == 0:
+            total_balance = response['result']['list'][0].get('totalWalletBalance', '0')
+            return total_balance
+        else:
+            await message.answer(f"⚠️ Ошибка API: {response.get('retMsg')}")
+            return 0
     except Exception as e:
-        await message.answer('⚠️ Неверные данные API, перепроверьте их')
+        logging.error(f"Ошибка при получении общего баланса: {e}")
+        await message.answer('⚠️ Ошибка при получении баланса')
         return 0
\ No newline at end of file