b3c9efc901
- Updated environment variable assignments to use double quotes for consistency. - Enhanced logging configuration format for clarity. - Improved configuration loading logic to correctly determine the executable's directory. - Cleaned up temporary file handling in cleanup_temp_files function. - Removed obsolete config.ini and related files from the dist directory. - Deleted unnecessary macOS application files from the dist folder.
503 lines
15 KiB
Python
503 lines
15 KiB
Python
import os
|
||
|
||
os.environ["PYTHONVERBOSE"] = "0"
|
||
os.environ["PYINSTALLER_VERBOSE"] = "0"
|
||
|
||
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
|
||
import logging
|
||
|
||
# 在文件开头设置日志
|
||
logging.basicConfig(
|
||
level=logging.WARNING,
|
||
format="%(asctime)s - %(levelname)s - %(message)s",
|
||
handlers=[
|
||
logging.StreamHandler(),
|
||
logging.FileHandler("cursor_keep_alive.log", encoding="utf-8"),
|
||
],
|
||
)
|
||
|
||
|
||
def load_config():
|
||
"""加载配置文件"""
|
||
config = ConfigParser()
|
||
|
||
# 获取程序运行的实际目录
|
||
if getattr(sys, "frozen", False):
|
||
# 打包后的情况：使用可执行文件所在目录
|
||
root_dir = os.path.dirname(sys.executable)
|
||
else:
|
||
# 开发环境：使用当前工作目录
|
||
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}")
|
||
print(config["Account"]["email"])
|
||
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}")
|
||
|
||
|
||
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
|
||
|
||
|
||
def get_browser_options():
|
||
co = ChromiumOptions()
|
||
try:
|
||
extension_path = get_extension_path()
|
||
co.add_extension(extension_path)
|
||
except FileNotFoundError as e:
|
||
print(f"警告: {e}")
|
||
|
||
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()
|
||
|
||
# Mac 系统特殊处理
|
||
if sys.platform == "darwin":
|
||
co.set_argument("--no-sandbox")
|
||
co.set_argument("--disable-gpu")
|
||
|
||
return co
|
||
|
||
|
||
def cleanup_temp_files():
|
||
"""清理临时文件和缓存"""
|
||
try:
|
||
temp_dirs = [
|
||
os.path.join(os.getcwd(), "__pycache__"),
|
||
os.path.join(os.getcwd(), "build"),
|
||
]
|
||
|
||
for dir_path in temp_dirs:
|
||
if os.path.exists(dir_path):
|
||
import shutil
|
||
|
||
shutil.rmtree(dir_path)
|
||
except Exception as e:
|
||
logging.warning(f"清理临时文件失败: {str(e)}")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
browser = None
|
||
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 = get_browser_options()
|
||
|
||
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("脚本执行完毕")
|
||
|
||
except Exception as e:
|
||
print(f"程序执行出错: {str(e)}")
|
||
import traceback
|
||
|
||
print(traceback.format_exc())
|
||
finally:
|
||
# 确保浏览器实例被正确关闭
|
||
if browser:
|
||
try:
|
||
browser.quit()
|
||
except:
|
||
pass
|
||
input("\n按回车键退出...")
|