登录任务增加代理切换次数设置

- 后端: LoginBatchRequest 增加 max_proxy_retries 字段,默认10,0=无限
- 核心: DouyuLogin 极验验证循环根据 max_proxy_retries 控制重试
- 前端: 筛选栏增加代理切换次数输入框,Tooltip 提示0=无限切换直到成功

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
This commit is contained in:
yml2213
2026-06-22 20:40:30 +08:00
co-authored by Claude Fable 5
parent bc40d334b9
commit 2b8716cf72
6 changed files with 36 additions and 7 deletions
+13 -3
View File
@@ -61,11 +61,13 @@ class DouyuLogin:
proxy_api_url: Optional[str] = None,
timeout: tuple[float, float] = REQUEST_TIMEOUT,
max_geetest_retries: int = 5,
max_proxy_retries: int = 10,
):
self.account = account
self.proxy = proxy
self.timeout = timeout
self.max_geetest_retries = max_geetest_retries
self.max_proxy_retries = max_proxy_retries # 0=无限切换直到成功
self.session = requests.Session()
# 初始化代理管理器
@@ -317,9 +319,15 @@ class DouyuLogin:
"""
logger.info("开始极验 fullpage 验证...")
for attempt in range(self.max_geetest_retries):
# max_proxy_retries=0 表示无限重试直到成功
max_attempts = self.max_proxy_retries if self.max_proxy_retries > 0 else 999999
for attempt in range(max_attempts):
try:
logger.info(f"极验验证尝试 {attempt + 1}/{self.max_geetest_retries}")
if self.max_proxy_retries > 0:
logger.info(f"极验验证尝试 {attempt + 1}/{max_attempts}")
else:
logger.info(f"极验验证尝试 {attempt + 1} (无限重试)")
# 按斗鱼登录页 HAR:fullpage 智能检测流程,不进入图片滑块。
str_16 = _generate_seed()
@@ -372,7 +380,9 @@ class DouyuLogin:
time.sleep(2)
continue
raise ValueError(f"极验验证失败,已重试 {self.max_geetest_retries}")
if self.max_proxy_retries > 0:
raise ValueError(f"极验验证失败,已重试 {max_attempts}")
raise ValueError("极验验证失败(无限重试模式仍未能通过)")
def _second_login(self, gt: str, challenge: str, validate: str,
seccode: str, code_token: str) -> str:
+1
View File
@@ -58,6 +58,7 @@ async def create_batch(
created_by=current.id,
creator_role=current.role,
max_geetest_retries=req.max_geetest_retries,
max_proxy_retries=req.max_proxy_retries,
proxy_config=proxy,
log_queue=log_queue,
loop=loop,
+1
View File
@@ -125,6 +125,7 @@ class AccountOut(BaseModel):
class LoginBatchRequest(BaseModel):
account_ids: list[int]
max_geetest_retries: int = 5
max_proxy_retries: int = 10 # 代理切换次数,0=无限切换直到成功
concurrency: int = 3 # 并发数,1-10
+3
View File
@@ -26,6 +26,7 @@ class LoginBatchRunner:
created_by: int,
creator_role: str,
max_geetest_retries: int = 5,
max_proxy_retries: int = 10,
proxy_config: Optional[ProxyConfigModel] = None,
log_queue: Optional[asyncio.Queue] = None,
loop: Optional[asyncio.AbstractEventLoop] = None,
@@ -36,6 +37,7 @@ class LoginBatchRunner:
self.created_by = created_by
self.creator_role = creator_role
self.max_geetest_retries = max_geetest_retries
self.max_proxy_retries = max_proxy_retries
self.proxy_config = proxy_config
self.log_queue = log_queue
self.loop = loop
@@ -126,6 +128,7 @@ class LoginBatchRunner:
account,
proxy=proxy_dict,
max_geetest_retries=self.max_geetest_retries,
max_proxy_retries=self.max_proxy_retries,
)
result = loginer.login()
+2 -2
View File
@@ -56,8 +56,8 @@ export const accountApi = {
};
export const loginApi = {
createBatch: (account_ids: number[], max_geetest_retries?: number, concurrency?: number) =>
api.post<any, any>('/login/batch', { account_ids, max_geetest_retries, concurrency }),
createBatch: (account_ids: number[], max_geetest_retries?: number, concurrency?: number, max_proxy_retries?: number) =>
api.post<any, any>('/login/batch', { account_ids, max_geetest_retries, concurrency, max_proxy_retries }),
listTasks: (batch_id?: string) =>
api.get<any, any[]>('/login/tasks', { params: batch_id ? { batch_id } : {} }),
stop: (batch_id: string) => api.post<any, any>(`/login/stop/${batch_id}`),
+16 -2
View File
@@ -2,7 +2,7 @@ import { useEffect, useState, useRef, useMemo, useCallback } from 'react';
import {
Table, Button, Select, message, Tag, Space, Spin, InputNumber, Tooltip, Popconfirm, theme,
} from 'antd';
import { PlayCircleOutlined, StopOutlined, FilterOutlined, ThunderboltOutlined, ReloadOutlined, DownOutlined, UpOutlined, DeleteOutlined } from '@ant-design/icons';
import { PlayCircleOutlined, StopOutlined, FilterOutlined, ThunderboltOutlined, ReloadOutlined, DownOutlined, UpOutlined, DeleteOutlined, SwapOutlined } from '@ant-design/icons';
import { accountApi, loginApi } from '../api/modules';
import { getUser, hasPerm } from '../store/auth';
import { formatTime } from '../utils/time';
@@ -33,6 +33,7 @@ export default function LoginTasksPage() {
const [wsConnected, setWsConnected] = useState(false);
const [selectedTags, setSelectedTags] = useState<string[]>([]);
const [concurrency, setConcurrency] = useState(3);
const [maxProxyRetries, setMaxProxyRetries] = useState(10);
const [logVisible, setLogVisible] = useState(true);
const [selectedRowKeys, setSelectedRowKeys] = useState<number[]>([]);
const wsRef = useRef<WebSocket | null>(null);
@@ -134,7 +135,7 @@ export default function LoginTasksPage() {
setLoading(true);
setLogs([]);
try {
const result = await loginApi.createBatch(accountIds, 5, concurrency);
const result = await loginApi.createBatch(accountIds, 5, concurrency, maxProxyRetries);
setBatchId(result.batch_id);
message.success(`已创建登录任务,共 ${result.count} 个账号`);
@@ -346,6 +347,19 @@ export default function LoginTasksPage() {
/>
</Space>
</Tooltip>
<Tooltip title="代理切换次数,0=无限切换直到成功">
<Space size={4}>
<SwapOutlined style={{ color: token.colorTextSecondary }} />
<InputNumber
min={0}
max={999}
value={maxProxyRetries}
onChange={(v) => setMaxProxyRetries(v ?? 10)}
style={{ width: 55 }}
size="small"
/>
</Space>
</Tooltip>
<Button
type="primary"
icon={<PlayCircleOutlined />}