From 7583136f0879ea4d6dfc73fae8b718f61e9f90a6 Mon Sep 17 00:00:00 2001
From: cheng zhen <chengchongzhen6@gmail.com>
Date: Tue, 4 Feb 2025 14:57:00 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E4=BC=98=E5=8C=96=E4=BA=86=E5=A4=84?=
 =?UTF-8?q?=E7=90=86=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cursor_pro_keep_alive.py | 26 ++++++++++++++------------
 1 file changed, 14 insertions(+), 12 deletions(-)

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