feat: 优化了处理逻辑

This commit is contained in:
cheng zhen
2025-02-04 14:57:00 +08:00
parent 12eb044692
commit 7583136f08

View File

@@ -370,7 +370,9 @@ def check_cursor_version():
return None return None
if not os.path.exists(package_path): if not os.path.exists(package_path):
logging.warning("未找到 Cursor 安装") logging.warning(
"未找到 Cursor 安装, 或者你自定义了安装路径默认版本小于0.45"
)
return None return None
with open(package_path, "r", encoding="utf-8") as f: with open(package_path, "r", encoding="utf-8") as f:
@@ -382,9 +384,9 @@ def check_cursor_version():
if len(version_parts) >= 2: if len(version_parts) >= 2:
major_minor = float(f"{version_parts[0]}.{version_parts[1]}") major_minor = float(f"{version_parts[0]}.{version_parts[1]}")
if major_minor > 0.44: if major_minor > 0.44:
return False
else:
return True return True
else:
return False
else: else:
logging.warning("无法获取版本信息") logging.warning("无法获取版本信息")
return None return None
@@ -394,18 +396,19 @@ def check_cursor_version():
return None return None
def reset_machine_id(less_than_0_45): def reset_machine_id(greater_than_0_45):
if less_than_0_45: if greater_than_0_45:
MachineIDResetter().reset_machine_ids()
else:
# 提示请手动执行脚本 https://github.com/chengazhen/cursor-auto-free/blob/main/patch_cursor_get_machine_id.py # 提示请手动执行脚本 https://github.com/chengazhen/cursor-auto-free/blob/main/patch_cursor_get_machine_id.py
logging.info( logging.info(
f"{Fore.RED}请手动执行脚本 https://github.com/chengazhen/cursor-auto-free/blob/main/patch_cursor_get_machine_id.py{Style.RESET_ALL}" f"{Fore.RED}请手动执行脚本 https://github.com/chengazhen/cursor-auto-free/blob/main/patch_cursor_get_machine_id.py{Style.RESET_ALL}"
) )
else:
MachineIDResetter().reset_machine_ids()
if __name__ == "__main__": if __name__ == "__main__":
print_logo() print_logo()
greater_than_0_45 = check_cursor_version()
browser_manager = None browser_manager = None
try: try:
logging.info("\n=== 初始化程序 ===") logging.info("\n=== 初始化程序 ===")
@@ -413,6 +416,7 @@ if __name__ == "__main__":
print("\n请选择操作模式:") print("\n请选择操作模式:")
print("1. 仅重置机器码") print("1. 仅重置机器码")
print("2. 完整注册流程") print("2. 完整注册流程")
while True: while True:
try: try:
choice = int(input("请输入选项 (1 或 2): ").strip()) choice = int(input("请输入选项 (1 或 2): ").strip())
@@ -425,14 +429,12 @@ if __name__ == "__main__":
if choice == 1: if choice == 1:
# 仅执行重置机器码 # 仅执行重置机器码
less_than_0_45 = check_cursor_version() reset_machine_id(greater_than_0_45)
reset_machine_id(less_than_0_45)
logging.info("机器码重置完成") logging.info("机器码重置完成")
sys.exit(0) sys.exit(0)
# 小于0.45的版本需要打补丁 # 小于0.45的版本需要打补丁
less_than_0_45 = check_cursor_version() if not greater_than_0_45:
if less_than_0_45:
ExitCursor() ExitCursor()
logging.info("正在初始化浏览器...") logging.info("正在初始化浏览器...")
@@ -488,7 +490,7 @@ if __name__ == "__main__":
) )
logging.info("重置机器码...") logging.info("重置机器码...")
reset_machine_id(less_than_0_45) reset_machine_id(greater_than_0_45)
logging.info("所有操作已完成") logging.info("所有操作已完成")
else: else:
logging.error("获取会话令牌失败,注册流程未完成") logging.error("获取会话令牌失败,注册流程未完成")