v3.2.6 稳定版本备份
This commit is contained in:
huangzhenpc
295
test_compare_reset.py
Normal file
295
test_compare_reset.py
Normal file
@@ -0,0 +1,295 @@
|
||||
import json
|
||||
import logging
|
||||
import os
|
||||
import time
|
||||
import subprocess
|
||||
from pathlib import Path
|
||||
from datetime import datetime
|
||||
from account_switcher import AccountSwitcher
|
||||
|
||||
def get_full_machine_info():
|
||||
"""获取更完整的机器码相关信息"""
|
||||
info = {}
|
||||
|
||||
# 1. 获取 package.json 中的所有信息（检查两个可能的路径）
|
||||
cursor_paths = [
|
||||
Path(os.path.expanduser("~")) / "AppData" / "Local" / "Programs" / "Cursor",
|
||||
Path(os.path.expanduser("~")) / "AppData" / "Local" / "cursor"
|
||||
]
|
||||
|
||||
info["package_json"] = {}
|
||||
for cursor_path in cursor_paths:
|
||||
package_json = cursor_path / "resources" / "app" / "package.json"
|
||||
if package_json.exists():
|
||||
with open(package_json, "r", encoding="utf-8") as f:
|
||||
info["package_json"][str(package_json)] = json.load(f)
|
||||
|
||||
# 2. 获取 storage.json 的内容
|
||||
storage_path = Path(os.getenv('APPDATA')) / "Cursor" / "User" / "globalStorage" / "storage.json"
|
||||
info["storage_json"] = None
|
||||
if storage_path.exists():
|
||||
try:
|
||||
with open(storage_path, "r", encoding="utf-8") as f:
|
||||
info["storage_json"] = json.load(f)
|
||||
except:
|
||||
pass
|
||||
|
||||
# 3. 检查注册表项和其值
|
||||
import winreg
|
||||
registry_paths = {
|
||||
"cryptography": r"SOFTWARE\Microsoft\Cryptography", # 系统 MachineGuid
|
||||
"cursor_shell": r"Software\Classes\Directory\Background\shell\Cursor",
|
||||
"cursor_command": r"Software\Classes\Directory\Background\shell\Cursor\command",
|
||||
"cursor_auth": r"Software\Cursor\Auth",
|
||||
"cursor_updates": r"Software\Cursor\Updates",
|
||||
"cursor_main": r"Software\Cursor"
|
||||
}
|
||||
|
||||
info["registry"] = {}
|
||||
|
||||
# HKEY_LOCAL_MACHINE 项
|
||||
try:
|
||||
key = winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, registry_paths["cryptography"], 0, winreg.KEY_READ | winreg.KEY_WOW64_64KEY)
|
||||
info["registry"]["HKLM_MachineGuid"] = {
|
||||
"exists": True,
|
||||
"value": winreg.QueryValueEx(key, "MachineGuid")[0]
|
||||
}
|
||||
except WindowsError:
|
||||
info["registry"]["HKLM_MachineGuid"] = {
|
||||
"exists": False,
|
||||
"value": None
|
||||
}
|
||||
|
||||
# HKEY_CURRENT_USER 项
|
||||
for name, path in registry_paths.items():
|
||||
if name == "cryptography":
|
||||
continue
|
||||
try:
|
||||
key = winreg.OpenKey(winreg.HKEY_CURRENT_USER, path, 0, winreg.KEY_READ)
|
||||
info["registry"][f"HKCU_{name}"] = {"exists": True, "values": {}}
|
||||
try:
|
||||
i = 0
|
||||
while True:
|
||||
name, value, type = winreg.EnumValue(key, i)
|
||||
info["registry"][f"HKCU_{name}"]["values"][name] = value
|
||||
i += 1
|
||||
except WindowsError:
|
||||
pass
|
||||
except WindowsError:
|
||||
info["registry"][f"HKCU_{name}"] = {"exists": False, "values": {}}
|
||||
|
||||
# 4. 检查关键文件的存在状态
|
||||
paths_to_check = {
|
||||
"storage": storage_path,
|
||||
"storage_backup": Path(os.getenv('APPDATA')) / "Cursor" / "User" / "globalStorage" / "backups",
|
||||
"user_data": Path(os.getenv('LOCALAPPDATA')) / "Cursor" / "User",
|
||||
"global_storage": Path(os.getenv('APPDATA')) / "Cursor" / "User" / "globalStorage",
|
||||
"cache": Path(os.getenv('LOCALAPPDATA')) / "Cursor" / "Cache",
|
||||
"updater": Path(os.getenv('LOCALAPPDATA')) / "cursor-updater"
|
||||
}
|
||||
|
||||
info["files"] = {}
|
||||
for name, path in paths_to_check.items():
|
||||
if path.exists():
|
||||
info["files"][name] = {
|
||||
"exists": True,
|
||||
"is_dir": path.is_dir(),
|
||||
"size": os.path.getsize(path) if path.is_file() else None,
|
||||
"modified_time": datetime.fromtimestamp(os.path.getmtime(path)).isoformat()
|
||||
}
|
||||
else:
|
||||
info["files"][name] = {
|
||||
"exists": False,
|
||||
"is_dir": None,
|
||||
"size": None,
|
||||
"modified_time": None
|
||||
}
|
||||
|
||||
# 5. 获取遥测相关的配置值
|
||||
info["telemetry"] = {
|
||||
"machineId": info["storage_json"].get("telemetry.machineId") if info["storage_json"] else None,
|
||||
"macMachineId": info["storage_json"].get("telemetry.macMachineId") if info["storage_json"] else None,
|
||||
"devDeviceId": info["storage_json"].get("telemetry.devDeviceId") if info["storage_json"] else None,
|
||||
"sqmId": info["storage_json"].get("telemetry.sqmId") if info["storage_json"] else None
|
||||
}
|
||||
|
||||
return info
|
||||
|
||||
def save_info(info, filename):
|
||||
"""保存信息到文件"""
|
||||
with open(filename, "w", encoding="utf-8") as f:
|
||||
json.dump(info, f, indent=2, ensure_ascii=False)
|
||||
|
||||
def compare_info(before_file, after_file):
|
||||
"""比较两个信息文件的差异"""
|
||||
with open(before_file, "r", encoding="utf-8") as f:
|
||||
before = json.load(f)
|
||||
with open(after_file, "r", encoding="utf-8") as f:
|
||||
after = json.load(f)
|
||||
|
||||
differences = {
|
||||
"package_json_changes": {},
|
||||
"registry_changes": {},
|
||||
"file_changes": {}
|
||||
}
|
||||
|
||||
# 比较 package.json 变化
|
||||
if "package_json" in before and "package_json" in after:
|
||||
for key in set(before["package_json"].keys()) | set(after["package_json"].keys()):
|
||||
before_val = before["package_json"].get(key)
|
||||
after_val = after["package_json"].get(key)
|
||||
if before_val != after_val:
|
||||
differences["package_json_changes"][key] = {
|
||||
"before": before_val,
|
||||
"after": after_val
|
||||
}
|
||||
|
||||
# 比较注册表变化
|
||||
for path in set(before["registry"].keys()) | set(after["registry"].keys()):
|
||||
before_reg = before["registry"].get(path, {})
|
||||
after_reg = after["registry"].get(path, {})
|
||||
if before_reg != after_reg:
|
||||
differences["registry_changes"][path] = {
|
||||
"before": before_reg,
|
||||
"after": after_reg
|
||||
}
|
||||
|
||||
# 比较文件变化
|
||||
for name in set(before["files"].keys()) | set(after["files"].keys()):
|
||||
before_file = before["files"].get(name, {})
|
||||
after_file = after["files"].get(name, {})
|
||||
if before_file != after_file:
|
||||
differences["file_changes"][name] = {
|
||||
"before": before_file,
|
||||
"after": after_file
|
||||
}
|
||||
|
||||
return differences
|
||||
|
||||
def main():
|
||||
"""主测试流程"""
|
||||
# 设置日志
|
||||
logging.basicConfig(level=logging.INFO)
|
||||
|
||||
print("选择操作:")
|
||||
print("1. 记录当前状态（在运行 GitHub 脚本前执行）")
|
||||
print("2. 记录重置后状态并对比（在运行 GitHub 脚本后执行）")
|
||||
print("3. 运行我们的重置方法并记录对比")
|
||||
|
||||
choice = input("\n请选择操作 (1/2/3): ").strip()
|
||||
|
||||
if choice == "1":
|
||||
# 记录初始状态
|
||||
print("\n记录初始状态...")
|
||||
initial_info = get_full_machine_info()
|
||||
save_info(initial_info, "before_github_reset.json")
|
||||
print("初始状态已保存到 before_github_reset.json")
|
||||
|
||||
elif choice == "2":
|
||||
# 检查是否存在初始状态文件
|
||||
if not os.path.exists("before_github_reset.json"):
|
||||
print("错误：找不到初始状态文件 before_github_reset.json")
|
||||
print("请先执行选项 1 记录初始状态")
|
||||
return
|
||||
|
||||
# 记录 GitHub 脚本执行后的状态
|
||||
print("\n记录 GitHub 脚本执行后的状态...")
|
||||
after_info = get_full_machine_info()
|
||||
save_info(after_info, "after_github_reset.json")
|
||||
|
||||
# 对比变化
|
||||
print("\n=== GitHub 脚本执行前后的变化 ===")
|
||||
differences = compare_info("before_github_reset.json", "after_github_reset.json")
|
||||
|
||||
# 显示 package.json 的变化
|
||||
if differences["package_json_changes"]:
|
||||
print("\npackage.json 变化:")
|
||||
for key, change in differences["package_json_changes"].items():
|
||||
print(f"- {key}:")
|
||||
print(f" 之前: {change['before']}")
|
||||
print(f" 之后: {change['after']}")
|
||||
|
||||
# 显示注册表的变化
|
||||
if differences["registry_changes"]:
|
||||
print("\n注册表变化:")
|
||||
for path, change in differences["registry_changes"].items():
|
||||
print(f"\n- {path}:")
|
||||
print(f" 之前: {change['before']}")
|
||||
print(f" 之后: {change['after']}")
|
||||
|
||||
# 显示文件的变化
|
||||
if differences["file_changes"]:
|
||||
print("\n文件变化:")
|
||||
for name, change in differences["file_changes"].items():
|
||||
print(f"\n- {name}:")
|
||||
before_status = "存在" if change["before"].get("exists") else "不存在"
|
||||
after_status = "存在" if change["after"].get("exists") else "不存在"
|
||||
print(f" 状态: {before_status} -> {after_status}")
|
||||
if change["before"].get("exists") and change["after"].get("exists"):
|
||||
if change["before"].get("size") != change["after"].get("size"):
|
||||
print(f" 大小: {change['before'].get('size')} -> {change['after'].get('size')}")
|
||||
if change["before"].get("modified_time") != change["after"].get("modified_time"):
|
||||
print(f" 修改时间: {change['before'].get('modified_time')} -> {change['after'].get('modified_time')}")
|
||||
|
||||
# 保存差异到文件
|
||||
save_info(differences, "github_script_changes.json")
|
||||
print("\n详细的变化信息已保存到 github_script_changes.json")
|
||||
|
||||
elif choice == "3":
|
||||
if not os.path.exists("github_script_changes.json"):
|
||||
print("错误：找不到 GitHub 脚本的变化记录文件 github_script_changes.json")
|
||||
print("请先执行选项 1 和 2 来记录 GitHub 脚本的变化")
|
||||
return
|
||||
|
||||
# 记录运行我们的重置方法前的状态
|
||||
print("\n记录当前状态...")
|
||||
before_our_reset = get_full_machine_info()
|
||||
save_info(before_our_reset, "before_our_reset.json")
|
||||
|
||||
# 运行我们的重置方法
|
||||
print("\n运行我们的重置方法...")
|
||||
switcher = AccountSwitcher()
|
||||
original_restart = switcher.restart_cursor
|
||||
try:
|
||||
switcher.restart_cursor = lambda: True
|
||||
switcher.reset_machine_id()
|
||||
finally:
|
||||
switcher.restart_cursor = original_restart
|
||||
|
||||
# 记录重置后的状态
|
||||
print("\n记录重置后的状态...")
|
||||
after_our_reset = get_full_machine_info()
|
||||
save_info(after_our_reset, "after_our_reset.json")
|
||||
|
||||
# 对比我们的方法与 GitHub 脚本的差异
|
||||
print("\n=== 对比我们的重置方法与 GitHub 脚本的差异 ===")
|
||||
with open("github_script_changes.json", "r", encoding="utf-8") as f:
|
||||
github_changes = json.load(f)
|
||||
our_changes = compare_info("before_our_reset.json", "after_our_reset.json")
|
||||
|
||||
# 分析差异
|
||||
print("\n=== 差异分析 ===")
|
||||
print("1. GitHub 脚本改变但我们没有改变的项:")
|
||||
for category in ["package_json_changes", "registry_changes", "file_changes"]:
|
||||
for key in github_changes[category]:
|
||||
if key not in our_changes[category]:
|
||||
print(f"- {category}: {key}")
|
||||
|
||||
print("\n2. 我们改变但 GitHub 脚本没有改变的项:")
|
||||
for category in ["package_json_changes", "registry_changes", "file_changes"]:
|
||||
for key in our_changes[category]:
|
||||
if key not in github_changes[category]:
|
||||
print(f"- {category}: {key}")
|
||||
|
||||
# 保存分析结果
|
||||
save_info({
|
||||
"github_changes": github_changes,
|
||||
"our_changes": our_changes
|
||||
}, "reset_methods_comparison.json")
|
||||
print("\n详细的对比结果已保存到 reset_methods_comparison.json")
|
||||
|
||||
else:
|
||||
print("无效的选择")
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user