执行 Ruff 安全自动修复

This commit is contained in:
yml2213
2026-08-31 10:28:14 +08:00
parent 2a1d27f953
commit b58c6b4357
145 changed files with 940 additions and 993 deletions
+23 -25
View File
@@ -3,14 +3,11 @@
import asyncio
import csv
import io
import json
import shutil
import threading
from datetime import datetime, timezone
from datetime import UTC, datetime
from typing import cast
from pathlib import Path
from fastapi import (
APIRouter,
Depends,
@@ -37,6 +34,8 @@ from core.huya.cookie_utils import normalize_huya_cookie
from core.huya.device_fingerprint import account_state_dir as _fp_state_dir
from core.huya.device_profile import (
_load_db as _load_profile_db,
)
from core.huya.device_profile import (
_save_db as _save_profile_db,
)
from core.sms_provider import parse_sms_lines
@@ -62,6 +61,7 @@ from ..schemas import (
AccountTag,
BatchAssign,
HuyaAccountOut,
HuyaAppPasswordLoginRequest,
HuyaAutoRegisterBatchOut,
HuyaAutoRegisterRequest,
HuyaAutoRegisterRetryRequest,
@@ -72,7 +72,6 @@ from ..schemas import (
HuyaGoodsOut,
HuyaPasswordAccountImport,
HuyaPasswordLoginRequest,
HuyaAppPasswordLoginRequest,
HuyaPasswordLoginSelectedRequest,
HuyaRechargeGoodsOut,
HuyaRegisterSuccessLogOut,
@@ -81,6 +80,13 @@ from ..schemas import (
HuyaTaskBatchRequest,
HuyaTaskOut,
)
from ..services.audit_service import record_audit
from ..services.huya_register_runner import (
export_success_logs_text,
huya_register_registry,
list_success_logs,
)
from ..services.huya_runner import HuyaBatchRunner, huya_batch_registry
from ..services.huya_service import (
HUYA_CONFIG_FIELDS,
SUPPORTED_TASK_TYPES,
@@ -94,14 +100,6 @@ from ..services.huya_service import (
save_huya_login_cookie_to_account,
upsert_huya_cookie,
)
from ..services.huya_runner import HuyaBatchRunner, huya_batch_registry
from ..services.audit_service import record_audit
from ..services.huya_register_runner import (
export_success_logs_text,
huya_register_registry,
list_success_logs,
)
router = APIRouter(prefix="/api/huya", tags=["虎牙"])
@@ -760,7 +758,7 @@ def sms_login_account(
phone = (req.phone or "").strip()
if phone:
account.game_phone = phone
account.updated_at = datetime.now(timezone.utc)
account.updated_at = datetime.now(UTC)
db.commit()
db.refresh(account)
except ValueError as exc:
@@ -1011,7 +1009,7 @@ def password_login_selected_accounts(
)
if not result.success or not result.cookie:
account.status = "login_failed"
account.updated_at = datetime.now(timezone.utc)
account.updated_at = datetime.now(UTC)
db.commit()
failed_count += 1
results.append(
@@ -1051,7 +1049,7 @@ def password_login_selected_accounts(
)
except HuyaCredentialError as exc:
account.status = "login_failed"
account.updated_at = datetime.now(timezone.utc)
account.updated_at = datetime.now(UTC)
db.commit()
failed_count += 1
results.append(
@@ -1065,7 +1063,7 @@ def password_login_selected_accounts(
)
except (HuyaLoginError, ValueError) as exc:
account.status = "login_failed"
account.updated_at = datetime.now(timezone.utc)
account.updated_at = datetime.now(UTC)
db.commit()
failed_count += 1
results.append(
@@ -1079,7 +1077,7 @@ def password_login_selected_accounts(
)
except Exception as exc:
account.status = "login_failed"
account.updated_at = datetime.now(timezone.utc)
account.updated_at = datetime.now(UTC)
db.commit()
failed_count += 1
results.append(
@@ -1163,7 +1161,7 @@ def app_password_login_selected_accounts(
)
if not result.success or not result.cookie:
account.status = "login_failed"
account.updated_at = datetime.now(timezone.utc)
account.updated_at = datetime.now(UTC)
db.commit()
failed_count += 1
results.append(
@@ -1201,7 +1199,7 @@ def app_password_login_selected_accounts(
)
except HuyaAppQrAuthRequiredError as exc:
account.status = "login_failed"
account.updated_at = datetime.now(timezone.utc)
account.updated_at = datetime.now(UTC)
db.commit()
failed_count += 1
results.append(
@@ -1215,7 +1213,7 @@ def app_password_login_selected_accounts(
)
except HuyaCredentialError as exc:
account.status = "login_failed"
account.updated_at = datetime.now(timezone.utc)
account.updated_at = datetime.now(UTC)
db.commit()
failed_count += 1
results.append(
@@ -1229,7 +1227,7 @@ def app_password_login_selected_accounts(
)
except (HuyaAppLoginError, HuyaLoginError, ValueError) as exc:
account.status = "login_failed"
account.updated_at = datetime.now(timezone.utc)
account.updated_at = datetime.now(UTC)
db.commit()
failed_count += 1
results.append(
@@ -1243,7 +1241,7 @@ def app_password_login_selected_accounts(
)
except Exception as exc:
account.status = "login_failed"
account.updated_at = datetime.now(timezone.utc)
account.updated_at = datetime.now(UTC)
db.commit()
failed_count += 1
results.append(
@@ -1729,7 +1727,7 @@ def update_config(
setattr(config, field, value.strip())
changed_fields.append(field)
apply_huya_config_defaults(config)
config.updated_at = datetime.now(timezone.utc)
config.updated_at = datetime.now(UTC)
if changed_fields:
record_audit(
db,
@@ -1995,7 +1993,7 @@ async def ws_huya_logs(websocket: WebSocket, batch_id: str):
if msg.get("level") == "result":
await asyncio.sleep(0.1)
break
except asyncio.TimeoutError:
except TimeoutError:
await websocket.send_json({"level": "heartbeat", "message": ""})
except WebSocketDisconnect:
pass