209a744d94
后端重构： - 新增分层架构：API Routes -> Services -> Repositories -> Infrastructure - 彻底移除全局单例，全面采用 FastAPI 依赖注入 - 新增 api/ 目录拆分路由（proxies, plugins, scheduler, settings, stats） - 新增 services/ 业务逻辑层：ProxyService, PluginService, SchedulerService, ValidatorService, SettingsService - 新增 repositories/ 数据访问层：ProxyRepository, SettingsRepository, PluginSettingsRepository - 新增 models/ 层：Pydantic Schemas + Domain Models - 重写 core/config.py：采用 Pydantic Settings 管理配置 - 新增 core/db.py：基于 asynccontextmanager 的连接管理，支持数据库迁移 - 新增 core/exceptions.py：统一业务异常体系 插件系统重构（核心）： - 新增 core/plugin_system/：BaseCrawlerPlugin + PluginRegistry - 采用显式注册模式（装饰器 + plugins/__init__.py），类型安全、测试友好 - 新增 plugins/base.py：BaseHTTPPlugin 通用 HTTP 爬虫基类 - 迁移全部 7 个插件到新架构（fate0, proxylist_download, ip3366, ip89, kuaidaili, speedx, yundaili） - 插件状态持久化到 plugin_settings 表 任务调度重构： - 新增 core/tasks/queue.py：ValidationQueue + WorkerPool - 解耦爬取与验证：爬虫只负责爬取，代理提交队列后由 Worker 异步验证 - 调度器定时从数据库拉取存量代理并分批投入验证队列 前端调整： - 新增 frontend/src/services/ 层拆分 API 调用逻辑 - 调整 stores/ 和 views/ 使用 Service 层 - 保持 API 兼容性，页面无需大幅修改 其他： - 新增 main.py 作为新入口 - 新增 DESIGN.md 架构设计文档 - 更新 requirements.txt 增加 pydantic-settings
115 lines
3.6 KiB
Python
115 lines
3.6 KiB
Python
"""代理相关路由"""
|
|
from typing import Optional
|
|
from fastapi import APIRouter, Depends, Query
|
|
from services.proxy_service import ProxyService
|
|
from models.schemas import ProxyListRequest, BatchDeleteRequest
|
|
from api.deps import get_proxy_service
|
|
|
|
router = APIRouter(prefix="/api/proxies", tags=["proxies"])
|
|
|
|
|
|
def success_response(message: str, data=None):
|
|
return {"code": 200, "message": message, "data": data}
|
|
|
|
|
|
def error_response(message: str, code: int = 500):
|
|
return {"code": code, "message": message, "data": None}
|
|
|
|
|
|
@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,
|
|
)
|
|
return success_response(
|
|
"获取代理列表成功",
|
|
{
|
|
"list": [
|
|
{
|
|
"ip": p.ip,
|
|
"port": p.port,
|
|
"protocol": p.protocol,
|
|
"score": p.score,
|
|
"last_check": p.last_check.isoformat() if p.last_check else None,
|
|
}
|
|
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:
|
|
return error_response("没有找到可用的代理", 404)
|
|
return success_response(
|
|
"获取随机代理成功",
|
|
{
|
|
"ip": proxy.ip,
|
|
"port": proxy.port,
|
|
"protocol": proxy.protocol,
|
|
"score": proxy.score,
|
|
"last_check": proxy.last_check.isoformat() if proxy.last_check else None,
|
|
},
|
|
)
|
|
|
|
|
|
@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"):
|
|
return error_response("不支持的导出格式", 400)
|
|
|
|
from fastapi.responses import StreamingResponse
|
|
|
|
media_types = {"csv": "text/csv", "txt": "text/plain", "json": "application/json"}
|
|
|
|
async def generate():
|
|
async for chunk in service.export_proxies(fmt, protocol, limit):
|
|
yield chunk
|
|
|
|
return StreamingResponse(
|
|
generate(),
|
|
media_type=media_types[fmt],
|
|
headers={"Content-Disposition": f"attachment; filename=proxies.{fmt}"},
|
|
)
|
|
|
|
|
|
@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})
|