mygo-kyx/codexTool
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

4

Browse Source
This commit is contained in:
kyx236
2026-01-26 06:58:04 +08:00
parent 20cdf8060d
commit adb60cdfd6
3 changed files with 160 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
api_register.py
View File

@@ -20,6 +20,20 @@ except ImportError:
import requests
def _is_shutdown_requested():
"""检查是否收到停止请求"""
try:
import run
return run._shutdown_requested
except Exception:
return False
class ShutdownRequested(Exception):
"""用户请求停止异常"""
pass
def log_status(step, message):
"""日志输出"""
timestamp = time.strftime("%H:%M:%S")
@@ -330,12 +344,20 @@ def get_verification_code_api(target_email: str, mail_api_base: str, mail_api_to
Returns:
str: 验证码，失败返回空字符串
Raises:
ShutdownRequested: 用户请求停止时抛出
"""
log_status("API监听", "正在监听邮件...")
headers = {"Authorization": mail_api_token, "Content-Type": "application/json"}
start_time = time.time()
for i in range(max_retries):
# 检查停止请求
if _is_shutdown_requested():
log_status("停止", "[!] 检测到停止请求，中断邮件监听")
raise ShutdownRequested("用户请求停止")
elapsed = int(time.time() - start_time)
try:
url = f"{mail_api_base}/api/public/emailList"
@@ -389,6 +411,9 @@ def api_register_flow(
Returns:
ChatGPTAPIRegister: 成功返回 reg 对象，失败返回 None
Raises:
ShutdownRequested: 用户请求停止时抛出
"""
def log_cb(msg):
if progress_callback:
@@ -396,57 +421,72 @@ def api_register_flow(
else:
log_progress(msg)
def check_shutdown():
"""检查停止请求"""
if _is_shutdown_requested():
log_cb("[!] 检测到停止请求")
raise ShutdownRequested("用户请求停止")
reg = ChatGPTAPIRegister(proxy=proxy)
try:
check_shutdown()
log_status("API注册", "初始化会话...")
if not reg.init_session():
log_cb("[X] 初始化失败")
return None
log_cb("[OK] 会话初始化成功")
check_shutdown()
log_status("API注册", "获取授权 URL...")
if not reg.get_authorize_url(email):
log_cb("[X] 获取授权 URL 失败")
return None
log_cb("[OK] 授权 URL 获取成功")
check_shutdown()
log_status("API注册", "开始授权流程...")
if not reg.start_authorize():
log_cb("[X] 授权流程启动失败")
return None
log_cb("[OK] 授权流程已启动")
check_shutdown()
log_status("API注册", "注册账户...")
if not reg.register(email, password):
log_cb("[X] 注册失败")
return None
log_cb("[OK] 账户注册成功")
check_shutdown()
log_status("API注册", "发送验证邮件...")
if not reg.send_verification_email():
log_cb("[X] 发送验证邮件失败")
return None
log_cb("[OK] 验证邮件已发送")
check_shutdown()
# 获取验证码
otp_code = get_verification_code_api(email, mail_api_base, mail_api_token)
if not otp_code:
log_cb("[X] 未能获取验证码")
return None
check_shutdown()
log_status("API注册", f"验证 OTP: {otp_code}")
if not reg.validate_otp(otp_code):
log_cb("[X] OTP 验证失败")
return None
log_cb("[OK] OTP 验证成功")
check_shutdown()
# 创建账户（带重试）
log_status("API注册", "创建账户...")
create_success = reg.create_account(real_name, birthdate)
# 如果创建失败，重新获取验证码再试一次
if not create_success:
check_shutdown()
log_cb("[!] 创建账户失败，尝试重新验证...")
# 重新发送验证邮件
@@ -456,6 +496,7 @@ def api_register_flow(
return None
log_cb("[OK] 验证邮件已重新发送")
check_shutdown()
# 重新获取验证码
time.sleep(2) # 等待新邮件
otp_code = get_verification_code_api(email, mail_api_base, mail_api_token)
@@ -463,12 +504,14 @@ def api_register_flow(
log_cb("[X] 未能获取新验证码")
return None
check_shutdown()
log_status("API注册", f"重新验证 OTP: {otp_code}")
if not reg.validate_otp(otp_code):
log_cb("[X] OTP 重新验证失败")
return None
log_cb("[OK] OTP 重新验证成功")
check_shutdown()
# 再次尝试创建账户
log_status("API注册", "重新创建账户...")
if not reg.create_account(real_name, birthdate):
@@ -477,6 +520,7 @@ def api_register_flow(
log_cb("[OK] 账户创建成功")
check_shutdown()
# 验证 session 是否有效
token = reg.get_session_token()
if token:
@@ -486,6 +530,8 @@ def api_register_flow(
return reg
except ShutdownRequested:
raise # 重新抛出停止请求异常
except Exception as e:
log_status("错误", f"注册异常: {e}")
return None

5
auto_gpt_team.py
View File

@@ -30,6 +30,7 @@ try:
api_login_flow,
is_api_mode_available,
get_verification_code_api,
ShutdownRequested,
)
API_MODE_AVAILABLE = is_api_mode_available()
except ImportError:
@@ -37,6 +38,7 @@ except ImportError:
ChatGPTAPIRegister = None
api_register_flow = None
api_login_flow = None
ShutdownRequested = Exception # 回退到基础异常类
# ================= 配置加载 =================
try:
@@ -2432,6 +2434,9 @@ def run_single_registration_api(progress_callback=None, step_callback=None, prox
log_status("失败", "注册成功但支付/获取token失败")
return {"success": False, "error": "支付流程失败", "account": email, "password": password}
except ShutdownRequested:
log_status("停止", "[!] 用户请求停止")
return {"success": False, "error": "用户停止", "stopped": True, "account": email, "password": password}
except Exception as e:
error_msg = str(e)
# 只有连接断开才认为是停止请求

110
telegram_bot.py
View File

@@ -2716,6 +2716,54 @@ class ProvisionerBot:
except Exception as e:
await update.message.reply_text(f"❌ 修改配置失败: {e}")
async def _test_mail_api_connection(self, mail_api_base: str, mail_api_token: str, domain: str) -> tuple[bool, str]:
"""测试邮件 API 连接
Args:
mail_api_base: 邮件 API 地址
mail_api_token: 邮件 API Token
domain: 测试用的邮箱域名
Returns:
tuple: (success, message)
"""
import requests
import random
import string
try:
# 生成测试邮箱
random_str = ''.join(random.choices(string.ascii_lowercase + string.digits, k=8))
test_email = f"test-{random_str}@{domain.lstrip('@')}"
# 测试 API 连接
url = f"{mail_api_base}/api/public/emailList"
headers = {
"Authorization": mail_api_token,
"Content-Type": "application/json"
}
payload = {
"toEmail": test_email,
"timeSort": "desc",
"size": 1
}
response = requests.post(url, headers=headers, json=payload, timeout=10)
data = response.json()
if data.get("code") == 200:
return True, "邮件 API 连接正常"
else:
error_msg = data.get("message", "未知错误")
return False, f"API 响应异常: {error_msg}"
except requests.exceptions.Timeout:
return False, "连接超时，无法连接到邮件 API 服务器"
except requests.exceptions.ConnectionError:
return False, "连接失败，请检查 mail_api_base 配置"
except Exception as e:
return False, f"测试失败: {e}"
@admin_only
async def cmd_team_register(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
"""开始 GPT Team 自动订阅注册"""
@@ -2757,6 +2805,25 @@ class ProvisionerBot:
parse_mode="HTML"
)
return
# 测试邮件 API 连接
await update.message.reply_text("⏳ 正在检测邮件 API 连接...")
import random
test_domain = random.choice(domains)
success, message = await self._test_mail_api_connection(MAIL_API_BASE, MAIL_API_TOKEN, test_domain)
if not success:
await update.message.reply_text(
f"<b>❌ 邮件 API 连接失败</b>\n\n"
f"错误: {message}\n\n"
f"请检查配置后重试:\n"
f"• mail_api_base: {MAIL_API_BASE}\n"
f"• mail_api_token: {'已配置' if MAIL_API_TOKEN else '未配置'}",
parse_mode="HTML"
)
return
except ImportError:
await update.message.reply_text("❌ auto_gpt_team 模块未找到")
return
@@ -2943,6 +3010,8 @@ class ProvisionerBot:
try:
import run
if run._shutdown_requested:
with step_lock:
current_step[0] = "用户请求停止..."
break
except:
pass
@@ -2953,6 +3022,13 @@ class ProvisionerBot:
import functools
def run_with_callback():
# 在执行前再次检查停止请求
try:
import run as run_module
if run_module._shutdown_requested:
return {"success": False, "error": "用户停止", "stopped": True}
except:
pass
return run_single_registration_auto(
progress_callback=None,
step_callback=step_callback
@@ -2971,6 +3047,8 @@ class ProvisionerBot:
if result.get("stopped"):
# 被 /stop 命令中断，不计入失败
log.info("注册被用户停止")
with step_lock:
current_step[0] = "已停止"
break
elif result.get("success"):
success_count += 1
@@ -2985,12 +3063,35 @@ class ProvisionerBot:
else:
fail_count += 1
log.warning(f"注册失败: {result.get('error', '未知错误')}")
except asyncio.CancelledError:
log.info("注册任务被取消")
with step_lock:
current_step[0] = "已取消"
break
except Exception as e:
fail_count += 1
log.error(f"注册异常: {e}")
# 清理浏览器进程
cleanup_chrome_processes()
# 每次注册后检查停止请求
try:
import run
if run._shutdown_requested:
with step_lock:
current_step[0] = "用户请求停止..."
break
except:
pass
# 检查是否被停止
stopped = False
try:
import run
stopped = run._shutdown_requested
except:
pass
# 停止进度更新任务
progress_task.cancel()
@@ -3001,8 +3102,15 @@ class ProvisionerBot:
# 完成进度
progress_bar = '▰' * 20
completed = success_count + fail_count
if stopped:
status_text = f"<b>🛑 注册已停止</b> {completed}/{count}"
else:
status_text = f"<b>🎉 注册完成!</b> {success_count}/{count}"
await progress_msg.edit_text(
f"<b>🎉 注册完成!</b> {success_count}/{count}\n"
f"{status_text}\n"
f"{progress_bar}\n\n"
f"✅ 成功: {success_count}\n"
f"❌ 失败: {fail_count}",