插件配置持久化:
- plugin_settings 表新增 config_json 字段,支持存储每个插件的自定义配置
- BaseCrawlerPlugin 新增 default_config 属性和 update_config 方法
- PluginSettingsRepository 新增 get_config / set_config 方法
- PluginService 新增 get_plugin_config 和 update_plugin_config
- api/routes/plugins.py 新增 GET /{id}/config 和 POST /{id}/config 接口
- 前端 Plugins.vue 增加配置编辑对话框,支持动态渲染数字/布尔/字符串类型配置
- ip3366 插件示例化:增加 max_pages 配置项,验证配置生效后会动态更新爬取 URL
任务队列持久化:
- 新建 validation_tasks 表:id, ip, port, protocol, status, result, response_time_ms, created_at, updated_at
- 新建 ValidationTaskRepository,提供 insert_batch / acquire_pending / complete_task / reset_processing 等方法
- ValidationQueue 重构:
- submit() 时把任务写入数据库并唤醒 Worker
- Worker 通过 acquire_pending 原子取任务并验证
- 验证完成后更新任务状态并入库有效代理
- 启动时自动恢复之前中断的 processing 任务为 pending
- 支持 drain() 等待所有 pending 完成
- 调度器验证流程同样自动持久化到任务表
其他适配:
- 更新 api/deps.py 和 api/lifespan.py,移除对已删除 settings_service 的残留引用
- 更新前端 pluginService.js 和 api/index.js 增加配置相关 API
56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
"""插件基类 - 所有爬虫插件必须继承此基类"""
|
|
from abc import ABC, abstractmethod
|
|
from dataclasses import dataclass
|
|
from typing import List, Dict, Any
|
|
|
|
|
|
@dataclass
|
|
class ProxyRaw:
|
|
"""爬虫产出的原始代理数据"""
|
|
ip: str
|
|
port: int
|
|
protocol: str = "http"
|
|
|
|
def __post_init__(self):
|
|
self.protocol = self.protocol.lower().strip()
|
|
if self.protocol not in ("http", "https", "socks4", "socks5"):
|
|
self.protocol = "http"
|
|
|
|
|
|
class BaseCrawlerPlugin(ABC):
|
|
"""爬虫插件基类
|
|
|
|
添加新爬虫只需:
|
|
1. 继承 BaseCrawlerPlugin
|
|
2. 实现 crawl() 方法返回 List[ProxyRaw]
|
|
3. 用 @registry.register 装饰或在 __init__ 中显式注册
|
|
"""
|
|
|
|
name: str = ""
|
|
display_name: str = ""
|
|
description: str = ""
|
|
enabled: bool = True
|
|
default_config: Dict[str, Any] = {}
|
|
|
|
def __init__(self):
|
|
self._config: Dict[str, Any] = dict(self.default_config or {})
|
|
|
|
@property
|
|
def config(self) -> Dict[str, Any]:
|
|
return self._config
|
|
|
|
def update_config(self, updates: Dict[str, Any]) -> None:
|
|
"""更新插件配置,只覆盖存在的键"""
|
|
for key, value in updates.items():
|
|
if key in self._config:
|
|
self._config[key] = value
|
|
|
|
@abstractmethod
|
|
async def crawl(self) -> List[ProxyRaw]:
|
|
"""爬取代理的核心方法。只负责爬取,不要在这里验证。"""
|
|
raise NotImplementedError
|
|
|
|
async def health_check(self) -> bool:
|
|
"""可选:检查插件健康状态"""
|
|
return True
|