1422 lines
56 KiB
Python
1422 lines
56 KiB
Python
# ==================== 主入口文件 ====================
|
||
# ChatGPT Team 批量注册自动化 - 主程序
|
||
#
|
||
# 流程:
|
||
# 1. 检查未完成账号 (自动恢复)
|
||
# 2. 批量创建邮箱 (4个)
|
||
# 3. 一次性邀请到 Team
|
||
# 4. 逐个注册 OpenAI 账号 (或并发处理)
|
||
# 5. 逐个 Codex 授权
|
||
# 6. 逐个添加到 CRS/S2A
|
||
# 7. 切换下一个 Team
|
||
|
||
import time
|
||
import random
|
||
import signal
|
||
import sys
|
||
import atexit
|
||
import threading
|
||
from concurrent.futures import ThreadPoolExecutor, as_completed
|
||
|
||
from config import (
|
||
TEAMS, ACCOUNTS_PER_TEAM, DEFAULT_PASSWORD, AUTH_PROVIDER,
|
||
add_domain_to_blacklist, get_domain_from_email, is_email_blacklisted,
|
||
save_team_json, get_next_proxy,
|
||
CONCURRENT_ENABLED, CONCURRENT_WORKERS
|
||
)
|
||
from email_service import batch_create_emails, unified_create_email
|
||
from team_service import batch_invite_to_team, print_team_summary, check_available_seats, invite_single_to_team, preload_all_account_ids
|
||
from crs_service import crs_add_account, crs_sync_team_owners, crs_verify_token
|
||
from cpa_service import cpa_verify_connection
|
||
from s2a_service import s2a_verify_connection
|
||
from browser_automation import register_and_authorize, login_and_authorize_with_otp, authorize_only, login_and_authorize_team_owner, ShutdownRequested, register_only
|
||
from utils import (
|
||
save_to_csv,
|
||
load_team_tracker,
|
||
save_team_tracker,
|
||
add_account_with_password,
|
||
update_account_status,
|
||
remove_account_from_tracker,
|
||
get_incomplete_accounts,
|
||
get_all_incomplete_accounts,
|
||
print_summary,
|
||
Timer,
|
||
add_team_owners_to_tracker
|
||
)
|
||
from logger import log
|
||
|
||
# 进度更新 (Telegram Bot 使用，导入失败时忽略)
|
||
try:
|
||
from bot_notifier import progress_start, progress_update, progress_account_done, progress_finish, notify_team_completed_sync
|
||
except ImportError:
|
||
# 如果没有 bot_notifier，使用空函数
|
||
def progress_start(team_name, total): pass
|
||
def progress_update(account=None, phase=None, step=None, role=None): pass
|
||
def progress_account_done(email, success): pass
|
||
def progress_finish(): pass
|
||
def notify_team_completed_sync(team_name, results): pass
|
||
|
||
|
||
# ==================== 全局状态 ====================
|
||
_tracker = None
|
||
_current_results = []
|
||
_shutdown_requested = False
|
||
_tracker_lock = threading.Lock() # 用于并发时保护 tracker 操作
|
||
_auth_callback_lock = threading.Lock() # 授权回调锁 - 确保同一时间只有一个线程进行授权回调
|
||
|
||
|
||
def _save_state():
|
||
"""保存当前状态 (用于退出时保存)"""
|
||
global _tracker
|
||
if _tracker:
|
||
log.info("保存状态...", icon="save")
|
||
save_team_tracker(_tracker)
|
||
log.success("状态已保存到 team_tracker.json")
|
||
|
||
|
||
def _signal_handler(signum, frame):
|
||
"""处理 Ctrl+C 信号"""
|
||
global _shutdown_requested
|
||
if _shutdown_requested:
|
||
log.warning("强制退出...")
|
||
sys.exit(1)
|
||
|
||
_shutdown_requested = True
|
||
log.warning("收到中断信号，正在安全退出...")
|
||
_save_state()
|
||
|
||
if _current_results:
|
||
log.info("当前进度:")
|
||
print_summary(_current_results)
|
||
|
||
log.info("提示: 下次运行将自动从未完成的账号继续")
|
||
sys.exit(0)
|
||
|
||
|
||
# 注册信号处理器 (仅在主线程中)
|
||
import threading
|
||
if threading.current_thread() is threading.main_thread():
|
||
signal.signal(signal.SIGINT, _signal_handler)
|
||
signal.signal(signal.SIGTERM, _signal_handler)
|
||
atexit.register(_save_state)
|
||
|
||
|
||
def process_single_team(team: dict, team_index: int = 0, teams_total: int = 0) -> tuple[list, list]:
|
||
"""处理单个 Team 的完整流程
|
||
|
||
Args:
|
||
team: Team 配置
|
||
team_index: 当前 Team 序号 (从 1 开始)
|
||
teams_total: Team 总数
|
||
|
||
Returns:
|
||
tuple: (处理结果列表, 待处理的 Owner 列表)
|
||
"""
|
||
global _tracker, _current_results, _shutdown_requested
|
||
|
||
results = []
|
||
team_name = team["name"]
|
||
|
||
# 只在 _tracker 为空时加载，避免覆盖已有的修改
|
||
if _tracker is None:
|
||
_tracker = load_team_tracker()
|
||
|
||
# 分离 Owner 和普通成员
|
||
all_accounts = _tracker.get("teams", {}).get(team_name, [])
|
||
owner_accounts = [acc for acc in all_accounts if acc.get("role") == "owner" and acc.get("status") != "completed"]
|
||
member_accounts = [acc for acc in all_accounts if acc.get("role") != "owner"]
|
||
|
||
# 统计完成数量 (只统计普通成员)
|
||
completed_count = sum(1 for acc in member_accounts if acc.get("status") == "completed")
|
||
member_count = len(member_accounts)
|
||
|
||
# 如果普通成员已完成目标数量，且没有未完成的 Owner，跳过
|
||
owner_incomplete = len(owner_accounts)
|
||
if member_count >= ACCOUNTS_PER_TEAM and completed_count == member_count and owner_incomplete == 0:
|
||
print_team_summary(team)
|
||
log.success(f"{team_name} 已完成 {completed_count}/{ACCOUNTS_PER_TEAM} 个成员账号，跳过")
|
||
return results, []
|
||
|
||
# 有未完成的才打印详细信息
|
||
log.header(f"开始处理 {team_name}")
|
||
|
||
# 打印 Team 当前状态
|
||
print_team_summary(team)
|
||
|
||
if completed_count > 0:
|
||
log.success(f"已完成 {completed_count} 个成员账号")
|
||
|
||
# ========== 检查可用席位 (用于邀请新成员) ==========
|
||
available_seats = check_available_seats(team)
|
||
log.info(f"Team 可用席位: {available_seats}")
|
||
|
||
# ========== 检查未完成的普通成员账号 ==========
|
||
incomplete_members = [acc for acc in member_accounts if acc.get("status") != "completed"]
|
||
|
||
invited_accounts = []
|
||
|
||
if incomplete_members:
|
||
# 有未完成的普通成员账号，优先处理
|
||
log.warning(f"发现 {len(incomplete_members)} 个未完成成员账号:")
|
||
for acc in incomplete_members:
|
||
log.step(f"{acc['email']} (状态: {acc.get('status', 'unknown')})")
|
||
|
||
invited_accounts = [{
|
||
"email": acc["email"],
|
||
"password": acc.get("password", DEFAULT_PASSWORD),
|
||
"status": acc.get("status", ""),
|
||
"role": acc.get("role", "member")
|
||
} for acc in incomplete_members]
|
||
log.info("继续处理未完成成员账号...", icon="start")
|
||
elif member_count >= ACCOUNTS_PER_TEAM:
|
||
# 普通成员已达到目标数量
|
||
log.success(f"已有 {member_count} 个成员账号，无需邀请新成员")
|
||
elif available_seats > 0:
|
||
# 需要邀请新成员
|
||
need_count = min(ACCOUNTS_PER_TEAM - member_count, available_seats)
|
||
|
||
if need_count > 0:
|
||
log.info(f"已有 {member_count} 个成员账号，可用席位 {available_seats}，将创建 {need_count} 个")
|
||
|
||
# ========== 阶段 1: 批量创建邮箱 ==========
|
||
log.section(f"阶段 1: 批量创建 {need_count} 个邮箱")
|
||
|
||
with Timer("邮箱创建"):
|
||
accounts = batch_create_emails(need_count)
|
||
|
||
if len(accounts) > 0:
|
||
# ========== 阶段 2: 批量邀请到 Team ==========
|
||
log.section(f"阶段 2: 批量邀请 {len(accounts)} 个邮箱到 {team_name}")
|
||
|
||
emails = [acc["email"] for acc in accounts]
|
||
|
||
with Timer("批量邀请"):
|
||
invite_result = batch_invite_to_team(emails, team)
|
||
|
||
# 更新追踪记录 (带密码) - 立即保存
|
||
for acc in accounts:
|
||
if acc["email"] in invite_result.get("success", []):
|
||
add_account_with_password(_tracker, team_name, acc["email"], acc["password"], "invited")
|
||
save_team_tracker(_tracker)
|
||
log.success("邀请记录已保存")
|
||
|
||
# 筛选成功邀请的账号
|
||
invited_accounts = [{
|
||
"email": acc["email"],
|
||
"password": acc["password"],
|
||
"status": "invited",
|
||
"role": "member"
|
||
} for acc in accounts if acc["email"] in invite_result.get("success", [])]
|
||
else:
|
||
log.warning(f"Team {team_name} 没有可用席位，无法邀请新成员")
|
||
|
||
# ========== 合并成员和 Owner 一起处理 ==========
|
||
all_to_process = invited_accounts.copy()
|
||
|
||
# 添加未完成的 Owner 到处理列表
|
||
from config import INCLUDE_TEAM_OWNERS
|
||
include_owner = INCLUDE_TEAM_OWNERS and len(owner_accounts) > 0
|
||
|
||
if include_owner:
|
||
for owner in owner_accounts:
|
||
all_to_process.append({
|
||
"email": owner["email"],
|
||
"password": owner.get("password", DEFAULT_PASSWORD),
|
||
"status": owner.get("status", ""),
|
||
"role": "owner"
|
||
})
|
||
log.info(f"包含 {len(owner_accounts)} 个 Owner 账号一起处理")
|
||
|
||
# ========== 阶段 3: 处理所有账号 (注册 + Codex 授权 + 入库) ==========
|
||
if all_to_process:
|
||
# 根据配置选择处理模式
|
||
if CONCURRENT_ENABLED and len(all_to_process) > 1:
|
||
log.section(f"阶段 3: 并发处理 {len(all_to_process)} 个账号 (并发数: {min(CONCURRENT_WORKERS, len(all_to_process))})")
|
||
all_results = process_accounts_concurrent(
|
||
all_to_process, team_name,
|
||
team_index=team_index, teams_total=teams_total,
|
||
include_owner=include_owner
|
||
)
|
||
else:
|
||
log.section(f"阶段 3: 逐个注册 OpenAI + Codex 授权 + 入库")
|
||
all_results = process_accounts(
|
||
all_to_process, team_name,
|
||
team_index=team_index, teams_total=teams_total,
|
||
include_owner=include_owner
|
||
)
|
||
results.extend(all_results)
|
||
|
||
# ========== Team 处理完成 ==========
|
||
success_count = sum(1 for r in results if r["status"] == "success")
|
||
if results:
|
||
log.success(f"{team_name} 处理完成: {success_count}/{len(results)} 成功")
|
||
# 发送 Team 完成报告到 Telegram
|
||
notify_team_completed_sync(team_name, results)
|
||
|
||
# 返回空列表，因为 Owner 已经在这里处理了
|
||
return results, []
|
||
|
||
|
||
def _get_team_by_name(team_name: str) -> dict:
|
||
"""根据名称获取 Team 配置"""
|
||
for team in TEAMS:
|
||
if team["name"] == team_name:
|
||
return team
|
||
return {}
|
||
|
||
|
||
def process_accounts(accounts: list, team_name: str, team_index: int = 0,
|
||
teams_total: int = 0, include_owner: bool = False) -> list:
|
||
"""处理账号列表 (注册/授权/CRS)
|
||
|
||
Args:
|
||
accounts: 账号列表 [{"email", "password", "status", "role"}]
|
||
team_name: Team 名称
|
||
team_index: 当前 Team 序号 (从 1 开始)
|
||
teams_total: Team 总数
|
||
include_owner: 是否包含 Owner
|
||
|
||
Returns:
|
||
list: 处理结果
|
||
"""
|
||
global _tracker, _current_results, _shutdown_requested
|
||
|
||
results = []
|
||
|
||
# 启动进度跟踪 (Telegram Bot)
|
||
progress_start(team_name, len(accounts), team_index, teams_total, include_owner)
|
||
|
||
for i, account in enumerate(accounts):
|
||
if _shutdown_requested:
|
||
log.warning("检测到中断请求，停止处理...")
|
||
break
|
||
|
||
email = account["email"]
|
||
password = account["password"]
|
||
role = account.get("role", "member")
|
||
|
||
# 检查邮箱域名是否在黑名单中
|
||
if is_email_blacklisted(email):
|
||
domain = get_domain_from_email(email)
|
||
log.warning(f"邮箱域名 {domain} 在黑名单中，跳过: {email}")
|
||
|
||
# 从 tracker 中移除
|
||
remove_account_from_tracker(_tracker, team_name, email)
|
||
save_team_tracker(_tracker)
|
||
|
||
# 尝试创建新邮箱替代
|
||
if role != "owner":
|
||
log.info("尝试创建新邮箱替代...")
|
||
new_email, new_password = unified_create_email()
|
||
if new_email and not is_email_blacklisted(new_email):
|
||
# 邀请新邮箱
|
||
if invite_single_to_team(new_email, _get_team_by_name(team_name)):
|
||
add_account_with_password(_tracker, team_name, new_email, new_password, "invited")
|
||
save_team_tracker(_tracker)
|
||
# 更新当前账号信息继续处理
|
||
email = new_email
|
||
password = new_password
|
||
account["email"] = email
|
||
account["password"] = password
|
||
log.success(f"已创建新邮箱替代: {email}")
|
||
else:
|
||
log.error("新邮箱邀请失败")
|
||
continue
|
||
else:
|
||
log.error("无法创建有效的新邮箱")
|
||
continue
|
||
else:
|
||
continue
|
||
|
||
log.separator("#", 50)
|
||
log.info(f"处理账号 {i + 1}/{len(accounts)}: {email}", icon="account")
|
||
log.separator("#", 50)
|
||
|
||
# 更新进度: 当前账号
|
||
progress_update(account=email, phase="准备", step="开始处理...", role=role)
|
||
|
||
result = {
|
||
"team": team_name,
|
||
"email": email,
|
||
"password": password,
|
||
"status": "failed",
|
||
"crs_id": ""
|
||
}
|
||
|
||
# 检查账号状态，决定处理流程
|
||
account_status = account.get("status", "")
|
||
account_role = account.get("role", "member")
|
||
|
||
# 已完成的账号跳过
|
||
if account_status == "completed":
|
||
log.info(f"账号已完成，跳过: {email}")
|
||
continue
|
||
|
||
# Team Owner 需要 OTP 登录 (仅限旧格式，状态为 team_owner)
|
||
is_team_owner_otp = account_status == "team_owner"
|
||
|
||
# 已授权但未入库的状态 (直接尝试入库，不重新授权)
|
||
# - authorized: 授权成功但入库失败
|
||
# - partial: 部分完成
|
||
# 注意: S2A 模式下 partial 状态需要重新授权，因为入库是授权过程的一部分
|
||
if AUTH_PROVIDER == "s2a":
|
||
need_crs_only = account_status == "authorized" # S2A: partial 需要重新授权
|
||
else:
|
||
need_crs_only = account_status in ["authorized", "partial"]
|
||
|
||
# 已注册但未授权的状态 (使用密码登录授权)
|
||
# - registered: 已注册，需要授权
|
||
# - auth_failed: 授权失败，重试
|
||
# - partial (S2A模式): 入库验证失败，需要重新授权
|
||
# - 新格式 Owner (role=owner 且状态不是 team_owner/completed) 也走密码登录
|
||
need_auth_only = (
|
||
account_status in ["registered", "auth_failed"]
|
||
or (AUTH_PROVIDER == "s2a" and account_status == "partial") # S2A: partial 需要重新授权
|
||
or (account_role == "owner" and account_status not in ["team_owner", "completed", "authorized", "partial"])
|
||
)
|
||
|
||
# 标记为处理中
|
||
update_account_status(_tracker, team_name, email, "processing")
|
||
save_team_tracker(_tracker)
|
||
|
||
try:
|
||
with Timer(f"账号 {email}"):
|
||
if is_team_owner_otp:
|
||
# 旧格式 Team Owner: 使用 OTP 登录授权
|
||
log.info("Team Owner 账号 (旧格式)，使用一次性验证码登录...", icon="auth")
|
||
progress_update(phase="授权", step="OTP 登录...")
|
||
auth_success, codex_data = login_and_authorize_with_otp(email)
|
||
register_success = auth_success
|
||
elif need_crs_only:
|
||
# 已授权但未入库: 跳过授权，直接尝试入库
|
||
log.info(f"已授权账号 (状态: {account_status})，跳过授权，直接入库...", icon="auth")
|
||
progress_update(phase="入库", step="添加到 CRS...")
|
||
register_success = True
|
||
codex_data = None # CPA/S2A 模式不需要 codex_data
|
||
# CRS 模式下，由于没有 codex_data，无法入库，需要重新授权
|
||
if AUTH_PROVIDER not in ("cpa", "s2a"):
|
||
log.warning("CRS 模式下已授权账号缺少 codex_data，需要重新授权")
|
||
auth_success, codex_data = authorize_only(email, password)
|
||
register_success = auth_success
|
||
elif need_auth_only:
|
||
# 已注册账号 (包括新格式 Owner): 使用密码登录授权
|
||
log.info(f"已注册账号 (状态: {account_status}, 角色: {account_role})，使用密码登录授权...", icon="auth")
|
||
progress_update(phase="授权", step="密码登录授权...")
|
||
auth_success, codex_data = authorize_only(email, password)
|
||
register_success = True
|
||
else:
|
||
# 新账号: 注册 + Codex 授权
|
||
progress_update(phase="注册", step="注册 OpenAI...")
|
||
register_success, codex_data = register_and_authorize(email, password)
|
||
|
||
# 检查是否是域名黑名单错误
|
||
if register_success == "domain_blacklisted":
|
||
domain = get_domain_from_email(email)
|
||
log.error(f"域名 {domain} 不被支持，加入黑名单")
|
||
add_domain_to_blacklist(domain)
|
||
|
||
# 从 tracker 中移除
|
||
remove_account_from_tracker(_tracker, team_name, email)
|
||
save_team_tracker(_tracker)
|
||
|
||
# 尝试创建新邮箱替代
|
||
log.info("尝试创建新邮箱替代...")
|
||
new_email, new_password = unified_create_email()
|
||
if new_email and not is_email_blacklisted(new_email):
|
||
# 邀请新邮箱
|
||
if invite_single_to_team(new_email, _get_team_by_name(team_name)):
|
||
add_account_with_password(_tracker, team_name, new_email, new_password, "invited")
|
||
save_team_tracker(_tracker)
|
||
log.success(f"已创建新邮箱: {new_email}，将在下次运行时处理")
|
||
else:
|
||
log.error("新邮箱邀请失败")
|
||
else:
|
||
log.error("无法创建有效的新邮箱")
|
||
|
||
continue # 跳过当前账号，继续下一个
|
||
|
||
if register_success and register_success != "domain_blacklisted":
|
||
update_account_status(_tracker, team_name, email, "registered")
|
||
save_team_tracker(_tracker)
|
||
|
||
# CPA 模式: codex_data 为 None，授权成功后直接标记完成
|
||
# CRS 模式: 需要 codex_data，手动添加到 CRS
|
||
if AUTH_PROVIDER == "s2a":
|
||
# S2A 模式: 授权成功后验证账号是否入库
|
||
from s2a_service import s2a_verify_account_in_pool
|
||
|
||
update_account_status(_tracker, team_name, email, "authorized")
|
||
save_team_tracker(_tracker)
|
||
|
||
# 验证账号是否成功入库
|
||
log.step("正在验证 S2A 账号入库状态...")
|
||
progress_update(phase="验证", step="检查入库状态...")
|
||
verified, account_data = s2a_verify_account_in_pool(email)
|
||
|
||
if verified:
|
||
account_id = account_data.get("id", "")
|
||
account_name = account_data.get("name", "")
|
||
result["status"] = "success"
|
||
result["crs_id"] = f"S2A-{account_id}"
|
||
|
||
update_account_status(_tracker, team_name, email, "completed")
|
||
save_team_tracker(_tracker)
|
||
|
||
log.success(f"✅ S2A 账号入库成功 (ID: {account_id}, 名称: {account_name})")
|
||
else:
|
||
log.warning("⚠️ S2A 授权成功但入库验证失败")
|
||
result["status"] = "partial"
|
||
update_account_status(_tracker, team_name, email, "partial")
|
||
save_team_tracker(_tracker)
|
||
|
||
elif AUTH_PROVIDER == "cpa":
|
||
# CPA 模式: 授权成功即完成 (后台自动处理账号)
|
||
update_account_status(_tracker, team_name, email, "authorized")
|
||
save_team_tracker(_tracker)
|
||
|
||
result["status"] = "success"
|
||
result["crs_id"] = "CPA-AUTO"
|
||
|
||
update_account_status(_tracker, team_name, email, "completed")
|
||
save_team_tracker(_tracker)
|
||
|
||
log.success(f"✅ CPA 账号处理完成: {email}")
|
||
else:
|
||
# CRS 模式: 原有逻辑
|
||
if codex_data:
|
||
update_account_status(_tracker, team_name, email, "authorized")
|
||
save_team_tracker(_tracker)
|
||
|
||
# 添加到 CRS
|
||
log.step("添加到 CRS...")
|
||
crs_result = crs_add_account(email, codex_data)
|
||
|
||
if crs_result:
|
||
crs_id = crs_result.get("id", "")
|
||
result["status"] = "success"
|
||
result["crs_id"] = crs_id
|
||
|
||
update_account_status(_tracker, team_name, email, "completed")
|
||
save_team_tracker(_tracker)
|
||
|
||
log.success(f"账号处理完成: {email}")
|
||
else:
|
||
log.warning("CRS 入库失败，但注册和授权成功")
|
||
result["status"] = "partial"
|
||
update_account_status(_tracker, team_name, email, "partial")
|
||
save_team_tracker(_tracker)
|
||
else:
|
||
log.warning("Codex 授权失败")
|
||
result["status"] = "auth_failed"
|
||
update_account_status(_tracker, team_name, email, "auth_failed")
|
||
save_team_tracker(_tracker)
|
||
elif register_success != "domain_blacklisted":
|
||
if is_team_owner_otp:
|
||
log.error(f"OTP 登录授权失败: {email}")
|
||
else:
|
||
log.error(f"注册/授权失败: {email}")
|
||
update_account_status(_tracker, team_name, email, "register_failed")
|
||
save_team_tracker(_tracker)
|
||
|
||
except ShutdownRequested:
|
||
# 用户请求停止，保存当前状态并退出
|
||
log.warning(f"用户请求停止，当前账号: {email}")
|
||
# 不改变账号状态，保持中断前的状态，下次继续处理
|
||
save_team_tracker(_tracker)
|
||
break
|
||
|
||
# 保存到 CSV
|
||
save_to_csv(
|
||
email=email,
|
||
password=password,
|
||
team_name=team_name,
|
||
status=result["status"],
|
||
crs_id=result.get("crs_id", "")
|
||
)
|
||
|
||
results.append(result)
|
||
|
||
# 更新进度: 账号完成
|
||
is_success = result["status"] in ("success", "completed")
|
||
progress_account_done(email, is_success)
|
||
|
||
# 账号之间的间隔
|
||
if i < len(accounts) - 1 and not _shutdown_requested:
|
||
wait_time = 1
|
||
log.info(f"等待 {wait_time}s 后处理下一个账号...", icon="wait")
|
||
time.sleep(wait_time)
|
||
|
||
return results
|
||
|
||
|
||
# ==================== 并发处理函数 ====================
|
||
|
||
def _process_single_account_worker(
|
||
account: dict,
|
||
team_name: str,
|
||
worker_id: int
|
||
) -> dict:
|
||
"""单个账号处理工作函数 (用于并发执行)
|
||
|
||
Args:
|
||
account: 账号信息 {"email", "password", "status", "role"}
|
||
team_name: Team 名称
|
||
worker_id: 工作线程 ID
|
||
|
||
Returns:
|
||
dict: 处理结果
|
||
"""
|
||
global _tracker, _shutdown_requested
|
||
|
||
email = account["email"]
|
||
password = account["password"]
|
||
role = account.get("role", "member")
|
||
account_status = account.get("status", "")
|
||
account_role = account.get("role", "member")
|
||
|
||
result = {
|
||
"team": team_name,
|
||
"email": email,
|
||
"password": password,
|
||
"status": "failed",
|
||
"crs_id": "",
|
||
"worker_id": worker_id
|
||
}
|
||
|
||
# 检查中断请求
|
||
if _shutdown_requested:
|
||
log.warning(f"[Worker-{worker_id}] 检测到中断请求，跳过: {email}")
|
||
return result
|
||
|
||
# 检查邮箱域名黑名单
|
||
if is_email_blacklisted(email):
|
||
domain = get_domain_from_email(email)
|
||
log.warning(f"[Worker-{worker_id}] 邮箱域名 {domain} 在黑名单中，跳过: {email}")
|
||
with _tracker_lock:
|
||
remove_account_from_tracker(_tracker, team_name, email)
|
||
save_team_tracker(_tracker)
|
||
return result
|
||
|
||
# 已完成的账号跳过
|
||
if account_status == "completed":
|
||
log.info(f"[Worker-{worker_id}] 账号已完成，跳过: {email}")
|
||
result["status"] = "completed"
|
||
return result
|
||
|
||
log.info(f"[Worker-{worker_id}] 开始处理: {email}", icon="account")
|
||
|
||
# 判断处理流程
|
||
is_team_owner_otp = account_status == "team_owner"
|
||
|
||
if AUTH_PROVIDER == "s2a":
|
||
need_crs_only = account_status == "authorized"
|
||
else:
|
||
need_crs_only = account_status in ["authorized", "partial"]
|
||
|
||
need_auth_only = (
|
||
account_status in ["registered", "auth_failed"]
|
||
or (AUTH_PROVIDER == "s2a" and account_status == "partial")
|
||
or (account_role == "owner" and account_status not in ["team_owner", "completed", "authorized", "partial"])
|
||
)
|
||
|
||
# 标记为处理中
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "processing")
|
||
save_team_tracker(_tracker)
|
||
|
||
try:
|
||
with Timer(f"[Worker-{worker_id}] 账号 {email}"):
|
||
if is_team_owner_otp:
|
||
log.info(f"[Worker-{worker_id}] Team Owner (OTP 登录)...", icon="auth")
|
||
auth_success, codex_data = login_and_authorize_with_otp(email)
|
||
register_success = auth_success
|
||
elif need_crs_only:
|
||
log.info(f"[Worker-{worker_id}] 已授权账号，跳过授权...", icon="auth")
|
||
register_success = True
|
||
codex_data = None
|
||
if AUTH_PROVIDER not in ("cpa", "s2a"):
|
||
auth_success, codex_data = authorize_only(email, password)
|
||
register_success = auth_success
|
||
elif need_auth_only:
|
||
log.info(f"[Worker-{worker_id}] 已注册账号，密码登录授权...", icon="auth")
|
||
auth_success, codex_data = authorize_only(email, password)
|
||
register_success = True
|
||
else:
|
||
log.info(f"[Worker-{worker_id}] 新账号，注册 + 授权...", icon="auth")
|
||
register_success, codex_data = register_and_authorize(email, password)
|
||
|
||
if register_success == "domain_blacklisted":
|
||
domain = get_domain_from_email(email)
|
||
log.error(f"[Worker-{worker_id}] 域名 {domain} 不被支持")
|
||
add_domain_to_blacklist(domain)
|
||
with _tracker_lock:
|
||
remove_account_from_tracker(_tracker, team_name, email)
|
||
save_team_tracker(_tracker)
|
||
return result
|
||
|
||
if register_success and register_success != "domain_blacklisted":
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "registered")
|
||
save_team_tracker(_tracker)
|
||
|
||
if AUTH_PROVIDER == "s2a":
|
||
from s2a_service import s2a_verify_account_in_pool
|
||
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "authorized")
|
||
save_team_tracker(_tracker)
|
||
|
||
log.step(f"[Worker-{worker_id}] 验证 S2A 入库状态...")
|
||
verified, account_data = s2a_verify_account_in_pool(email)
|
||
|
||
if verified:
|
||
account_id = account_data.get("id", "")
|
||
result["status"] = "success"
|
||
result["crs_id"] = f"S2A-{account_id}"
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "completed")
|
||
save_team_tracker(_tracker)
|
||
log.success(f"[Worker-{worker_id}] ✅ S2A 入库成功: {email} (ID: {account_id})")
|
||
else:
|
||
log.warning(f"[Worker-{worker_id}] ⚠️ S2A 入库验证失败: {email}")
|
||
result["status"] = "partial"
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "partial")
|
||
save_team_tracker(_tracker)
|
||
|
||
elif AUTH_PROVIDER == "cpa":
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "authorized")
|
||
save_team_tracker(_tracker)
|
||
result["status"] = "success"
|
||
result["crs_id"] = "CPA-AUTO"
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "completed")
|
||
save_team_tracker(_tracker)
|
||
log.success(f"[Worker-{worker_id}] ✅ CPA 处理完成: {email}")
|
||
|
||
else:
|
||
# CRS 模式
|
||
if codex_data:
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "authorized")
|
||
save_team_tracker(_tracker)
|
||
crs_result = crs_add_account(email, codex_data)
|
||
if crs_result:
|
||
crs_id = crs_result.get("id", "")
|
||
result["status"] = "success"
|
||
result["crs_id"] = crs_id
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "completed")
|
||
save_team_tracker(_tracker)
|
||
log.success(f"[Worker-{worker_id}] ✅ CRS 入库成功: {email}")
|
||
else:
|
||
result["status"] = "partial"
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "partial")
|
||
save_team_tracker(_tracker)
|
||
else:
|
||
result["status"] = "auth_failed"
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "auth_failed")
|
||
save_team_tracker(_tracker)
|
||
else:
|
||
log.error(f"[Worker-{worker_id}] 注册/授权失败: {email}")
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "register_failed")
|
||
save_team_tracker(_tracker)
|
||
|
||
except ShutdownRequested:
|
||
log.warning(f"[Worker-{worker_id}] 用户请求停止: {email}")
|
||
with _tracker_lock:
|
||
save_team_tracker(_tracker)
|
||
|
||
except Exception as e:
|
||
log.error(f"[Worker-{worker_id}] 处理异常: {email} - {e}")
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "error")
|
||
save_team_tracker(_tracker)
|
||
|
||
# 保存到 CSV
|
||
save_to_csv(
|
||
email=email,
|
||
password=password,
|
||
team_name=team_name,
|
||
status=result["status"],
|
||
crs_id=result.get("crs_id", "")
|
||
)
|
||
|
||
return result
|
||
|
||
|
||
def process_accounts_concurrent(
|
||
accounts: list,
|
||
team_name: str,
|
||
team_index: int = 0,
|
||
teams_total: int = 0,
|
||
include_owner: bool = False,
|
||
max_workers: int = None
|
||
) -> list:
|
||
"""并发处理账号列表 (两阶段模式: 并行注册 + 串行授权)
|
||
|
||
阶段 1: 并行注册所有新账号
|
||
阶段 2: 串行授权所有已注册的账号
|
||
|
||
Args:
|
||
accounts: 账号列表 [{"email", "password", "status", "role"}]
|
||
team_name: Team 名称
|
||
team_index: 当前 Team 序号
|
||
teams_total: Team 总数
|
||
include_owner: 是否包含 Owner
|
||
max_workers: 最大并发数 (默认使用配置值)
|
||
|
||
Returns:
|
||
list: 处理结果
|
||
"""
|
||
global _tracker, _shutdown_requested
|
||
|
||
if max_workers is None:
|
||
max_workers = CONCURRENT_WORKERS
|
||
|
||
stagger_delay = 3.0 # 线程错开启动间隔 (秒)
|
||
|
||
# 过滤已完成的账号
|
||
pending_accounts = [acc for acc in accounts if acc.get("status") != "completed"]
|
||
|
||
if not pending_accounts:
|
||
log.info("所有账号已完成，无需处理")
|
||
return []
|
||
|
||
total = len(pending_accounts)
|
||
actual_workers = min(max_workers, total)
|
||
|
||
# 分类账号: 需要注册的 vs 已注册待授权的
|
||
need_register = []
|
||
need_auth_only = []
|
||
|
||
for acc in pending_accounts:
|
||
status = acc.get("status", "")
|
||
role = acc.get("role", "member")
|
||
|
||
# 已注册但未授权的状态
|
||
if status in ["registered", "auth_failed"] or \
|
||
(AUTH_PROVIDER == "s2a" and status == "partial") or \
|
||
(role == "owner" and status not in ["team_owner", "completed", "authorized", ""]):
|
||
need_auth_only.append(acc)
|
||
elif status == "team_owner":
|
||
# Team Owner 使用 OTP，需要特殊处理
|
||
need_auth_only.append(acc)
|
||
elif status in ["invited", "processing", ""]:
|
||
# 新账号，需要注册
|
||
need_register.append(acc)
|
||
else:
|
||
# 其他状态，尝试注册
|
||
need_register.append(acc)
|
||
|
||
log.section(f"两阶段并发处理 {total} 个账号")
|
||
log.info(f"需要注册: {len(need_register)} 个, 需要授权: {len(need_auth_only)} 个")
|
||
|
||
# 启动进度跟踪
|
||
progress_start(team_name, total, team_index, teams_total, include_owner)
|
||
|
||
results = []
|
||
|
||
# ==================== 阶段 1: 并行注册 ====================
|
||
if need_register:
|
||
log.section(f"阶段 1: 并行注册 {len(need_register)} 个账号 (并发数: {actual_workers})")
|
||
|
||
registered_accounts = []
|
||
|
||
with ThreadPoolExecutor(max_workers=actual_workers) as executor:
|
||
future_to_account = {}
|
||
|
||
for i, account in enumerate(need_register):
|
||
if _shutdown_requested:
|
||
break
|
||
|
||
worker_id = i % actual_workers + 1
|
||
log.info(f"[Worker-{worker_id}] 启动注册: {account['email']}", icon="start")
|
||
|
||
future = executor.submit(
|
||
_register_single_account_worker,
|
||
account,
|
||
team_name,
|
||
worker_id
|
||
)
|
||
future_to_account[future] = account
|
||
|
||
# 错开启动
|
||
if i < len(need_register) - 1:
|
||
time.sleep(stagger_delay)
|
||
|
||
# 收集注册结果
|
||
for future in as_completed(future_to_account):
|
||
if _shutdown_requested:
|
||
log.warning("检测到中断请求，取消剩余任务...")
|
||
executor.shutdown(wait=False, cancel_futures=True)
|
||
break
|
||
|
||
account = future_to_account[future]
|
||
try:
|
||
reg_result = future.result()
|
||
if reg_result == "success":
|
||
log.success(f"✅ 注册成功: {account['email']}")
|
||
registered_accounts.append(account)
|
||
elif reg_result == "domain_blacklisted":
|
||
log.error(f"❌ 域名黑名单: {account['email']}")
|
||
domain = get_domain_from_email(account['email'])
|
||
add_domain_to_blacklist(domain)
|
||
with _tracker_lock:
|
||
remove_account_from_tracker(_tracker, team_name, account['email'])
|
||
save_team_tracker(_tracker)
|
||
else:
|
||
log.error(f"❌ 注册失败: {account['email']}")
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, account['email'], "register_failed")
|
||
save_team_tracker(_tracker)
|
||
except Exception as e:
|
||
log.error(f"注册异常: {account.get('email', 'unknown')} - {e}")
|
||
|
||
log.success(f"阶段 1 完成: {len(registered_accounts)}/{len(need_register)} 注册成功")
|
||
|
||
# 将成功注册的账号加入授权列表
|
||
need_auth_only.extend(registered_accounts)
|
||
|
||
# ==================== 阶段 2: 串行授权 ====================
|
||
if need_auth_only and not _shutdown_requested:
|
||
log.section(f"阶段 2: 串行授权 {len(need_auth_only)} 个账号")
|
||
|
||
for i, account in enumerate(need_auth_only):
|
||
if _shutdown_requested:
|
||
log.warning("检测到中断请求，停止授权...")
|
||
break
|
||
|
||
email = account["email"]
|
||
password = account["password"]
|
||
role = account.get("role", "member")
|
||
status = account.get("status", "")
|
||
|
||
log.info(f"[{i+1}/{len(need_auth_only)}] 授权: {email}", icon="auth")
|
||
|
||
result = {
|
||
"team": team_name,
|
||
"email": email,
|
||
"password": password,
|
||
"status": "failed",
|
||
"crs_id": ""
|
||
}
|
||
|
||
try:
|
||
with Timer(f"授权 {email}"):
|
||
# 判断授权方式
|
||
if status == "team_owner":
|
||
# Team Owner 使用 OTP
|
||
log.info("Team Owner，使用 OTP 登录...", icon="auth")
|
||
auth_success, codex_data = login_and_authorize_with_otp(email)
|
||
else:
|
||
# 普通账号使用密码登录授权
|
||
auth_success, codex_data = authorize_only(email, password)
|
||
|
||
if auth_success:
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "authorized")
|
||
save_team_tracker(_tracker)
|
||
|
||
# 验证入库
|
||
if AUTH_PROVIDER == "s2a":
|
||
from s2a_service import s2a_verify_account_in_pool
|
||
verified, account_data = s2a_verify_account_in_pool(email)
|
||
|
||
if verified:
|
||
account_id = account_data.get("id", "")
|
||
result["status"] = "success"
|
||
result["crs_id"] = f"S2A-{account_id}"
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "completed")
|
||
save_team_tracker(_tracker)
|
||
log.success(f"✅ S2A 入库成功: {email} (ID: {account_id})")
|
||
else:
|
||
log.warning(f"⚠️ S2A 入库验证失败: {email}")
|
||
result["status"] = "partial"
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "partial")
|
||
save_team_tracker(_tracker)
|
||
|
||
elif AUTH_PROVIDER == "cpa":
|
||
result["status"] = "success"
|
||
result["crs_id"] = "CPA-AUTO"
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "completed")
|
||
save_team_tracker(_tracker)
|
||
log.success(f"✅ CPA 处理完成: {email}")
|
||
|
||
else:
|
||
# CRS 模式
|
||
if codex_data:
|
||
crs_result = crs_add_account(email, codex_data)
|
||
if crs_result:
|
||
crs_id = crs_result.get("id", "")
|
||
result["status"] = "success"
|
||
result["crs_id"] = crs_id
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "completed")
|
||
save_team_tracker(_tracker)
|
||
log.success(f"✅ CRS 入库成功: {email}")
|
||
else:
|
||
result["status"] = "partial"
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "partial")
|
||
save_team_tracker(_tracker)
|
||
else:
|
||
log.error(f"❌ 授权失败: {email}")
|
||
result["status"] = "auth_failed"
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "auth_failed")
|
||
save_team_tracker(_tracker)
|
||
|
||
except ShutdownRequested:
|
||
log.warning(f"用户请求停止: {email}")
|
||
with _tracker_lock:
|
||
save_team_tracker(_tracker)
|
||
break
|
||
except Exception as e:
|
||
log.error(f"授权异常: {email} - {e}")
|
||
result["status"] = "error"
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "error")
|
||
save_team_tracker(_tracker)
|
||
|
||
# 保存到 CSV
|
||
save_to_csv(
|
||
email=email,
|
||
password=password,
|
||
team_name=team_name,
|
||
status=result["status"],
|
||
crs_id=result.get("crs_id", "")
|
||
)
|
||
|
||
results.append(result)
|
||
|
||
# 更新进度
|
||
is_success = result["status"] in ("success", "completed")
|
||
progress_account_done(email, is_success)
|
||
|
||
# 授权间隔
|
||
if i < len(need_auth_only) - 1 and not _shutdown_requested:
|
||
time.sleep(1)
|
||
|
||
# 统计结果
|
||
success_count = sum(1 for r in results if r["status"] in ("success", "completed"))
|
||
log.success(f"两阶段处理完成: {success_count}/{len(results)} 成功")
|
||
|
||
return results
|
||
|
||
|
||
def _register_single_account_worker(account: dict, team_name: str, worker_id: int) -> str:
|
||
"""单个账号注册工作函数 (用于阶段 1 并行注册)
|
||
|
||
Args:
|
||
account: 账号信息
|
||
team_name: Team 名称
|
||
worker_id: 工作线程 ID
|
||
|
||
Returns:
|
||
str: "success", "domain_blacklisted", or "failed"
|
||
"""
|
||
global _tracker, _shutdown_requested
|
||
|
||
email = account["email"]
|
||
password = account["password"]
|
||
|
||
# 检查中断请求
|
||
if _shutdown_requested:
|
||
return "failed"
|
||
|
||
# 检查邮箱域名黑名单
|
||
if is_email_blacklisted(email):
|
||
return "domain_blacklisted"
|
||
|
||
log.info(f"[Worker-{worker_id}] 开始注册: {email}", icon="account")
|
||
|
||
# 标记为处理中
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "processing")
|
||
save_team_tracker(_tracker)
|
||
|
||
try:
|
||
with Timer(f"[Worker-{worker_id}] 注册 {email}"):
|
||
result = register_only(email, password)
|
||
|
||
if result == "success":
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "registered")
|
||
save_team_tracker(_tracker)
|
||
return "success"
|
||
elif result == "domain_blacklisted":
|
||
return "domain_blacklisted"
|
||
else:
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "register_failed")
|
||
save_team_tracker(_tracker)
|
||
return "failed"
|
||
|
||
except ShutdownRequested:
|
||
return "failed"
|
||
except Exception as e:
|
||
log.error(f"[Worker-{worker_id}] 注册异常: {email} - {e}")
|
||
with _tracker_lock:
|
||
update_account_status(_tracker, team_name, email, "error")
|
||
save_team_tracker(_tracker)
|
||
return "failed"
|
||
|
||
|
||
def _print_system_config():
|
||
"""打印当前系统配置"""
|
||
from config import (
|
||
EMAIL_PROVIDER, AUTH_PROVIDER, ACCOUNTS_PER_TEAM,
|
||
INCLUDE_TEAM_OWNERS, BROWSER_RANDOM_FINGERPRINT,
|
||
S2A_API_BASE, CPA_API_BASE, CRS_API_BASE,
|
||
PROXY_ENABLED, PROXIES,
|
||
CONCURRENT_ENABLED, CONCURRENT_WORKERS
|
||
)
|
||
|
||
log.section("系统配置")
|
||
log.info(f"邮箱服务: {EMAIL_PROVIDER}", icon="email")
|
||
|
||
# 授权服务
|
||
if AUTH_PROVIDER == "s2a":
|
||
auth_url = S2A_API_BASE or "未配置"
|
||
elif AUTH_PROVIDER == "cpa":
|
||
auth_url = CPA_API_BASE or "未配置"
|
||
else:
|
||
auth_url = CRS_API_BASE or "未配置"
|
||
log.info(f"授权服务: {AUTH_PROVIDER.upper()} ({auth_url})", icon="auth")
|
||
|
||
log.info(f"每 Team 账号: {ACCOUNTS_PER_TEAM}", icon="account")
|
||
log.info(f"Owner 入库: {'✓ 开启' if INCLUDE_TEAM_OWNERS else '✗ 关闭'}", icon="config")
|
||
log.info(f"随机指纹: {'✓ 开启' if BROWSER_RANDOM_FINGERPRINT else '✗ 关闭'}", icon="config")
|
||
|
||
# 并发配置
|
||
if CONCURRENT_ENABLED:
|
||
log.info(f"并发处理: ✓ 开启 ({CONCURRENT_WORKERS} 并发, 授权串行)", icon="config")
|
||
else:
|
||
log.info("并发处理: ✗ 关闭 (串行模式)", icon="config")
|
||
|
||
if PROXY_ENABLED and PROXIES:
|
||
log.info(f"代理: 已启用 ({len(PROXIES)} 个)", icon="proxy")
|
||
else:
|
||
log.info("代理: 未启用", icon="proxy")
|
||
|
||
log.separator()
|
||
|
||
|
||
def run_all_teams():
|
||
"""主函数: 遍历所有 Team"""
|
||
global _tracker, _current_results, _shutdown_requested
|
||
|
||
log.header("ChatGPT Team 批量注册自动化")
|
||
|
||
# 打印系统配置
|
||
_print_system_config()
|
||
|
||
log.info(f"共 {len(TEAMS)} 个 Team 待处理", icon="team")
|
||
log.info(f"统一密码: {DEFAULT_PASSWORD}", icon="code")
|
||
log.info("按 Ctrl+C 可安全退出并保存进度")
|
||
log.separator()
|
||
|
||
# 先显示整体状态
|
||
_tracker = load_team_tracker()
|
||
all_incomplete = get_all_incomplete_accounts(_tracker)
|
||
|
||
if all_incomplete:
|
||
total_incomplete = sum(len(accs) for accs in all_incomplete.values())
|
||
log.warning(f"发现 {total_incomplete} 个未完成账号，将优先处理")
|
||
|
||
_current_results = []
|
||
teams_total = len(TEAMS)
|
||
|
||
with Timer("全部流程"):
|
||
# ========== 处理所有 Team (成员 + Owner 一起) ==========
|
||
for i, team in enumerate(TEAMS):
|
||
if _shutdown_requested:
|
||
log.warning("检测到中断请求，停止处理...")
|
||
break
|
||
|
||
log.separator("★", 60)
|
||
team_email = team.get('account') or team.get('owner_email', '')
|
||
log.highlight(f"Team {i + 1}/{teams_total}: {team['name']} ({team_email})", icon="team")
|
||
log.separator("★", 60)
|
||
|
||
# 传递 Team 序号信息
|
||
results, _ = process_single_team(team, team_index=i + 1, teams_total=teams_total)
|
||
_current_results.extend(results)
|
||
|
||
# Team 之间的间隔
|
||
if i < teams_total - 1 and not _shutdown_requested:
|
||
wait_time = 3
|
||
log.countdown(wait_time, "下一个 Team")
|
||
|
||
# 打印总结
|
||
print_summary(_current_results)
|
||
|
||
return _current_results
|
||
|
||
|
||
def run_single_team(team_index: int = 0):
|
||
"""只运行单个 Team (用于测试)
|
||
|
||
Args:
|
||
team_index: Team 索引 (从 0 开始)
|
||
"""
|
||
global _current_results
|
||
|
||
if team_index >= len(TEAMS):
|
||
log.error(f"Team 索引超出范围 (0-{len(TEAMS) - 1})")
|
||
return
|
||
|
||
team = TEAMS[team_index]
|
||
log.info(f"单 Team 模式: {team['name']}", icon="start")
|
||
|
||
_current_results = []
|
||
# 单 Team 模式：序号为 1/1
|
||
results, _ = process_single_team(team, team_index=1, teams_total=1)
|
||
_current_results.extend(results)
|
||
|
||
print_summary(_current_results)
|
||
|
||
return _current_results
|
||
|
||
|
||
def test_email_only():
|
||
"""测试模式: 只创建邮箱和邀请，不注册"""
|
||
global _tracker
|
||
|
||
log.info("测试模式: 仅邮箱创建 + 邀请", icon="debug")
|
||
|
||
if len(TEAMS) == 0:
|
||
log.error("没有配置 Team")
|
||
return
|
||
|
||
team = TEAMS[0]
|
||
team_name = team["name"]
|
||
log.step(f"使用 Team: {team_name}")
|
||
|
||
# 创建邮箱
|
||
accounts = batch_create_emails(2) # 测试只创建 2 个
|
||
|
||
if accounts:
|
||
# 批量邀请
|
||
emails = [acc["email"] for acc in accounts]
|
||
result = batch_invite_to_team(emails, team)
|
||
|
||
# 保存到 tracker
|
||
_tracker = load_team_tracker()
|
||
for acc in accounts:
|
||
if acc["email"] in result.get("success", []):
|
||
add_account_with_password(_tracker, team_name, acc["email"], acc["password"], "invited")
|
||
save_team_tracker(_tracker)
|
||
|
||
log.success(f"测试完成: {len(result.get('success', []))} 个邀请成功")
|
||
log.info("记录已保存到 team_tracker.json", icon="save")
|
||
|
||
|
||
def show_status():
|
||
"""显示当前状态"""
|
||
log.header("当前状态")
|
||
|
||
tracker = load_team_tracker()
|
||
|
||
if not tracker.get("teams"):
|
||
log.info("没有任何记录")
|
||
return
|
||
|
||
total_accounts = 0
|
||
total_completed = 0
|
||
total_incomplete = 0
|
||
|
||
for team_name, accounts in tracker["teams"].items():
|
||
log.info(f"{team_name}:", icon="team")
|
||
status_count = {}
|
||
for acc in accounts:
|
||
total_accounts += 1
|
||
status = acc.get("status", "unknown")
|
||
status_count[status] = status_count.get(status, 0) + 1
|
||
|
||
if status == "completed":
|
||
total_completed += 1
|
||
log.success(f"{acc['email']} ({status})")
|
||
elif status in ["invited", "registered", "authorized", "processing"]:
|
||
total_incomplete += 1
|
||
log.warning(f"{acc['email']} ({status})")
|
||
else:
|
||
total_incomplete += 1
|
||
log.error(f"{acc['email']} ({status})")
|
||
|
||
log.info(f"统计: {status_count}")
|
||
|
||
log.separator("-", 40)
|
||
log.info(f"总计: {total_accounts} 个账号")
|
||
log.success(f"完成: {total_completed}")
|
||
log.warning(f"未完成: {total_incomplete}")
|
||
log.info(f"最后更新: {tracker.get('last_updated', 'N/A')}", icon="time")
|
||
|
||
|
||
def process_team_with_login(team: dict, team_index: int, total: int):
|
||
"""处理单个 Team（包括获取 token、授权和后续流程）
|
||
|
||
用于格式3的 Team，登录时同时完成授权
|
||
"""
|
||
global _tracker
|
||
|
||
log.separator("★", 60)
|
||
log.highlight(f"Team {team_index + 1}/{total}: {team['name']} ({team['owner_email']})", icon="team")
|
||
log.separator("★", 60)
|
||
|
||
# 1. 登录并授权
|
||
log.info("登录并授权 Owner...", icon="auth")
|
||
proxy = get_next_proxy()
|
||
result = login_and_authorize_team_owner(
|
||
team["owner_email"],
|
||
team["owner_password"],
|
||
proxy
|
||
)
|
||
|
||
owner_result = None # Owner 的处理结果
|
||
|
||
if result.get("token"):
|
||
team["auth_token"] = result["token"]
|
||
if result.get("account_id"):
|
||
team["account_id"] = result["account_id"]
|
||
if result.get("authorized"):
|
||
team["authorized"] = True
|
||
|
||
# 立即保存
|
||
save_team_json()
|
||
|
||
if not result.get("token"):
|
||
log.error(f"登录失败，跳过此 Team")
|
||
return []
|
||
|
||
team["needs_login"] = False
|
||
|
||
if result.get("authorized"):
|
||
log.success(f"Owner 登录并授权成功")
|
||
# 记录 Owner 授权成功的结果
|
||
owner_result = {
|
||
"email": team["owner_email"],
|
||
"team": team["name"],
|
||
"status": "success",
|
||
"role": "owner"
|
||
}
|
||
else:
|
||
log.warning(f"Owner 登录成功但授权失败，后续可重试")
|
||
|
||
# 2. 添加 Owner 到 tracker (状态根据 authorized 决定)
|
||
_tracker = load_team_tracker()
|
||
add_team_owners_to_tracker(_tracker, DEFAULT_PASSWORD)
|
||
save_team_tracker(_tracker)
|
||
|
||
# 3. 处理该 Team 的成员
|
||
results, pending_owners = process_single_team(team)
|
||
|
||
# 4. 如果 Owner 授权失败，在这里重试
|
||
if pending_owners:
|
||
for owner in pending_owners:
|
||
# 只处理未授权的 Owner
|
||
if owner.get("status") != "authorized":
|
||
owner_data = {
|
||
"email": owner["email"],
|
||
"password": owner.get("password", DEFAULT_PASSWORD),
|
||
"status": owner.get("status", "registered"),
|
||
"role": "owner"
|
||
}
|
||
owner_results = process_accounts([owner_data], team["name"])
|
||
results.extend(owner_results)
|
||
|
||
# 添加 Owner 结果到返回列表
|
||
if owner_result:
|
||
results.insert(0, owner_result)
|
||
|
||
return results
|
||
|
||
|
||
if __name__ == "__main__":
|
||
# ========== 启动前置检查 ==========
|
||
# 1. 根据配置选择验证对应的授权服务
|
||
if AUTH_PROVIDER == "cpa":
|
||
log.info("授权服务: CPA", icon="auth")
|
||
is_valid, message = cpa_verify_connection()
|
||
if is_valid:
|
||
log.success(f"CPA {message}")
|
||
else:
|
||
log.error(f"CPA 验证失败: {message}")
|
||
sys.exit(1)
|
||
elif AUTH_PROVIDER == "s2a":
|
||
log.info("授权服务: S2A (Sub2API)", icon="auth")
|
||
is_valid, message = s2a_verify_connection()
|
||
if is_valid:
|
||
log.success(f"S2A {message}")
|
||
else:
|
||
log.error(f"S2A 验证失败: {message}")
|
||
sys.exit(1)
|
||
else:
|
||
log.info("授权服务: CRS", icon="auth")
|
||
is_valid, message = crs_verify_token()
|
||
if is_valid:
|
||
log.success(f"CRS {message}")
|
||
else:
|
||
log.error(f"CRS Token 验证失败: {message}")
|
||
sys.exit(1)
|
||
|
||
# 2. 分离需要登录和不需要登录的 Team
|
||
needs_login_teams = [t for t in TEAMS if t.get("format") == "new" and t.get("needs_login")]
|
||
ready_teams = [t for t in TEAMS if not (t.get("format") == "new" and t.get("needs_login"))]
|
||
|
||
# 3. 只对已有 token 的 Team 预加载 account_id 和添加到 tracker
|
||
if ready_teams:
|
||
success_count, fail_count = preload_all_account_ids()
|
||
_tracker = load_team_tracker()
|
||
add_team_owners_to_tracker(_tracker, DEFAULT_PASSWORD)
|
||
save_team_tracker(_tracker)
|
||
|
||
if len(sys.argv) > 1:
|
||
arg = sys.argv[1]
|
||
|
||
if arg == "test":
|
||
test_email_only()
|
||
elif arg == "single":
|
||
team_idx = int(sys.argv[2]) if len(sys.argv) > 2 else 0
|
||
run_single_team(team_idx)
|
||
elif arg == "status":
|
||
show_status()
|
||
else:
|
||
log.error(f"未知参数: {arg}")
|
||
log.info("用法: python run.py [test|single N|status]")
|
||
else:
|
||
# 默认运行
|
||
_current_results = []
|
||
|
||
# 先处理需要登录的 Team（获取 token 后立即处理）
|
||
if needs_login_teams:
|
||
log.separator("=", 60)
|
||
log.info(f"处理缺少 Token 的 Team ({len(needs_login_teams)} 个)")
|
||
log.separator("=", 60)
|
||
|
||
for i, team in enumerate(needs_login_teams):
|
||
if _shutdown_requested:
|
||
break
|
||
results = process_team_with_login(team, i, len(needs_login_teams))
|
||
_current_results.extend(results)
|
||
|
||
if i < len(needs_login_teams) - 1 and not _shutdown_requested:
|
||
wait_time = random.randint(3, 8)
|
||
log.info(f"等待 {wait_time}s...", icon="wait")
|
||
time.sleep(wait_time)
|
||
|
||
# 再处理已有 token 的 Team
|
||
if ready_teams and not _shutdown_requested:
|
||
run_all_teams()
|
||
|
||
if _current_results:
|
||
print_summary(_current_results)
|