add balance

This commit is contained in:
Yuriy Yuriev
2026-05-14 21:04:28 +07:00
parent bdf3fef7ec
commit fe9ada0558
7 changed files with 734 additions and 170 deletions
+16 -8
View File
@@ -48,7 +48,8 @@ class TaskParams:
successful_visits: int = 0
links_found: int = 0
chat_id: Optional[int] = None
user_id: Optional[int] = None # Telegram user_id, если задача создана пользователем
completed: bool = False # Задача завершена
stopped: bool = False # Задача остановлена пользователем
@@ -241,15 +242,22 @@ class TaskManager:
async def get_stats(self) -> dict:
"""Общая статистика."""
twitch_tasks = await self.get_twitch_tasks()
visit_tasks = await self.get_visit_tasks()
twitch = visit = paused = active = 0
for p in self._tasks.values():
if p.task_type == "twitch_irc":
twitch += 1
elif p.task_type == "visit":
visit += 1
if p.paused:
paused += 1
else:
active += 1
return {
"total": len(self._tasks),
"twitch": len(twitch_tasks),
"visit": len(visit_tasks),
"paused": sum(1 for p in self._tasks.values() if p.paused),
"active": sum(1 for p in self._tasks.values() if not p.paused),
"twitch": twitch,
"visit": visit,
"paused": paused,
"active": active,
}
def _get_task_info(self, task_id: str, params: TaskParams) -> str: