修正 windows 环境下用户目录的获取方式

bypass_token_limit.py

@@ -26,6 +26,13 @@ EMOJI = {
 def get_user_documents_path():
      """Get user Documents folder path"""
      if sys.platform == "win32":
+         try:
+             import winreg
+             with winreg.OpenKey(winreg.HKEY_CURRENT_USER, "Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\Shell Folders") as key:
+                 documents_path, _ = winreg.QueryValueEx(key, "Personal")
+                 return documents_path
+         except Exception as e:
+             # fallback
              return os.path.join(os.path.expanduser("~"), "Documents")
      elif sys.platform == "darwin":
          return os.path.join(os.path.expanduser("~"), "Documents")

new_signup.py

@@ -116,6 +116,13 @@ def fill_signup_form(page, first_name, last_name, email, config, translator=None
 def get_user_documents_path():
     """Get user Documents folder path"""
     if sys.platform == "win32":
+        try:
+            import winreg
+            with winreg.OpenKey(winreg.HKEY_CURRENT_USER, "Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\Shell Folders") as key:
+                documents_path, _ = winreg.QueryValueEx(key, "Personal")
+                return documents_path
+        except Exception as e:
+            # fallback
             return os.path.join(os.path.expanduser("~"), "Documents")
     elif sys.platform == "darwin":
         return os.path.join(os.path.expanduser("~"), "Documents")

reset_machine_manual.py

@@ -33,6 +33,13 @@ EMOJI = {
 def get_user_documents_path():
      """Get user Documents folder path"""
      if sys.platform == "win32":
+         try:
+             import winreg
+             with winreg.OpenKey(winreg.HKEY_CURRENT_USER, "Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\Shell Folders") as key:
+                 documents_path, _ = winreg.QueryValueEx(key, "Personal")
+                 return documents_path
+         except Exception as e:
+             # fallback
              return os.path.join(os.path.expanduser("~"), "Documents")
      elif sys.platform == "darwin":
          return os.path.join(os.path.expanduser("~"), "Documents")

totally_reset_cursor.py

@@ -32,6 +32,12 @@ EMOJI = {
 def get_user_documents_path():
      """Get user Documents folder path"""
      if sys.platform == "win32":
+         try:
+             import winreg
+             with winreg.OpenKey(winreg.HKEY_CURRENT_USER, "Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\Shell Folders") as key:
+                 documents_path, _ = winreg.QueryValueEx(key, "Personal")
+                 return documents_path
+         except Exception as e:
              return os.path.join(os.path.expanduser("~"), "Documents")
      elif sys.platform == "darwin":
          return os.path.join(os.path.expanduser("~"), "Documents")

utils.py

@@ -6,6 +6,15 @@ import random
 def get_user_documents_path():
     """Get user documents path"""
     if platform.system() == "Windows":
+        try:
+            import winreg
+            # 打开注册表
+            with winreg.OpenKey(winreg.HKEY_CURRENT_USER, "Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\Shell Folders") as key:
+                # 获取 "Personal" 键的值，这指向用户的文档目录
+                documents_path, _ = winreg.QueryValueEx(key, "Personal")
+                return documents_path
+        except Exception as e:
+            # fallback
             return os.path.expanduser("~\\Documents")
     else:
         return os.path.expanduser("~/Documents")