0131c8b408
- Add Free_Proxy_Website-style fpw_* plugins and register them - Per-plugin crawl timeout (crawl_timeout_seconds=120); remove global crawl_timeout setting - Validator: fix connect vs total timeout on save; SOCKS session LRU cache; drop redundant semaphore - Validation handler uses single DB connection; batch upsert after crawl; WorkerPool put_nowait - Remove unused max_retries from settings API/UI; settings maintenance SQL + init_db cleanup of deprecated keys - WebSocket dashboard stats; ProxyList pool_filter and API alignment - POST /api/proxies/delete-one for IPv6-safe deletes; task poll stops on 404 - pytest uses PROXYPOOL_DB_PATH=db/proxies.test.sqlite so tests do not wipe production DB - .gitignore: explicit proxies.test.sqlite patterns; fix plugin_service ValidationException import Made-with: Cursor
108 lines
3.7 KiB
Python
108 lines
3.7 KiB
Python
"""代理相关路由（含统计信息）"""
|
||
from typing import Optional
|
||
from fastapi import APIRouter, Depends, Query
|
||
from fastapi.responses import StreamingResponse
|
||
|
||
from app.services.proxy_service import ProxyService
|
||
from app.services.scheduler_service import SchedulerService
|
||
from app.services.dashboard_stats import get_dashboard_stats
|
||
from app.models.schemas import ProxyListRequest, BatchDeleteRequest, ProxyDeleteItem
|
||
from app.api.deps import get_proxy_service, get_scheduler_service
|
||
from app.api.common import success_response, format_proxy
|
||
from app.core.exceptions import ProxyPoolException, ProxyNotFoundException
|
||
|
||
router = APIRouter(prefix="/api/proxies", tags=["proxies"])
|
||
|
||
|
||
@router.get("/stats")
|
||
async def get_stats(
|
||
scheduler_service: SchedulerService = Depends(get_scheduler_service),
|
||
):
|
||
stats = await get_dashboard_stats(scheduler_service.running)
|
||
return success_response("获取统计信息成功", stats)
|
||
|
||
|
||
@router.post("")
|
||
async def list_proxies(
|
||
request: ProxyListRequest,
|
||
service: ProxyService = Depends(get_proxy_service),
|
||
):
|
||
proxies, total = await service.list_proxies(
|
||
page=request.page,
|
||
page_size=request.page_size,
|
||
protocol=request.protocol,
|
||
min_score=request.min_score,
|
||
max_score=request.max_score,
|
||
sort_by=request.sort_by,
|
||
sort_order=request.sort_order,
|
||
pool_filter=request.pool_filter,
|
||
)
|
||
return success_response(
|
||
"获取代理列表成功",
|
||
{
|
||
"list": [format_proxy(p) for p in proxies],
|
||
"total": total,
|
||
"page": request.page,
|
||
"page_size": request.page_size,
|
||
},
|
||
)
|
||
|
||
|
||
@router.get("/random")
|
||
async def get_random_proxy(service: ProxyService = Depends(get_proxy_service)):
|
||
proxy = await service.get_random_proxy()
|
||
if not proxy:
|
||
raise ProxyPoolException("暂无可用代理", 404)
|
||
return success_response("获取随机代理成功", format_proxy(proxy))
|
||
|
||
|
||
@router.get("/export/{fmt}")
|
||
async def export_proxies(
|
||
fmt: str,
|
||
protocol: Optional[str] = None,
|
||
limit: int = Query(default=10000, ge=1, le=100000),
|
||
service: ProxyService = Depends(get_proxy_service),
|
||
):
|
||
if fmt not in ("csv", "txt", "json"):
|
||
raise ProxyPoolException("不支持的导出格式", 400)
|
||
|
||
media_types = {"csv": "text/csv", "txt": "text/plain", "json": "application/json"}
|
||
|
||
return StreamingResponse(
|
||
service.export_proxies(fmt, protocol, limit),
|
||
media_type=media_types[fmt],
|
||
headers={"Content-Disposition": f"attachment; filename=proxies.{fmt}"},
|
||
)
|
||
|
||
|
||
@router.post("/delete-one")
|
||
async def delete_proxy_one(
|
||
item: ProxyDeleteItem,
|
||
service: ProxyService = Depends(get_proxy_service),
|
||
):
|
||
"""JSON 删除（推荐）：IPv6 等含冒号 IP 不受路径分段影响。"""
|
||
await service.delete_proxy(item.ip, item.port)
|
||
return success_response("删除代理成功")
|
||
|
||
|
||
@router.delete("/{ip}/{port}")
|
||
async def delete_proxy(ip: str, port: int, service: ProxyService = Depends(get_proxy_service)):
|
||
await service.delete_proxy(ip, port)
|
||
return success_response("删除代理成功")
|
||
|
||
|
||
@router.post("/batch-delete")
|
||
async def batch_delete(
|
||
request: BatchDeleteRequest,
|
||
service: ProxyService = Depends(get_proxy_service),
|
||
):
|
||
proxies = [(item.ip, item.port) for item in request.proxies]
|
||
deleted = await service.batch_delete(proxies)
|
||
return success_response(f"批量删除 {deleted} 个代理成功", {"deleted_count": deleted})
|
||
|
||
|
||
@router.delete("/clean-invalid")
|
||
async def clean_invalid(service: ProxyService = Depends(get_proxy_service)):
|
||
count = await service.clean_invalid()
|
||
return success_response(f"清理了 {count} 个无效代理", {"deleted_count": count})
|