578 lines
18 KiB
Python
578 lines
18 KiB
Python
"""
|
||
配置管理模块
|
||
|
||
使用 Pydantic Settings 从环境变量和 .env 文件加载配置
|
||
|
||
支持的配置项:
|
||
- 代理设置（代理池、轮换策略）
|
||
- 并发控制（最大并发数、重试次数）
|
||
- 日志级别
|
||
- 邮箱配置（可选）
|
||
- Sentinel 配置（可选）
|
||
"""
|
||
|
||
from pydantic import BaseModel, Field, field_validator
|
||
from pydantic_settings import BaseSettings, SettingsConfigDict
|
||
from typing import List, Optional, Literal, Dict, Any
|
||
from dotenv import load_dotenv
|
||
from pathlib import Path
|
||
import random
|
||
|
||
# 加载 .env 文件
|
||
load_dotenv()
|
||
|
||
|
||
class ProxyConfig(BaseModel):
|
||
"""代理配置"""
|
||
|
||
enabled: bool = Field(default=False, description="是否启用代理")
|
||
pool: List[str] = Field(default_factory=list, description="代理池列表")
|
||
rotation: Literal["random", "round_robin"] = Field(
|
||
default="random",
|
||
description="代理轮换策略"
|
||
)
|
||
|
||
_current_index: int = 0 # 用于 round_robin 策略
|
||
|
||
def get_next_proxy(self) -> Optional[str]:
|
||
"""
|
||
获取下一个代理地址
|
||
|
||
根据配置的轮换策略返回代理
|
||
- random: 随机选择
|
||
- round_robin: 轮流使用
|
||
|
||
返回:
|
||
代理地址，如果代理池为空则返回 None
|
||
"""
|
||
if not self.enabled or not self.pool:
|
||
return None
|
||
|
||
if self.rotation == "random":
|
||
return random.choice(self.pool)
|
||
else: # round_robin
|
||
proxy = self.pool[self._current_index % len(self.pool)]
|
||
self._current_index += 1
|
||
return proxy
|
||
|
||
def validate_proxy_format(self, proxy: str) -> bool:
|
||
"""
|
||
验证代理格式是否正确
|
||
|
||
支持格式:
|
||
- http://ip:port
|
||
- http://user:pass@ip:port
|
||
- https://ip:port
|
||
- socks5://ip:port
|
||
|
||
参数:
|
||
proxy: 代理地址
|
||
|
||
返回:
|
||
True 如果格式正确，否则 False
|
||
"""
|
||
import re
|
||
pattern = r'^(http|https|socks5)://([^:]+:[^@]+@)?[\w.-]+:\d+$'
|
||
return bool(re.match(pattern, proxy))
|
||
|
||
|
||
class MailConfig(BaseModel):
|
||
"""邮箱配置"""
|
||
|
||
enabled: bool = Field(default=False, description="是否启用邮箱功能")
|
||
type: Literal["imap", "tempmail", "api", "manual", "cloudmail"] = Field(
|
||
default="manual",
|
||
description="邮箱类型"
|
||
)
|
||
|
||
# IMAP 配置
|
||
imap_host: Optional[str] = Field(default=None, description="IMAP 服务器地址")
|
||
imap_port: int = Field(default=993, description="IMAP 端口")
|
||
imap_username: Optional[str] = Field(default=None, description="邮箱用户名")
|
||
imap_password: Optional[str] = Field(default=None, description="邮箱密码")
|
||
|
||
# 临时邮箱 API 配置
|
||
api_key: Optional[str] = Field(default=None, description="临时邮箱 API Key")
|
||
api_endpoint: Optional[str] = Field(default=None, description="临时邮箱 API 端点")
|
||
|
||
# CloudMail 配置
|
||
cloudmail_api_url: Optional[str] = Field(
|
||
default=None,
|
||
description="Cloud Mail API 基础 URL"
|
||
)
|
||
cloudmail_token: Optional[str] = Field(
|
||
default=None,
|
||
description="Cloud Mail 身份令牌（预先生成）"
|
||
)
|
||
cloudmail_domain: Optional[str] = Field(
|
||
default=None,
|
||
description="Cloud Mail 邮箱域名（例如 mygoband.com）"
|
||
)
|
||
|
||
def to_dict(self) -> Dict[str, Any]:
|
||
"""转换为字典格式（供 MailHandler 使用）"""
|
||
config = {"type": self.type}
|
||
|
||
if self.type == "imap":
|
||
config.update({
|
||
"host": self.imap_host,
|
||
"port": self.imap_port,
|
||
"username": self.imap_username,
|
||
"password": self.imap_password,
|
||
})
|
||
elif self.type in ["tempmail", "api"]:
|
||
config.update({
|
||
"api_key": self.api_key,
|
||
"api_endpoint": self.api_endpoint,
|
||
})
|
||
elif self.type == "cloudmail":
|
||
config.update({
|
||
"api_base_url": self.cloudmail_api_url,
|
||
"token": self.cloudmail_token,
|
||
"domain": self.cloudmail_domain,
|
||
})
|
||
|
||
return config
|
||
|
||
|
||
class SentinelConfig(BaseModel):
|
||
"""Sentinel 配置"""
|
||
|
||
enabled: bool = Field(default=False, description="是否启用 Sentinel 解决器")
|
||
solver_type: Literal["external_script", "api", "module"] = Field(
|
||
default="external_script",
|
||
description="解决器类型"
|
||
)
|
||
|
||
# 外部脚本配置
|
||
script_path: Optional[str] = Field(
|
||
default=None,
|
||
description="外部脚本路径（Node.js 或其他）"
|
||
)
|
||
|
||
# API 配置
|
||
api_endpoint: Optional[str] = Field(
|
||
default=None,
|
||
description="Sentinel solver API 端点"
|
||
)
|
||
api_key: Optional[str] = Field(default=None, description="API Key")
|
||
|
||
# Python 模块配置
|
||
module_name: Optional[str] = Field(
|
||
default=None,
|
||
description="Python 模块名称（例如 'my_sentinel_solver'）"
|
||
)
|
||
|
||
|
||
class AppConfig(BaseSettings):
|
||
"""
|
||
应用配置（从环境变量加载）
|
||
|
||
环境变量优先级:
|
||
1. 系统环境变量
|
||
2. .env 文件
|
||
3. 默认值
|
||
"""
|
||
|
||
model_config = SettingsConfigDict(
|
||
env_file=".env",
|
||
env_file_encoding="utf-8",
|
||
case_sensitive=False,
|
||
extra="ignore" # 忽略未定义的环境变量
|
||
)
|
||
|
||
# ========== 代理配置 ==========
|
||
proxy_enabled: bool = Field(
|
||
default=False,
|
||
description="是否启用代理"
|
||
)
|
||
proxy_pool: str = Field(
|
||
default="",
|
||
description="代理池（逗号分隔），例如: http://ip1:port,socks5://ip2:port"
|
||
)
|
||
proxy_rotation: Literal["random", "round_robin"] = Field(
|
||
default="random",
|
||
description="代理轮换策略"
|
||
)
|
||
|
||
# ========== 并发配置 ==========
|
||
max_workers: int = Field(
|
||
default=1,
|
||
ge=1,
|
||
le=50,
|
||
description="最大并发任务数"
|
||
)
|
||
retry_limit: int = Field(
|
||
default=3,
|
||
ge=0,
|
||
le=10,
|
||
description="失败重试次数"
|
||
)
|
||
retry_delay: int = Field(
|
||
default=5,
|
||
ge=0,
|
||
description="重试延迟（秒）"
|
||
)
|
||
|
||
# ========== 日志配置 ==========
|
||
log_level: Literal["DEBUG", "INFO", "WARNING", "ERROR"] = Field(
|
||
default="INFO",
|
||
description="日志级别"
|
||
)
|
||
log_to_file: bool = Field(
|
||
default=True,
|
||
description="是否记录日志到文件"
|
||
)
|
||
|
||
# ========== 邮箱配置 ==========
|
||
mail_enabled: bool = Field(default=False, description="是否启用邮箱功能")
|
||
mail_type: Literal["imap", "tempmail", "api", "manual", "cloudmail"] = Field(
|
||
default="manual",
|
||
description="邮箱类型"
|
||
)
|
||
mail_imap_host: Optional[str] = Field(default=None, description="IMAP 服务器")
|
||
mail_imap_port: int = Field(default=993, description="IMAP 端口")
|
||
mail_imap_username: Optional[str] = Field(default=None, description="邮箱用户名")
|
||
mail_imap_password: Optional[str] = Field(default=None, description="邮箱密码")
|
||
mail_api_key: Optional[str] = Field(default=None, description="临时邮箱 API Key")
|
||
mail_api_endpoint: Optional[str] = Field(default=None, description="API 端点")
|
||
|
||
# CloudMail 配置
|
||
mail_cloudmail_api_url: Optional[str] = Field(
|
||
default=None,
|
||
description="Cloud Mail API URL"
|
||
)
|
||
mail_cloudmail_token: Optional[str] = Field(
|
||
default=None,
|
||
description="Cloud Mail 身份令牌"
|
||
)
|
||
mail_cloudmail_domain: Optional[str] = Field(
|
||
default=None,
|
||
description="Cloud Mail 邮箱域名（例如 mygoband.com）"
|
||
)
|
||
|
||
# ========== Sentinel 配置 ==========
|
||
sentinel_enabled: bool = Field(default=False, description="是否启用 Sentinel")
|
||
sentinel_solver_type: Literal["external_script", "api", "module"] = Field(
|
||
default="external_script",
|
||
description="Sentinel 解决器类型"
|
||
)
|
||
sentinel_script_path: Optional[str] = Field(default=None, description="脚本路径")
|
||
sentinel_api_endpoint: Optional[str] = Field(default=None, description="API 端点")
|
||
sentinel_api_key: Optional[str] = Field(default=None, description="API Key")
|
||
sentinel_module_name: Optional[str] = Field(default=None, description="模块名称")
|
||
|
||
# Sentinel 内部配置
|
||
sentinel_debug: bool = Field(default=False, description="Sentinel 调试模式")
|
||
sentinel_sdk_path: str = Field(
|
||
default="sdk/sdk.js",
|
||
description="Sentinel SDK JS 文件路径（相对于项目根目录）"
|
||
)
|
||
|
||
# ========== TLS 指纹配置 ==========
|
||
tls_impersonate: Literal["chrome110", "chrome120", "chrome124"] = Field(
|
||
default="chrome124",
|
||
description="模拟的浏览器版本"
|
||
)
|
||
|
||
# ========== 其他配置 ==========
|
||
accounts_output_file: str = Field(
|
||
default="accounts.txt",
|
||
description="成功账号保存文件路径"
|
||
)
|
||
request_timeout: int = Field(
|
||
default=30,
|
||
ge=5,
|
||
le=300,
|
||
description="HTTP 请求超时时间（秒）"
|
||
)
|
||
|
||
# ========== Telegram Bot 配置 ==========
|
||
telegram_bot_token: Optional[str] = Field(
|
||
default=None,
|
||
description="Telegram Bot Token (从 @BotFather 获取)"
|
||
)
|
||
telegram_allowed_users: str = Field(
|
||
default="",
|
||
description="允许使用 Bot 的用户 ID（逗号分隔）"
|
||
)
|
||
telegram_admin_users: str = Field(
|
||
default="",
|
||
description="管理员用户 ID（逗号分隔，拥有更高权限）"
|
||
)
|
||
telegram_bot_enabled: bool = Field(
|
||
default=True,
|
||
description="是否启用 Telegram Bot"
|
||
)
|
||
|
||
@field_validator("proxy_pool")
|
||
@classmethod
|
||
def validate_proxy_pool(cls, v: str) -> str:
|
||
"""验证代理池格式"""
|
||
if not v:
|
||
return v
|
||
|
||
proxies = [p.strip() for p in v.split(",") if p.strip()]
|
||
for proxy in proxies:
|
||
# 基本格式检查
|
||
if not any(proxy.startswith(prefix) for prefix in ["http://", "https://", "socks5://"]):
|
||
raise ValueError(
|
||
f"Invalid proxy format: {proxy}. "
|
||
"Must start with http://, https://, or socks5://"
|
||
)
|
||
|
||
return v
|
||
|
||
@property
|
||
def proxy(self) -> ProxyConfig:
|
||
"""获取代理配置对象"""
|
||
pool = [p.strip() for p in self.proxy_pool.split(",") if p.strip()]
|
||
return ProxyConfig(
|
||
enabled=self.proxy_enabled,
|
||
pool=pool,
|
||
rotation=self.proxy_rotation
|
||
)
|
||
|
||
@property
|
||
def mail(self) -> MailConfig:
|
||
"""获取邮箱配置对象"""
|
||
return MailConfig(
|
||
enabled=self.mail_enabled,
|
||
type=self.mail_type,
|
||
imap_host=self.mail_imap_host,
|
||
imap_port=self.mail_imap_port,
|
||
imap_username=self.mail_imap_username,
|
||
imap_password=self.mail_imap_password,
|
||
api_key=self.mail_api_key,
|
||
api_endpoint=self.mail_api_endpoint,
|
||
cloudmail_api_url=self.mail_cloudmail_api_url,
|
||
cloudmail_token=self.mail_cloudmail_token,
|
||
cloudmail_domain=self.mail_cloudmail_domain,
|
||
)
|
||
|
||
@property
|
||
def sentinel(self) -> SentinelConfig:
|
||
"""获取 Sentinel 配置对象"""
|
||
return SentinelConfig(
|
||
enabled=self.sentinel_enabled,
|
||
solver_type=self.sentinel_solver_type,
|
||
script_path=self.sentinel_script_path,
|
||
api_endpoint=self.sentinel_api_endpoint,
|
||
api_key=self.sentinel_api_key,
|
||
module_name=self.sentinel_module_name,
|
||
)
|
||
|
||
@property
|
||
def fingerprint_config(self) -> Dict[str, Any]:
|
||
"""获取指纹配置"""
|
||
return {
|
||
"user_agent": "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/143.0.0.0 Safari/537.36",
|
||
"screen_width": 1920,
|
||
"screen_height": 1080,
|
||
"languages": ["en-US", "en"],
|
||
"hardware_concurrency": 8,
|
||
}
|
||
|
||
def validate_config(self) -> List[str]:
|
||
"""
|
||
验证配置完整性，返回警告列表
|
||
|
||
返回:
|
||
警告信息列表，空列表表示无警告
|
||
"""
|
||
warnings = []
|
||
|
||
# 检查代理配置
|
||
if self.proxy_enabled and not self.proxy_pool:
|
||
warnings.append("Proxy enabled but proxy_pool is empty")
|
||
|
||
# 检查邮箱配置
|
||
if self.mail_enabled:
|
||
if self.mail_type == "imap":
|
||
if not all([self.mail_imap_host, self.mail_imap_username, self.mail_imap_password]):
|
||
warnings.append("IMAP mail enabled but credentials incomplete")
|
||
elif self.mail_type in ["tempmail", "api"]:
|
||
if not self.mail_api_key:
|
||
warnings.append(f"{self.mail_type} enabled but api_key not configured")
|
||
elif self.mail_type == "cloudmail":
|
||
if not all([self.mail_cloudmail_api_url, self.mail_cloudmail_token, self.mail_cloudmail_domain]):
|
||
warnings.append(
|
||
"CloudMail enabled but config incomplete. "
|
||
"Required: MAIL_CLOUDMAIL_API_URL, MAIL_CLOUDMAIL_TOKEN, MAIL_CLOUDMAIL_DOMAIN"
|
||
)
|
||
|
||
# 检查 Sentinel 配置
|
||
if self.sentinel_enabled:
|
||
if self.sentinel_solver_type == "external_script" and not self.sentinel_script_path:
|
||
warnings.append("Sentinel external_script enabled but script_path not configured")
|
||
elif self.sentinel_solver_type == "api" and not self.sentinel_api_endpoint:
|
||
warnings.append("Sentinel API enabled but api_endpoint not configured")
|
||
elif self.sentinel_solver_type == "module" and not self.sentinel_module_name:
|
||
warnings.append("Sentinel module enabled but module_name not configured")
|
||
|
||
# 检查并发设置
|
||
if self.max_workers > 10 and not self.proxy_enabled:
|
||
warnings.append(
|
||
f"High concurrency ({self.max_workers} workers) without proxy may trigger rate limits"
|
||
)
|
||
|
||
return warnings
|
||
|
||
def print_summary(self):
|
||
"""打印配置摘要"""
|
||
from utils.logger import logger
|
||
|
||
logger.info("=" * 60)
|
||
logger.info("Configuration Summary")
|
||
logger.info("=" * 60)
|
||
logger.info(f"Proxy: {'Enabled' if self.proxy_enabled else 'Disabled'}")
|
||
if self.proxy_enabled:
|
||
logger.info(f" - Pool size: {len(self.proxy.pool)}")
|
||
logger.info(f" - Rotation: {self.proxy_rotation}")
|
||
|
||
logger.info(f"Mail: {'Enabled' if self.mail_enabled else 'Disabled'}")
|
||
if self.mail_enabled:
|
||
logger.info(f" - Type: {self.mail_type}")
|
||
|
||
logger.info(f"Sentinel: {'Enabled' if self.sentinel_enabled else 'Disabled'}")
|
||
if self.sentinel_enabled:
|
||
logger.info(f" - Solver: {self.sentinel_solver_type}")
|
||
|
||
logger.info(f"Concurrency: {self.max_workers} workers")
|
||
logger.info(f"Retry limit: {self.retry_limit}")
|
||
logger.info(f"Log level: {self.log_level}")
|
||
logger.info(f"TLS impersonate: {self.tls_impersonate}")
|
||
logger.info("=" * 60)
|
||
|
||
# 打印警告
|
||
warnings = self.validate_config()
|
||
if warnings:
|
||
logger.warning("Configuration warnings:")
|
||
for warning in warnings:
|
||
logger.warning(f" ⚠️ {warning}")
|
||
|
||
|
||
def load_config() -> AppConfig:
|
||
"""
|
||
加载配置
|
||
|
||
返回:
|
||
AppConfig 实例
|
||
|
||
示例:
|
||
config = load_config()
|
||
print(config.proxy.enabled)
|
||
print(config.mail.type)
|
||
"""
|
||
config = AppConfig()
|
||
return config
|
||
|
||
|
||
def create_default_env_file(path: str = ".env.example"):
|
||
"""
|
||
创建默认的 .env 示例文件
|
||
|
||
参数:
|
||
path: 文件保存路径
|
||
"""
|
||
content = """# OpenAI 账号自动注册系统 - 配置文件示例
|
||
# 复制此文件为 .env 并根据实际情况修改
|
||
|
||
# ========== 代理配置 ==========
|
||
# 是否启用代理（true/false）
|
||
PROXY_ENABLED=false
|
||
|
||
# 代理池（逗号分隔，支持 http/https/socks5）
|
||
# 格式: protocol://[user:pass@]ip:port
|
||
# 示例: http://user:pass@1.2.3.4:8080,socks5://5.6.7.8:1080
|
||
PROXY_POOL=
|
||
|
||
# 代理轮换策略（random: 随机选择, round_robin: 轮流使用）
|
||
PROXY_ROTATION=random
|
||
|
||
# ========== 并发配置 ==========
|
||
# 最大并发任务数（建议 1-5，过高可能触发风控）
|
||
MAX_WORKERS=1
|
||
|
||
# 失败重试次数
|
||
RETRY_LIMIT=3
|
||
|
||
# 重试延迟（秒）
|
||
RETRY_DELAY=5
|
||
|
||
# ========== 日志配置 ==========
|
||
# 日志级别（DEBUG, INFO, WARNING, ERROR）
|
||
LOG_LEVEL=INFO
|
||
|
||
# 是否记录日志到文件
|
||
LOG_TO_FILE=true
|
||
|
||
# ========== 邮箱配置 ==========
|
||
# 是否启用邮箱功能（true/false）
|
||
MAIL_ENABLED=false
|
||
|
||
# 邮箱类型（imap, tempmail, api, cloudmail, manual）
|
||
MAIL_TYPE=manual
|
||
|
||
# IMAP 配置（如果使用 IMAP）
|
||
MAIL_IMAP_HOST=imap.gmail.com
|
||
MAIL_IMAP_PORT=993
|
||
MAIL_IMAP_USERNAME=your@email.com
|
||
MAIL_IMAP_PASSWORD=your_app_password
|
||
|
||
# 临时邮箱 API 配置（如果使用临时邮箱）
|
||
MAIL_API_KEY=
|
||
MAIL_API_ENDPOINT=
|
||
|
||
# CloudMail 配置（如果使用 CloudMail）
|
||
# 1. 先通过 Cloud Mail 管理界面或 API 生成 Token
|
||
# 2. 将 Token 填入 MAIL_CLOUDMAIL_TOKEN
|
||
# 3. 填写你的邮箱域名（不带 @）
|
||
# 4. Token 失效时需要手动更新
|
||
MAIL_CLOUDMAIL_API_URL=https://your-cloudmail-domain.com
|
||
MAIL_CLOUDMAIL_TOKEN=9f4e298e-7431-4c76-bc15-4931c3a73984
|
||
MAIL_CLOUDMAIL_DOMAIN=mygoband.com
|
||
|
||
# ========== Sentinel 配置 ==========
|
||
# 是否启用 Sentinel 解决器（true/false）
|
||
SENTINEL_ENABLED=false
|
||
|
||
# Sentinel 解决器类型（external_script, api, module）
|
||
SENTINEL_SOLVER_TYPE=external_script
|
||
|
||
# 外部脚本路径（如果使用 external_script）
|
||
SENTINEL_SCRIPT_PATH=./sentinel_solver.js
|
||
|
||
# API 配置（如果使用 api）
|
||
SENTINEL_API_ENDPOINT=http://localhost:8000/solve
|
||
SENTINEL_API_KEY=
|
||
|
||
# Python 模块名称（如果使用 module）
|
||
SENTINEL_MODULE_NAME=
|
||
|
||
# ========== TLS 指纹配置 ==========
|
||
# 模拟的浏览器版本（chrome110, chrome120, chrome124）
|
||
TLS_IMPERSONATE=chrome124
|
||
|
||
# ========== 其他配置 ==========
|
||
# 成功账号保存文件路径
|
||
ACCOUNTS_OUTPUT_FILE=accounts.txt
|
||
|
||
# HTTP 请求超时时间（秒）
|
||
REQUEST_TIMEOUT=30
|
||
"""
|
||
|
||
Path(path).write_text(content, encoding="utf-8")
|
||
print(f"✅ Default .env.example created at: {path}")
|
||
|
||
|
||
# 导出主要接口
|
||
__all__ = [
|
||
"AppConfig",
|
||
"ProxyConfig",
|
||
"MailConfig",
|
||
"SentinelConfig",
|
||
"load_config",
|
||
"create_default_env_file",
|
||
]
|