Add initial implementation of Cursor Pro Keep Alive tool

- Created .gitignore to exclude unnecessary files. - Added config.ini for account configuration. - Implemented cursor_auth_manager.py for managing authentication. - Developed cursor_pro_keep_alive.py for automated account management. - Included CursorKeepAlive.spec for packaging. - Added gpt-accesstoken.py for obtaining ChatGPT access tokens. - Created README.md with usage instructions. - Updated requirements.txt with necessary dependencies. - Added run_cursor.bat for script execution. - Implemented update_cursor_auth.py for updating authentication details. - Included turnstilePatch for handling CAPTCHA challenges. - Added logging functionality in log.txt for tracking script execution. - Created dist folder with packaged executable and necessary files.
2024-12-28 15:14:40 +08:00
commit d48b1a9d6d
18 changed files with 1045 additions and 0 deletions
--- a/cursor_pro_keep_alive.py
+++ b/cursor_pro_keep_alive.py
@@ -0,0 +1,453 @@
+from DrissionPage import ChromiumOptions, Chromium
+from DrissionPage.common import Keys
+import re
+import time
+import random
+from cursor_auth_manager import CursorAuthManager
+from configparser import ConfigParser
+import os
+import sys
+
+
+def load_config():
+    """加载配置文件"""
+    config = ConfigParser()
+
+    # 获取程序运行的根目录
+    root_dir = os.getcwd()
+    config_path = os.path.join(root_dir, "config.ini")
+
+    if os.path.exists(config_path):
+        config.read(config_path, encoding="utf-8")
+        print(f"已加载配置文件: {config_path}")
+        return {
+            "account": config["Account"]["email"],
+            "password": config["Account"]["password"],
+            "first_name": config["Account"]["first_name"],
+            "last_name": config["Account"]["last_name"],
+        }
+
+    raise FileNotFoundError(
+        f"配置文件不存在: {config_path}\n请确保配置文件在程序运行目录下"
+    )
+
+
+def get_veri_code(tab):
+    """获取验证码"""
+    username = account.split("@")[0]
+    try:
+        while True:
+            if tab.ele("@id=pre_button"):
+                tab.actions.click("@id=pre_button").type(Keys.CTRL_A).key_down(
+                    Keys.BACKSPACE
+                ).key_up(Keys.BACKSPACE).input(username).key_down(Keys.ENTER).key_up(
+                    Keys.ENTER
+                )
+                break
+            time.sleep(1)
+
+        while True:
+            new_mail = tab.ele("@class=mail")
+            if new_mail:
+                if new_mail.text:
+                    print("最新的邮件：", new_mail.text)
+                    tab.actions.click("@class=mail")
+                    break
+                else:
+                    print(new_mail)
+                    break
+            time.sleep(1)
+
+        if tab.ele("@class=overflow-auto mb-20"):
+            email_content = tab.ele("@class=overflow-auto mb-20").text
+            verification_code = re.search(
+                r"verification code is (\d{6})", email_content
+            )
+            if verification_code:
+                code = verification_code.group(1)
+                print("验证码：", code)
+            else:
+                print("未找到验证码")
+
+        if tab.ele("@id=delete_mail"):
+            tab.actions.click("@id=delete_mail")
+            time.sleep(1)
+
+        if tab.ele("@id=confirm_mail"):
+            tab.actions.click("@id=confirm_mail")
+            print("删除邮件")
+        tab.close()
+    except Exception as e:
+        print(e)
+
+    return code
+
+
+def handle_turnstile(tab):
+    """处理 Turnstile 验证"""
+    print("准备处理验证")
+    try:
+        while True:
+            try:
+                challengeCheck = (
+                    tab.ele("@id=cf-turnstile", timeout=2)
+                    .child()
+                    .shadow_root.ele("tag:iframe")
+                    .ele("tag:body")
+                    .sr("tag:input")
+                )
+
+                if challengeCheck:
+                    print("验证框加载完成")
+                    time.sleep(random.uniform(1, 3))
+                    challengeCheck.click()
+                    print("验证按钮已点击，等待验证完成...")
+                    time.sleep(2)
+                    return True
+            except:
+                pass
+
+            if tab.ele("@name=password"):
+                print("无需验证")
+                break
+            if tab.ele("@data-index=0"):
+                print("无需验证")
+                break
+            if tab.ele("Account Settings"):
+                print("无需验证")
+                break
+
+            time.sleep(random.uniform(1, 2))
+    except Exception as e:
+        print(e)
+        print("跳过验证")
+        return False
+
+
+def delete_account(browser, tab):
+    """删除账户流程"""
+    print("\n开始删除账户...")
+
+    try:
+        if tab.ele("@name=email"):
+            tab.ele("@name=email").input(account)
+            print("输入账号")
+            time.sleep(random.uniform(1, 3))
+    except Exception as e:
+        print(f"输入账号失败: {str(e)}")
+
+    try:
+        if tab.ele("Continue"):
+            tab.ele("Continue").click()
+            print("点击Continue")
+    except Exception as e:
+        print(f"点击Continue失败: {str(e)}")
+
+    handle_turnstile(tab)
+    time.sleep(5)
+
+    try:
+        if tab.ele("@name=password"):
+            tab.ele("@name=password").input(password)
+            print("输入密码")
+            time.sleep(random.uniform(1, 3))
+    except Exception as e:
+        print("输入密码失败")
+
+    sign_in_button = tab.ele(
+        "xpath:/html/body/div[1]/div/div/div[2]/div/form/div/button"
+    )
+    try:
+        if sign_in_button:
+            sign_in_button.click(by_js=True)
+            print("点击Sign in")
+    except Exception as e:
+        print(f"点击Sign in失败: {str(e)}")
+
+    handle_turnstile(tab)
+
+    # 处理验证码
+    while True:
+        try:
+            if tab.ele("Account Settings"):
+                break
+            if tab.ele("@data-index=0"):
+                tab_mail = browser.new_tab(mail_url)
+                browser.activate_tab(tab_mail)
+                print("打开邮箱页面")
+                code = get_veri_code(tab_mail)
+
+                if code:
+                    print("获取验证码成功：", code)
+                    browser.activate_tab(tab)
+                else:
+                    print("获取验证码失败，程序退出")
+                    return False
+
+                i = 0
+                for digit in code:
+                    tab.ele(f"@data-index={i}").input(digit)
+                    time.sleep(random.uniform(0.1, 0.3))
+                    i += 1
+                break
+        except Exception as e:
+            print(e)
+
+    handle_turnstile(tab)
+    time.sleep(random.uniform(1, 3))
+    # tab.get_screenshot('sign-in_success.png')
+    # print("登录账户截图")
+
+    tab.get(settings_url)
+    print("进入设置页面")
+
+    try:
+        if tab.ele("@class=mt-1"):
+            tab.ele("@class=mt-1").click()
+            print("点击Adavance")
+            time.sleep(random.uniform(1, 2))
+    except Exception as e:
+        print(f"点击Adavance失败: {str(e)}")
+
+    try:
+        if tab.ele("Delete Account"):
+            tab.ele("Delete Account").click()
+            print("点击Delete Account")
+            time.sleep(random.uniform(1, 2))
+    except Exception as e:
+        print(f"点击Delete Account失败: {str(e)}")
+
+    try:
+        if tab.ele("tag:input"):
+            tab.actions.click("tag:input").type("delete")
+            print("输入delete")
+            time.sleep(random.uniform(1, 2))
+    except Exception as e:
+        print(f"输入delete失败: {str(e)}")
+
+    delete_button = tab.ele(
+        "xpath:/html/body/main/div/div/div/div/div/div[1]/div[2]/div[3]/div[2]/div/div/div[2]/button[2]"
+    )
+    try:
+        if delete_button:
+            print("点击Delete")
+            delete_button.click()
+            time.sleep(5)
+            # tab.get_screenshot('delete_account.png')
+            # print("删除账户截图")
+            return True
+    except Exception as e:
+        print(f"点击Delete失败: {str(e)}")
+        return False
+
+
+def get_cursor_session_token(tab):
+    """获取cursor session token"""
+    cookies = tab.cookies()
+    cursor_session_token = None
+    for cookie in cookies:
+        if cookie["name"] == "WorkosCursorSessionToken":
+            cursor_session_token = cookie["value"].split("%3A%3A")[1]
+            break
+    return cursor_session_token
+
+
+def update_cursor_auth(email=None, access_token=None, refresh_token=None):
+    """
+    更新Cursor的认证信息的便捷函数
+    """
+    auth_manager = CursorAuthManager()
+    return auth_manager.update_auth(email, access_token, refresh_token)
+
+
+def sign_up_account(browser, tab):
+    """注册账户流程"""
+    print("\n开始注册新账户...")
+    tab.get(sign_up_url)
+
+    try:
+        if tab.ele("@name=first_name"):
+            print("已打开注册页面")
+            tab.actions.click("@name=first_name").input(first_name)
+            time.sleep(random.uniform(1, 3))
+
+            tab.actions.click("@name=last_name").input(last_name)
+            time.sleep(random.uniform(1, 3))
+
+            tab.actions.click("@name=email").input(account)
+            print("输入邮箱")
+            time.sleep(random.uniform(1, 3))
+
+            tab.actions.click("@type=submit")
+            print("点击注册按钮")
+
+    except Exception as e:
+        print("打开注册页面失败")
+        return False
+
+    handle_turnstile(tab)
+
+    try:
+        if tab.ele("@name=password"):
+            tab.ele("@name=password").input(password)
+            print("输入密码")
+            time.sleep(random.uniform(1, 3))
+
+            tab.ele("@type=submit").click()
+            print("点击Continue按钮")
+
+    except Exception as e:
+        print("输入密码失败")
+        return False
+
+    time.sleep(random.uniform(1, 3))
+    if tab.ele("This email is not available."):
+        print("This email is not available.")
+        return False
+
+    handle_turnstile(tab)
+
+    while True:
+        try:
+            if tab.ele("Account Settings"):
+                break
+            if tab.ele("@data-index=0"):
+                tab_mail = browser.new_tab(mail_url)
+                browser.activate_tab(tab_mail)
+                print("打开邮箱页面")
+                code = get_veri_code(tab_mail)
+
+                if code:
+                    print("获取验证码成功：", code)
+                    browser.activate_tab(tab)
+                else:
+                    print("获取验证码失败，程序退出")
+                    return False
+
+                i = 0
+                for digit in code:
+                    tab.ele(f"@data-index={i}").input(digit)
+                    time.sleep(random.uniform(0.1, 0.3))
+                    i += 1
+                break
+        except Exception as e:
+            print(e)
+
+    handle_turnstile(tab)
+
+    time.sleep(random.uniform(1, 3))
+    print("进入设置页面")
+    tab.get(settings_url)
+    try:
+        usage_ele = tab.ele(
+            "xpath:/html/body/main/div/div/div/div/div/div[2]/div/div/div/div[1]/div[1]/span[2]"
+        )
+        if usage_ele:
+            usage_info = usage_ele.text
+            total_usage = usage_info.split("/")[-1].strip()
+            print("可用上限: " + total_usage)
+    except Exception as e:
+        print("获取可用上限失败")
+    # tab.get_screenshot("sign_up_success.png")
+    # print("注册账户截图")
+    print("注册完成")
+    print("Cursor 账号： " + account)
+    print("       密码： " + password)
+    return True
+
+
+def get_extension_path():
+    """获取插件路径"""
+    root_dir = os.getcwd()
+    extension_path = os.path.join(root_dir, "turnstilePatch")
+
+    if hasattr(sys, "_MEIPASS"):
+        print("运行在打包环境中")
+        extension_path = os.path.join(sys._MEIPASS, "turnstilePatch")
+
+    print(f"尝试加载插件路径: {extension_path}")
+
+    if not os.path.exists(extension_path):
+        raise FileNotFoundError(
+            f"插件不存在: {extension_path}\n请确保 turnstilePatch 文件夹在正确位置"
+        )
+
+    return extension_path
+
+
+if __name__ == "__main__":
+    try:
+        # 加载配置
+        config = load_config()
+
+        # 固定的 URL 配置
+        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"
+
+        # 账号信息
+        account = config["account"]
+        password = config["password"]
+        first_name = config["first_name"]
+        last_name = config["last_name"]
+
+        auto_update_cursor_auth = True
+
+        # 浏览器配置
+        co = ChromiumOptions()
+        try:
+            extension_path = get_extension_path()
+            co.add_extension(extension_path)
+        except FileNotFoundError as e:
+            print(f"警告: {e}")
+            print("将尝试继续执行...")
+
+        co.headless()  # 无头模式
+        co.set_user_agent(
+            "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/130.0.6723.92 Safari/537.36"
+        )
+        co.set_pref("credentials_enable_service", False)
+        co.set_argument("--hide-crash-restore-bubble")
+        co.auto_port()
+        # co.set_argument('--no-sandbox')        # 无沙盒模式     用于linux
+        # co.set_argument('--headless=new')      #无界面系统启动参数   用于linux
+        # co.set_proxy('127.0.0.1:10809')        #设置代理
+
+        browser = Chromium(co)
+        tab = browser.latest_tab
+        tab.run_js("try { turnstile.reset() } catch(e) { }")
+
+        print("开始执行删除和注册流程")
+        print("***请确认已经用https://tempmail.plus/zh邮箱成功申请过cursor账号！***")
+        tab.get(login_url)
+
+        # 执行删除和注册流程
+        if delete_account(browser, tab):
+            print("账户删除成功")
+            time.sleep(3)
+            if sign_up_account(browser, tab):
+                token = get_cursor_session_token(tab)
+                print(f"CursorSessionToken: {token}")
+                print("账户注册成功")
+                if auto_update_cursor_auth:
+                    update_cursor_auth(
+                        email=account, access_token=token, refresh_token=token
+                    )
+            else:
+                print("账户注册失败")
+        else:
+            print("账户删除失败")
+
+        print("脚本执行完毕")
+        browser.quit()
+
+        # 添加等待用户输入后再退出
+        input("\n按回车键退出...")
+    except FileNotFoundError as e:
+        print(f"<EFBFBD><EFBFBD><EFBFBD>误: {e}")
+        print("请确保 config.ini 文件存在并包含正确的配置信息")
+        input("\n按回车键退出...")
+    except Exception as e:
+        print(f"加载配置时出错: {e}")
+        input("\n按回车键退出...")