From a953db3bb89f83b7350806b97310f7af678ee02e Mon Sep 17 00:00:00 2001 From: chengchongzhen <15939054361@163.com> Date: Sat, 11 Jan 2025 13:41:49 +0800 Subject: [PATCH] refactor: replace print statements with logging for improved traceability and error handling in cursor_pro_keep_alive.py --- cursor_pro_keep_alive.py | 92 +++++++++++++++++++++++++--------------- 1 file changed, 58 insertions(+), 34 deletions(-) diff --git a/cursor_pro_keep_alive.py b/cursor_pro_keep_alive.py index 8266700..2555845 100644 --- a/cursor_pro_keep_alive.py +++ b/cursor_pro_keep_alive.py @@ -18,7 +18,7 @@ from config import Config def handle_turnstile(tab): - print("开始突破难关") + logging.info("正在检测 Turnstile 验证...") try: while True: try: @@ -31,29 +31,28 @@ def handle_turnstile(tab): ) if challengeCheck: - print("开始突破") + logging.info("检测到 Turnstile 验证，正在处理...") time.sleep(random.uniform(1, 3)) challengeCheck.click() time.sleep(2) - print("突破成功") + logging.info("Turnstile 验证通过") return True except: pass if tab.ele("@name=password"): - print("突破成功") + logging.info("验证成功 - 已到达密码输入页面") break if tab.ele("@data-index=0"): - print("突破成功") + logging.info("验证成功 - 已到达验证码输入页面") break if tab.ele("Account Settings"): - print("突破成功") + logging.info("验证成功 - 已到达账户设置页面") break time.sleep(random.uniform(1, 2)) except Exception as e: - print(e) - print("突破失败") + logging.error(f"Turnstile 验证失败: {str(e)}") return False @@ -65,7 +64,7 @@ def get_cursor_session_token(tab, max_attempts=3, retry_interval=2): :param retry_interval: 重试间隔(秒) :return: session token 或 None """ - print("开始获取cookie") + logging.info("开始获取cookie") attempts = 0 while attempts < max_attempts: @@ -77,18 +76,20 @@ def get_cursor_session_token(tab, max_attempts=3, retry_interval=2): attempts += 1 if attempts < max_attempts: - print( + logging.warning( f"第 {attempts} 次尝试未获取到CursorSessionToken，{retry_interval}秒后重试..." ) time.sleep(retry_interval) else: - print(f"已达到最大尝试次数({max_attempts})，获取CursorSessionToken失败") + logging.error( + f"已达到最大尝试次数({max_attempts})，获取CursorSessionToken失败" + ) except Exception as e: - print(f"获取cookie失败: {str(e)}") + logging.error(f"获取cookie失败: {str(e)}") attempts += 1 if attempts < max_attempts: - print(f"将在 {retry_interval} 秒后重试...") + logging.info(f"将在 {retry_interval} 秒后重试...") time.sleep(retry_interval) return None @@ -103,43 +104,51 @@ def update_cursor_auth(email=None, access_token=None, refresh_token=None): def sign_up_account(browser, tab): - print("开始执行...") + logging.info("=== 开始注册账号流程 ===") + logging.info(f"正在访问注册页面: {sign_up_url}") tab.get(sign_up_url) try: if tab.ele("@name=first_name"): + logging.info("正在填写个人信息...") tab.actions.click("@name=first_name").input(first_name) + logging.info(f"已输入名字: {first_name}") time.sleep(random.uniform(1, 3)) tab.actions.click("@name=last_name").input(last_name) + logging.info(f"已输入姓氏: {last_name}") time.sleep(random.uniform(1, 3)) tab.actions.click("@name=email").input(account) + logging.info(f"已输入邮箱: {account}") time.sleep(random.uniform(1, 3)) + logging.info("提交个人信息...") tab.actions.click("@type=submit") except Exception as e: - print("打开注册页面失败") + logging.error(f"注册页面访问失败: {str(e)}") return False handle_turnstile(tab) try: if tab.ele("@name=password"): + logging.info("正在设置密码...") tab.ele("@name=password").input(password) time.sleep(random.uniform(1, 3)) + logging.info("提交密码...") tab.ele("@type=submit").click() - print("请稍等...") + logging.info("密码设置完成，等待系统响应...") except Exception as e: - print("执行失败") + logging.error(f"密码设置失败: {str(e)}") return False time.sleep(random.uniform(1, 3)) if tab.ele("This email is not available."): - print("执行失败") + logging.error("注册失败：邮箱已被使用") return False handle_turnstile(tab) @@ -147,26 +156,34 @@ def sign_up_account(browser, tab): while True: try: if tab.ele("Account Settings"): + logging.info("注册成功 - 已进入账户设置页面") break if tab.ele("@data-index=0"): + logging.info("正在获取邮箱验证码...") code = email_handler.get_verification_code() if not code: + logging.error("获取验证码失败") return False + logging.info(f"成功获取验证码: {code}") + logging.info("正在输入验证码...") i = 0 for digit in code: tab.ele(f"@data-index={i}").input(digit) time.sleep(random.uniform(0.1, 0.3)) i += 1 + logging.info("验证码输入完成") break except Exception as e: - print(e) + logging.error(f"验证码处理过程出错: {str(e)}") handle_turnstile(tab) wait_time = random.randint(3, 6) for i in range(wait_time): - print(f"等待中... {wait_time-i}秒") + logging.info(f"等待系统处理中... 剩余 {wait_time-i} 秒") time.sleep(1) + + logging.info("正在获取账户信息...") tab.get(settings_url) try: usage_selector = ( @@ -178,11 +195,12 @@ def sign_up_account(browser, tab): if usage_ele: usage_info = usage_ele.text total_usage = usage_info.split("/")[-1].strip() - print("可用上限: " + total_usage) + logging.info(f"账户可用额度上限: {total_usage}") except Exception as e: - print("获取可用上限失败") - print("注册完成") - account_info = f"\nCursor 账号： {account} 密码： {password}" + logging.error(f"获取账户额度信息失败: {str(e)}") + + logging.info("\n=== 注册完成 ===") + account_info = f"Cursor 账号信息:\n邮箱: {account}\n密码: {password}" logging.info(account_info) time.sleep(5) return True @@ -225,53 +243,59 @@ if __name__ == "__main__": print_logo() browser_manager = None try: - ExitCursor() - # 初始化浏览器 + logging.info("\n=== 初始化程序 ===") + # ExitCursor() + logging.info("正在初始化浏览器...") browser_manager = BrowserManager() browser = browser_manager.init_browser() - # 初始化邮箱验证处理器 + logging.info("正在初始化邮箱验证模块...") email_handler = EmailVerificationHandler(browser) - # 固定的 URL 配置 + logging.info("\n=== 配置信息 ===") login_url = "https://authenticator.cursor.sh" sign_up_url = "https://authenticator.cursor.sh/sign-up" settings_url = "https://www.cursor.com/settings" mail_url = "https://tempmail.plus" - # 生成随机邮箱 + logging.info("正在生成随机账号信息...") email_generator = EmailGenerator() account = email_generator.generate_email() password = email_generator.default_password first_name = email_generator.default_first_name last_name = email_generator.default_last_name + logging.info(f"生成的邮箱账号: {account}") auto_update_cursor_auth = True tab = browser.latest_tab tab.run_js("try { turnstile.reset() } catch(e) { }") + logging.info("\n=== 开始注册流程 ===") + logging.info(f"正在访问登录页面: {login_url}") tab.get(login_url) if sign_up_account(browser, tab): + logging.info("正在获取会话令牌...") token = get_cursor_session_token(tab) if token: + logging.info("更新认证信息...") update_cursor_auth( email=account, access_token=token, refresh_token=token ) + logging.info("重置机器码...") MachineIDResetter().reset_machine_ids() + logging.info("所有操作已完成") else: - print("账户注册失败") - - print("执行完毕") + logging.error("获取会话令牌失败，注册流程未完成") except Exception as e: - logging.error(f"程序执行出错: {str(e)}") + logging.error(f"程序执行出现错误: {str(e)}") import traceback logging.error(traceback.format_exc()) finally: if browser_manager: browser_manager.quit() - input("\n按回车键退出...") + input("\n程序执行完毕，按回车键退出...")