openclaw/grokzhuce
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化注册流程：即时NSFW验证与完善邮箱清理

Browse Source 
- 新增 current_email 追踪机制，确保所有异常路径都能正确删除临时邮箱
- 将 NSFW 二次验证 (enable_unhinged) 从批量后处理改为注册时即时执行
- 在所有退出点（验证失败、SSO获取失败、TOS/NSFW设置失败、CAPTCHA重试耗尽等）添加邮箱清理
- 优化成功输出格式：显示进度、邮箱、SSO预览、平均耗时、NSFW状态
- 添加目标数量达成提示信息
- 移除冗余的批量 NSFW 后处理代码
This commit is contained in:
muqing-kg
2026-02-05 18:06:31 +08:00
parent a89a7c5912
commit 42be5c85d9
1 changed files with 51 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

80
grok.py
View File

@@ -102,7 +102,7 @@ def verify_email_code_grpc(session, email, code):
def register_single_thread(): def register_single_thread():
# 错峰启动，防止瞬时并发过高 # 错峰启动，防止瞬时并发过高
time.sleep(random.uniform(0, 5)) time.sleep(random.uniform(0, 5))
try: try:
email_service = EmailService() email_service = EmailService()
turnstile_service = TurnstileService() turnstile_service = TurnstileService()
@@ -111,16 +111,21 @@ def register_single_thread():
except Exception as e: except Exception as e:
print(f"[-] 服务初始化失败: {e}") print(f"[-] 服务初始化失败: {e}")
return return
# 修正：直接从 config 获取 # 修正：直接从 config 获取
final_action_id = config["action_id"] final_action_id = config["action_id"]
if not final_action_id: if not final_action_id:
print("[-] 线程退出：缺少 Action ID") print("[-] 线程退出：缺少 Action ID")
return return
current_email = None # 追踪当前邮箱，确保异常时能删除
while True: while True:
try: try:
if stop_event.is_set(): if stop_event.is_set():
if current_email:
try: email_service.delete_email(current_email)
except: pass
return return
impersonate_fingerprint, account_user_agent = get_random_chrome_profile() impersonate_fingerprint, account_user_agent = get_random_chrome_profile()
with requests.Session(impersonate=impersonate_fingerprint, proxies=PROXIES) as session: with requests.Session(impersonate=impersonate_fingerprint, proxies=PROXIES) as session:
@@ -129,49 +134,52 @@ def register_single_thread():
except: pass except: pass
password = generate_random_string() password = generate_random_string()
# print(f"[debug] 线程-{threading.get_ident()} 正在请求创建邮箱...")
try: try:
jwt, email = email_service.create_email() jwt, email = email_service.create_email()
current_email = email
except Exception as e: except Exception as e:
print(f"[-] 邮箱服务抛出异常: {e}") print(f"[-] 邮箱服务抛出异常: {e}")
jwt, email = None, None jwt, email, current_email = None, None, None
if not email: if not email:
time.sleep(5); continue time.sleep(5); continue
if stop_event.is_set(): if stop_event.is_set():
if email: email_service.delete_email(email)
email_service.delete_email(email) current_email = None
return return
print(f"[*] 开始注册: {email}") print(f"[*] 开始注册: {email}")
# Step 1: 发送验证码 # Step 1: 发送验证码
if not send_email_code_grpc(session, email): if not send_email_code_grpc(session, email):
email_service.delete_email(email) email_service.delete_email(email)
current_email = None
time.sleep(5); continue time.sleep(5); continue
# Step 2: 获取验证码 # Step 2: 获取验证码
verify_code = email_service.fetch_verification_code(email) verify_code = email_service.fetch_verification_code(email)
if not verify_code: if not verify_code:
email_service.delete_email(email) email_service.delete_email(email)
current_email = None
continue continue
# Step 3: 验证验证码 # Step 3: 验证验证码
if not verify_email_code_grpc(session, email, verify_code): if not verify_email_code_grpc(session, email, verify_code):
email_service.delete_email(email) email_service.delete_email(email)
current_email = None
continue continue
# Step 4: 注册重试循环 # Step 4: 注册重试循环
for attempt in range(3): for attempt in range(3):
if stop_event.is_set(): if stop_event.is_set():
email_service.delete_email(email) email_service.delete_email(email)
current_email = None
return return
task_id = turnstile_service.create_task(site_url, config["site_key"]) task_id = turnstile_service.create_task(site_url, config["site_key"])
# 这里不再打印获取 Token 的过程，只在失败时报错
token = turnstile_service.get_response(task_id) token = turnstile_service.get_response(task_id)
if not token or token == "CAPTCHA_FAIL": if not token or token == "CAPTCHA_FAIL":
continue continue
@@ -188,14 +196,15 @@ def register_single_thread():
}, },
"turnstileToken": token, "promptOnDuplicateEmail": True "turnstileToken": token, "promptOnDuplicateEmail": True
}] }]
with post_lock: with post_lock:
res = session.post(f"{site_url}/sign-up", json=payload, headers=headers) res = session.post(f"{site_url}/sign-up", json=payload, headers=headers)
if res.status_code == 200: if res.status_code == 200:
match = re.search(r'(https://[^" \s]+set-cookie\?q=[^:" \s]+)1:', res.text) match = re.search(r'(https://[^" \s]+set-cookie\?q=[^:" \s]+)1:', res.text)
if not match: if not match:
email_service.delete_email(email) email_service.delete_email(email)
current_email = None
break break
if match: if match:
verify_url = match.group(1) verify_url = match.group(1)
@@ -204,6 +213,7 @@ def register_single_thread():
sso_rw = session.cookies.get("sso-rw") sso_rw = session.cookies.get("sso-rw")
if not sso: if not sso:
email_service.delete_email(email) email_service.delete_email(email)
current_email = None
break break
tos_result = user_agreement_service.accept_tos_version( tos_result = user_agreement_service.accept_tos_version(
@@ -215,6 +225,7 @@ def register_single_thread():
tos_hex = tos_result.get("hex_reply") or "" tos_hex = tos_result.get("hex_reply") or ""
if not tos_result.get("ok") or not tos_hex: if not tos_result.get("ok") or not tos_hex:
email_service.delete_email(email) email_service.delete_email(email)
current_email = None
break break
nsfw_result = nsfw_service.enable_nsfw( nsfw_result = nsfw_service.enable_nsfw(
@@ -226,32 +237,56 @@ def register_single_thread():
nsfw_hex = nsfw_result.get("hex_reply") or "" nsfw_hex = nsfw_result.get("hex_reply") or ""
if not nsfw_result.get("ok") or not nsfw_hex: if not nsfw_result.get("ok") or not nsfw_hex:
email_service.delete_email(email) email_service.delete_email(email)
current_email = None
break break
# 立即进行二次验证 (enable_unhinged)
unhinged_result = nsfw_service.enable_unhinged(sso)
unhinged_ok = unhinged_result.get("ok", False)
with file_lock: with file_lock:
global success_count global success_count
if success_count >= target_count: if success_count >= target_count:
if not stop_event.is_set(): if not stop_event.is_set():
stop_event.set() stop_event.set()
print(f"[*] 已达到目标数量，删除邮箱: {email}")
email_service.delete_email(email) email_service.delete_email(email)
current_email = None
break
try:
with open(output_file, "a") as f: f.write(sso + "\n")
except Exception as write_err:
print(f"[-] 写入文件失败: {write_err}")
email_service.delete_email(email)
current_email = None
break break
with open(output_file, "a") as f: f.write(sso + "\n")
success_count += 1 success_count += 1
avg = (time.time() - start_time) / success_count avg = (time.time() - start_time) / success_count
print(f"[+] {success_count}/{target_count} {email} | {avg:.1f}s/个") nsfw_tag = "✓" if unhinged_ok else "✗"
print(f"[✓] 注册成功: {success_count}/{target_count} | {email} | SSO: {sso[:15]}... | 平均: {avg:.1f}s | NSFW: {nsfw_tag}")
email_service.delete_email(email) email_service.delete_email(email)
current_email = None
if success_count >= target_count and not stop_event.is_set(): if success_count >= target_count and not stop_event.is_set():
stop_event.set() stop_event.set()
print(f"[*] 已达到目标数量: {success_count}/{target_count}，停止新注册")
break # 跳出 for 循环，继续 while True 注册下一个 break # 跳出 for 循环，继续 while True 注册下一个
time.sleep(3) time.sleep(3)
else: else:
# 如果重试 3 次都失败 (for 循环没有被 break) # 如果重试 3 次都失败 (for 循环没有被 break)
email_service.delete_email(email) email_service.delete_email(email)
current_email = None
time.sleep(5) time.sleep(5)
except Exception as e: except Exception as e:
print(f"[-] 异常: {str(e)[:50]}") print(f"[-] 异常: {str(e)[:50]}")
# 异常时确保删除邮箱
if current_email:
try:
email_service.delete_email(current_email)
except:
pass
current_email = None
time.sleep(5) time.sleep(5)
def main(): def main():
@@ -310,18 +345,5 @@ def main():
futures = [executor.submit(register_single_thread) for _ in range(t)] futures = [executor.submit(register_single_thread) for _ in range(t)]
concurrent.futures.wait(futures) concurrent.futures.wait(futures)
# 二次验证 NSFW
if os.path.exists(output_file):
print(f"\n[*] 开始二次验证 NSFW...")
nsfw_service = NsfwSettingsService()
with open(output_file, "r") as f:
tokens = [line.strip() for line in f if line.strip()]
ok_count = 0
for sso in tokens:
result = nsfw_service.enable_unhinged(sso)
if result.get("ok"):
ok_count += 1
print(f"[*] 二次验证完成: {ok_count}/{len(tokens)}")
if __name__ == "__main__": if __name__ == "__main__":
main() main()