Merge pull request 'Switched to demo mode' (#26) from Alex/stcs:devel into stable

Reviewed-on: kodorvan/stcs#26

app/bybit/__init__.py

@@ -15,7 +15,7 @@ async def get_bybit_client(tg_id: int) -> HTTP | None:
     """
     try:
         api_key, api_secret = await rq.get_user_api(tg_id=tg_id)
-        return HTTP(api_key=api_key, api_secret=api_secret)
+        return HTTP(demo=True, api_key=api_key, api_secret=api_secret)
     except Exception as e:
         logger.error("Error getting bybit client for user %s: %s", tg_id, e)
         return None

app/bybit/open_positions.py

@@ -152,11 +152,13 @@ async def trading_cycle_profit(
         else:
             s_side = side
 
+        next_series = current_series + 1
+
         await rq.set_user_deal(
             tg_id=tg_id,
             symbol=symbol,
             current_step=1,
-            current_series=current_series,
+            current_series=next_series,
             trade_mode=trade_mode,
             side_mode=side_mode,
             margin_type=margin_type,

app/bybit/telegram_message_handler.py

@@ -190,10 +190,6 @@ class TelegramMessageHandler:
                     res = await trading_cycle_profit(
                         tg_id=tg_id, symbol=symbol, side=r_side
                     )
-                    current_series += 1
-                    await rq.set_current_series(
-                        tg_id=tg_id, symbol=symbol, current_series=current_series
-                    )
                     if res == "OK":
                         pass
                     else:

app/bybit/web_socket.py

@@ -58,7 +58,8 @@ class WebSocketBot:
                     )
                     logger.info("User %s connected to WebSocket", tg_id)
                 else:
-                    await asyncio.sleep(30)
+                    await asyncio.sleep(5)
+                    await self.try_connect_user(api_key, api_secret, tg_id)
 
             await asyncio.sleep(10)
 
@@ -73,6 +74,7 @@ class WebSocketBot:
         """Try to connect a user to the WebSocket."""
         try:
             self.ws_private = WebSocket(
+                demo=True,
                 testnet=False,
                 channel_type="private",
                 api_key=api_key,