Stage 07.3.1 - auto trading background runner and live screen

app/src/trading/auto/runner.py

@@ -0,0 +1,130 @@
+# app/src/trading/auto/runner.py
+
+from __future__ import annotations
+
+import asyncio
+from typing import Callable
+
+from aiogram import Bot
+
+from src.trading.auto.service import AutoTradeService
+
+
+class AutoTradeRunner:
+    _task: asyncio.Task | None = None
+    _bot: Bot | None = None
+    _chat_id: int | None = None
+    _message_id: int | None = None
+    _render_text: Callable[[], str] | None = None
+    _render_markup: Callable[[], object] | None = None
+    _current_screen: str | None = None
+    _interval_seconds = 5
+
+    # зарегистрировать live-экран для автообновления
+    @classmethod
+    def register_screen(
+        cls,
+        *,
+        bot: Bot,
+        chat_id: int,
+        message_id: int,
+        render_text: Callable[[], str],
+        render_markup: Callable[[], object],
+    ) -> None:
+        cls._bot = bot
+        cls._chat_id = chat_id
+        cls._message_id = message_id
+        cls._render_text = render_text
+        cls._render_markup = render_markup
+
+    # удалить ранее зарегистрированный live-экран
+    @classmethod
+    async def delete_registered_screen(
+        cls,
+        *,
+        bot: Bot,
+        chat_id: int,
+    ) -> None:
+        if cls._chat_id is None or cls._message_id is None:
+            return
+
+        if cls._chat_id != chat_id:
+            return
+
+        try:
+            await bot.delete_message(
+                chat_id=cls._chat_id,
+                message_id=cls._message_id,
+            )
+        except Exception:
+            pass
+
+        cls._message_id = None
+        cls._render_text = None
+        cls._render_markup = None
+
+    # переключить активный экран
+    @classmethod
+    def set_current_screen(cls, screen: str) -> None:
+        cls._current_screen = screen
+
+    # запустить background runner
+    @classmethod
+    def start(cls) -> None:
+        if cls._task is not None and not cls._task.done():
+            return
+
+        cls._task = asyncio.create_task(cls._worker())
+
+    # остановить background runner
+    @classmethod
+    def stop(cls) -> None:
+        if cls._task is None:
+            return
+
+        cls._task.cancel()
+        cls._task = None
+
+    # background loop автоторговли
+    @classmethod
+    async def _worker(cls) -> None:
+        service = AutoTradeService()
+
+        while True:
+            state = service.get_state()
+
+            if state.status == "OFF":
+                cls._task = None
+                break
+
+            service.run_cycle()
+
+            await cls._refresh_screen()
+            await asyncio.sleep(cls._interval_seconds)
+
+    # обновить live-экран Telegram
+    @classmethod
+    async def _refresh_screen(cls) -> None:
+        if cls._current_screen != "auto":
+            return
+
+        if not all(
+            [
+                cls._bot,
+                cls._chat_id,
+                cls._message_id,
+                cls._render_text,
+                cls._render_markup,
+            ]
+        ):
+            return
+
+        try:
+            await cls._bot.edit_message_text(
+                chat_id=cls._chat_id,
+                message_id=cls._message_id,
+                text=cls._render_text(),
+                reply_markup=cls._render_markup(),
+            )
+        except Exception:
+            pass