执行 Ruff 安全自动修复
This commit is contained in:
@@ -6,8 +6,7 @@ import threading
|
||||
import uuid
|
||||
from concurrent.futures import ThreadPoolExecutor, as_completed
|
||||
from dataclasses import dataclass, field
|
||||
from datetime import datetime, timezone
|
||||
from typing import Optional
|
||||
from datetime import UTC, datetime
|
||||
|
||||
from core.douyu.proxy_fetcher import ProxyFetcher
|
||||
from core.huya.auto_register import HuyaAutoRegisterResult, register_huya_with_sms_line
|
||||
@@ -17,13 +16,18 @@ from core.sms_provider import SmsLine
|
||||
from ..database import SessionLocal
|
||||
from ..models import (
|
||||
HuyaRegisterBatch as HuyaRegisterBatchModel,
|
||||
)
|
||||
from ..models import (
|
||||
HuyaRegisterItem as HuyaRegisterItemModel,
|
||||
)
|
||||
from ..models import (
|
||||
HuyaRegisterSuccessLog,
|
||||
)
|
||||
from ..models import (
|
||||
ProxyConfig as ProxyConfigModel,
|
||||
)
|
||||
from .huya_service import upsert_huya_cookie
|
||||
|
||||
|
||||
# 运行中状态:服务中断后视为未完成,可继续
|
||||
RUNNING_ITEM_STATUSES = frozenset({"sending", "waiting", "changing", "logging"})
|
||||
# 续跑(continue):只跑「还没跑完」的,不重试已明确失败(error)、不重跑成功(success)
|
||||
@@ -37,7 +41,7 @@ TERMINAL_BATCH_STATUSES = frozenset({"finished", "stopped", "error", "interrupte
|
||||
|
||||
|
||||
def _now() -> datetime:
|
||||
return datetime.now(timezone.utc)
|
||||
return datetime.now(UTC)
|
||||
|
||||
|
||||
def _cookie_preview(cookie: str) -> str:
|
||||
@@ -409,7 +413,7 @@ class HuyaRegisterRunner:
|
||||
self,
|
||||
batch: HuyaRegisterBatch,
|
||||
sms_lines: list[SmsLine],
|
||||
proxy_config: Optional[ProxyConfigModel] = None,
|
||||
proxy_config: ProxyConfigModel | None = None,
|
||||
*,
|
||||
item_indices: list[int] | None = None,
|
||||
):
|
||||
@@ -812,7 +816,7 @@ class HuyaRegisterRegistry:
|
||||
password_prefix: str = "hy",
|
||||
fixed_password: str = "",
|
||||
use_proxy: bool = False,
|
||||
proxy_config: Optional[ProxyConfigModel] = None,
|
||||
proxy_config: ProxyConfigModel | None = None,
|
||||
) -> HuyaRegisterRunner:
|
||||
batch_id = uuid.uuid4().hex[:12]
|
||||
concurrency = max(1, min(int(concurrency or 1), 5))
|
||||
@@ -918,7 +922,7 @@ class HuyaRegisterRegistry:
|
||||
def retry(
|
||||
self,
|
||||
batch_id: str,
|
||||
proxy_config: Optional[ProxyConfigModel] = None,
|
||||
proxy_config: ProxyConfigModel | None = None,
|
||||
*,
|
||||
mode: str = "continue",
|
||||
concurrency: int | None = None,
|
||||
|
||||
Reference in New Issue
Block a user