72 lines
2.7 KiB
Python
72 lines
2.7 KiB
Python
import os
|
|
import winreg
|
|
import logging
|
|
from pathlib import Path
|
|
|
|
class CursorRegistry:
|
|
"""Cursor注册表操作工具类"""
|
|
|
|
def __init__(self):
|
|
self.cursor_path = Path(os.path.expanduser("~")) / "AppData" / "Local" / "Programs" / "Cursor"
|
|
self.app_path = self.cursor_path / "resources" / "app"
|
|
|
|
def refresh_registry(self) -> bool:
|
|
"""刷新Cursor相关的注册表项
|
|
|
|
Returns:
|
|
bool: 是否成功
|
|
"""
|
|
try:
|
|
# 获取Cursor安装路径
|
|
cursor_exe = self.cursor_path / "Cursor.exe"
|
|
if not cursor_exe.exists():
|
|
logging.error("未找到Cursor.exe")
|
|
return False
|
|
|
|
# 刷新注册表项
|
|
try:
|
|
# 打开HKEY_CURRENT_USER\Software\Classes\Directory\Background\shell\Cursor
|
|
key_path = r"Software\Classes\Directory\Background\shell\Cursor"
|
|
with winreg.OpenKey(winreg.HKEY_CURRENT_USER, key_path, 0, winreg.KEY_WRITE) as key:
|
|
winreg.SetValueEx(key, "Icon", 0, winreg.REG_SZ, str(cursor_exe))
|
|
|
|
# 打开command子键
|
|
with winreg.OpenKey(winreg.HKEY_CURRENT_USER, key_path + r"\command", 0, winreg.KEY_WRITE) as key:
|
|
winreg.SetValueEx(key, "", 0, winreg.REG_SZ, f'"{str(cursor_exe)}" "%V"')
|
|
|
|
logging.info("注册表刷新成功")
|
|
return True
|
|
|
|
except WindowsError as e:
|
|
logging.error(f"刷新注册表失败: {str(e)}")
|
|
return False
|
|
|
|
except Exception as e:
|
|
logging.error(f"刷新注册表过程出错: {str(e)}")
|
|
return False
|
|
|
|
def clean_registry(self) -> bool:
|
|
"""清理Cursor相关的注册表项
|
|
|
|
Returns:
|
|
bool: 是否成功
|
|
"""
|
|
try:
|
|
# 删除注册表项
|
|
try:
|
|
key_path = r"Software\Classes\Directory\Background\shell\Cursor"
|
|
winreg.DeleteKey(winreg.HKEY_CURRENT_USER, key_path + r"\command")
|
|
winreg.DeleteKey(winreg.HKEY_CURRENT_USER, key_path)
|
|
logging.info("注册表清理成功")
|
|
return True
|
|
|
|
except WindowsError as e:
|
|
if e.winerror == 2: # 找不到注册表项
|
|
logging.info("注册表项不存在，无需清理")
|
|
return True
|
|
logging.error(f"清理注册表失败: {str(e)}")
|
|
return False
|
|
|
|
except Exception as e:
|
|
logging.error(f"清理注册表过程出错: {str(e)}")
|
|
return False |