完善虎牙积分查询日志

This commit is contained in:
yml2213
2026-07-04 17:57:33 +08:00
parent 56ffaf531c
commit 3b88653e5d
3 changed files with 54 additions and 16 deletions
+5 -1
View File
@@ -6,6 +6,7 @@ from concurrent.futures import ThreadPoolExecutor, as_completed
from datetime import datetime, timezone
from typing import Optional
from loguru import logger
from sqlalchemy.orm import Session, joinedload
from core.huya import HuyaHttpClient
@@ -42,6 +43,9 @@ class HuyaBatchRunner:
self._stop.set()
def _push_log(self, level: str, message: str):
if level != "result" and message:
log_func = getattr(logger, level, logger.info)
log_func(f"[huya] {message}")
if self.log_queue and self.loop:
asyncio.run_coroutine_threadsafe(
self.log_queue.put({"level": level, "message": message}),
@@ -113,7 +117,7 @@ class HuyaBatchRunner:
self._mark_task(worker_db, task, "failed", "账号 Cookie 为空")
return
client = HuyaHttpClient(logger=lambda _msg: None)
client = HuyaHttpClient(logger=lambda msg: self._push_log("info", f"[{uid}] {msg}"))
response = client.query_user_score(uid=uid, cookie=cookie, sid=sid_int)
if response is None:
self._mark_task(worker_db, task, "error", "虎牙积分接口无响应")