Deleted tasks.py

app/services/Bybit/functions/bybit_ws.py

@@ -3,26 +3,53 @@ import logging.config
 from pybit.unified_trading import WebSocket
 from websocket import WebSocketConnectionClosedException
 from logger_helper.logger_helper import LOGGING_CONFIG
+import app.telegram.database.requests as rq
 
 logging.config.dictConfig(LOGGING_CONFIG)
 logger = logging.getLogger("bybit_ws")
 
 event_loop = None  # Сюда нужно будет установить event loop из основного приложения
 
+def get_or_create_event_loop() -> asyncio.AbstractEventLoop:
+    """
+    Возвращает текущий активный цикл событий asyncio или создает новый, если его нет.
+    """
+    try:
+        return asyncio.get_running_loop()
+    except RuntimeError:
+        loop = asyncio.new_event_loop()
+        asyncio.set_event_loop(loop)
+        return loop
+
 
 def set_event_loop(loop: asyncio.AbstractEventLoop):
     global event_loop
     event_loop = loop
 
 
-def on_execution_callback(message, msg):
+async def run_ws_for_user(tg_id, message) -> None:
     """
-    Callback на событие исполнения сделки.
-    Безопасно запускает асинхронный обработчик из sync callback.
+    Запускает WebSocket Bybit для пользователя с указанным tg_id.
     """
+
+    api_key = await rq.get_bybit_api_key(tg_id)
+    api_secret = await rq.get_bybit_secret_key(tg_id)
+
+    await start_execution_ws(api_key, api_secret, message)
+
+
+def on_order_callback(message, msg):
     if event_loop is not None:
-        from app.services.Bybit.functions.Futures import handle_execution_message  # Импорт внутри, чтобы избежать циклических импортов
-        asyncio.run_coroutine_threadsafe(handle_execution_message(message, msg), event_loop)
+        from app.services.Bybit.functions.Futures import handle_order_message
+        asyncio.run_coroutine_threadsafe(handle_order_message(message, msg), event_loop)
+    else:
+        logger.error("Event loop не установлен, callback пропущен.")
+
+
+def on_execution_callback(message, ws_msg):
+    if event_loop is not None:
+        from app.services.Bybit.functions.Futures import handle_execution_message
+        asyncio.run_coroutine_threadsafe(handle_execution_message(message, ws_msg), event_loop)
     else:
         logger.error("Event loop не установлен, callback пропущен.")
 
@@ -35,8 +62,14 @@ async def start_execution_ws(api_key: str, api_secret: str, message):
     reconnect_delay = 5
     while True:
         try:
+            if not api_key or not api_secret:
+                logger.error("API_KEY и API_SECRET должны быть указаны для подключения к приватным каналам.")
+                await asyncio.sleep(reconnect_delay)
+                continue
             ws = WebSocket(api_key=api_key, api_secret=api_secret, testnet=False, channel_type="private")
-            ws.execution_stream(lambda msg: on_execution_callback(message, msg))
+
+            ws.subscribe("order", lambda ws_msg: on_order_callback(message, ws_msg))
+            ws.subscribe("execution", lambda ws_msg: on_execution_callback(message, ws_msg))
             while True:
                 await asyncio.sleep(1)  # Поддержание активности
         except WebSocketConnectionClosedException: