📝

Author: jiangyangcreate

blog/2024-9-30.md

@@ -113,46 +113,45 @@ description: 九月生活与职业成长随想
 
 ## 将Cursor添加到右键菜单
 
-通过脚本读取当前用户名，并在桌面生成名为`AddCursorToContextMenu.reg`的注册表文件。如权限不足，可在桌面找到该文件，以管理员身份运行。
+通过脚本读取当前用户名，并在桌面生成名为`AddCursorToContextMenu.reg`的注册表文件。可在桌面找到该文件，以管理员身份运行。
 
 ```python
 import os
-import subprocess
 
-# 获取当前用户名
 username = os.getenv('USERNAME')
-
-# 获取当前用户的桌面路径
-desktop_path = os.path.join(os.path.expanduser("~"), 'Desktop')
-
-# 构造 Cursor.exe 的路径
 cursor_path = fr"C:\\Users\\{username}\\AppData\\Local\\Programs\\Cursor\\Cursor.exe"
 
-# 生成注册表内容
+
 reg_content = f"""Windows Registry Editor Version 5.00
 
 [HKEY_CLASSES_ROOT\\*\\shell\\Open with Cursor]
 @="Open with Cursor"
-"Icon"="\"{cursor_path}\""
+"Icon"="{cursor_path}"
+
 [HKEY_CLASSES_ROOT\\*\\shell\\Open with Cursor\\command]
-@="\"{cursor_path}\" \"%1\""
+@="\\"{cursor_path}\\" \\"%1\\""
 
 [HKEY_CLASSES_ROOT\\Directory\\shell\\Open with Cursor]
 @="Open with Cursor"
-"Icon"="\"{cursor_path}\""
+"Icon"="{cursor_path}"
+
 [HKEY_CLASSES_ROOT\\Directory\\shell\\Open with Cursor\\command]
-@="\"{cursor_path}\" \"%1\""
+@="\\"{cursor_path}\\" \\"%1\\""
 
 [HKEY_CLASSES_ROOT\\Directory\\Background\\shell\\Open with Cursor]
-@="Open with Cursor当前文件夹"
-"Icon"="\"{cursor_path}\""
+@="Open with Cursor current folder"
+"Icon"="{cursor_path}"
+
 [HKEY_CLASSES_ROOT\\Directory\\Background\\shell\\Open with Cursor\\command]
-@="\"{cursor_path}\" \"%V\""
+@="\\"{cursor_path}\\" \\"%V\\""
 
-[HKEY_CURRENT_USER\\Software\\Classes\\CLSID\\{{86ca1aa0-34aa-4e8b-a509-50c905bae2a2}}\\InprocServer32]
-@=""
 """
 
+# 获取当前用户的桌面路径
+desktop_path = os.path.join(os.path.expanduser("~"), 'Desktop')
+
+# 构造 Cursor.exe 的路径
+
 # 生成 .reg 文件的完整路径，保存在桌面
 reg_file_path = os.path.join(desktop_path, "AddCursorToContextMenu.reg")
 
@@ -162,11 +161,71 @@ with open(reg_file_path, "w", encoding="utf-8") as file:
 
 print(f".reg 文件已保存到: {reg_file_path}")
 
-# 执行 .reg 文件来导入注册表
-try:
-    subprocess.run(['regedit.exe', '/s', reg_file_path], check=True)
-    print("注册表文件已成功导入。")
-except subprocess.CalledProcessError as e:
-    print(f"导入注册表文件时出错: {e}")
+```
+
+## 导入模块
+
+现在大家都知道可以通过`pip install -r requirements.txt`安装模块，但是有些模块可能安装失败，需要人手动再次安装。不能做到自动化。这里提供一个脚本，可以自动安装失败的模块2次，并保存到`failed_requirements.txt`。
+
+```python showLineNumbers
+import subprocess
+
+def install_requirements(requirements_file, failed_file):
+    failed_packages = []
+
+    # 打开 requirements.txt 并读取模块名称
+    with open(requirements_file, 'r') as file:
+        packages = file.readlines()
+
+    # 遍历每个包名并尝试安装
+    for package in packages:
+        package = package.strip()
+        if package:  # 跳过空行
+            print(f"正在安装: {package}")
+            try:
+                # 使用 subprocess 执行 pip 安装命令
+                result = subprocess.run(
+                    ['pip', 'install', package],
+                    stdout=subprocess.PIPE,
+                    stderr=subprocess.PIPE,
+                    text=True
+                )
+
+                # 检查安装结果，如果返回码不为0则表示安装失败
+                if result.returncode != 0:
+                    print(f"安装失败: {package}")
+                    failed_packages.append(package)
+
+            except Exception as e:
+                print(f"发生错误: {e}")
+                failed_packages.append(package)
+
+    # 将安装失败的模块写入 failed_requirements.txt
+    if failed_packages:
+        with open(failed_file, 'w') as f:
+            f.write("\n".join(failed_packages))
+        print(f"安装失败的模块已保存到 {failed_file}")
+    else:
+        print("所有模块均安装成功")
+def remove_versions(requirements_file, output_file):
+    with open(requirements_file, 'r') as file:
+        lines = file.readlines()
+
+    cleaned_lines = []
+    for line in lines:
+        # 去除版本号部分（==后面的内容）
+        package = line.split('==')[0].strip()
+        if package:  # 跳过空行
+            cleaned_lines.append(package)
+
+    # 将去掉版本号的包名写入新的文件
+    with open(output_file, 'w') as output:
+        output.write("\n".join(cleaned_lines))
+
+    print(f"已将无版本号的依赖项写入 {output_file}")
+
+for i in range(2):
+    install_requirements('requirements.txt', 'failed_requirements.txt')
+    remove_versions('failed_requirements.txt', 'requirements.txt')
 
 ```