feat: 会员激活码验证功能完善 - 修复激活码验证逻辑,正确处理API返回数据;添加新增天数显示;完善会员状态显示;添加历史激活记录显示;优化状态信息的保存和读取

This commit is contained in:
huangzhenpc
2025-02-11 16:06:23 +08:00
parent d28e2f35c8
commit d82928785d
2 changed files with 729 additions and 0 deletions

View File

@@ -0,0 +1,426 @@
import os
import json
import requests
import logging
import subprocess
import uuid
import hashlib
from typing import Optional, Dict, Tuple
from pathlib import Path
from utils.config import Config
def get_hardware_id() -> str:
"""获取硬件唯一标识"""
try:
# 获取CPU信息
cpu_info = subprocess.check_output('wmic cpu get ProcessorId').decode()
cpu_id = cpu_info.split('\n')[1].strip()
# 获取主板序列号
board_info = subprocess.check_output('wmic baseboard get SerialNumber').decode()
board_id = board_info.split('\n')[1].strip()
# 获取BIOS序列号
bios_info = subprocess.check_output('wmic bios get SerialNumber').decode()
bios_id = bios_info.split('\n')[1].strip()
# 组合信息并生成哈希
combined = f"{cpu_id}:{board_id}:{bios_id}"
return hashlib.md5(combined.encode()).hexdigest()
except Exception as e:
logging.error(f"获取硬件ID失败: {str(e)}")
# 如果获取失败使用UUID作为备选方案
return str(uuid.uuid4())
class CursorAuthManager:
def __init__(self):
self.cursor_path = Path(os.path.expanduser("~")) / "AppData" / "Local" / "Programs" / "Cursor"
self.app_path = self.cursor_path / "resources" / "app"
self.package_json = self.app_path / "package.json"
def update_auth(self, email: str, access_token: str, refresh_token: str) -> bool:
"""更新Cursor认证信息"""
try:
# 读取package.json
with open(self.package_json, "r", encoding="utf-8") as f:
data = json.load(f)
# 更新认证信息
data["email"] = email
data["accessToken"] = access_token
data["refreshToken"] = refresh_token
# 保存更新后的文件
with open(self.package_json, "w", encoding="utf-8") as f:
json.dump(data, f, indent=2)
logging.info(f"认证信息更新成功: {email}")
return True
except Exception as e:
logging.error(f"更新认证信息失败: {str(e)}")
return False
class AccountSwitcher:
def __init__(self):
self.cursor_path = Path(os.path.expanduser("~")) / "AppData" / "Local" / "Programs" / "Cursor"
self.app_path = self.cursor_path / "resources" / "app"
self.package_json = self.app_path / "package.json"
self.auth_manager = CursorAuthManager()
self.config = Config()
self.hardware_id = get_hardware_id()
def get_device_info(self) -> dict:
"""获取设备信息"""
try:
import platform
import socket
import requests
# 获取操作系统信息
os_info = f"{platform.system()} {platform.version()}"
# 获取设备名称
device_name = platform.node()
# 获取地理位置(可选)
try:
ip_info = requests.get('https://ipapi.co/json/', timeout=5).json()
location = f"{ip_info.get('country_name', '')}-{ip_info.get('region', '')}-{ip_info.get('city', '')}"
except:
location = ""
return {
"os": os_info,
"device_name": device_name,
"location": location
}
except Exception as e:
logging.error(f"获取设备信息失败: {str(e)}")
return {
"os": "Windows 10",
"device_name": "未知设备",
"location": ""
}
def check_activation_code(self, code: str) -> Tuple[bool, str, Optional[Dict]]:
"""验证激活码
Returns:
Tuple[bool, str, Optional[Dict]]: (是否成功, 提示消息, 账号信息)
"""
try:
# 获取当前状态
member_info = self.config.load_member_info()
# 分割多个激活码
codes = [c.strip() for c in code.split(',')]
success_codes = []
failed_codes = []
activation_results = []
# 获取设备信息
device_info = self.get_device_info()
# 逐个验证激活码
for single_code in codes:
if not single_code:
continue
# 验证激活码
endpoint = "https://cursorapi.nosqli.com/admin/api.member/activate"
data = {
"code": single_code,
"machine_id": self.hardware_id,
"os": device_info["os"],
"device_name": device_info["device_name"],
"location": device_info["location"]
}
headers = {
"Content-Type": "application/json"
}
try:
response = requests.post(
endpoint,
json=data,
headers=headers,
timeout=10
)
response_data = response.json()
if response_data.get("code") == 200:
result_data = response_data.get("data", {})
logging.info(f"激活码 {single_code} 验证成功: {response_data.get('msg', '')}")
activation_results.append(result_data)
success_codes.append(single_code)
elif response_data.get("code") == 400:
error_msg = response_data.get("msg", "参数错误")
if "已被使用" in error_msg or "已激活" in error_msg:
logging.warning(f"激活码 {single_code} 已被使用")
failed_codes.append(f"{single_code} (已被使用)")
else:
logging.error(f"激活码 {single_code} 验证失败: {error_msg}")
failed_codes.append(f"{single_code} ({error_msg})")
elif response_data.get("code") == 500:
error_msg = response_data.get("msg", "系统错误")
logging.error(f"激活码 {single_code} 验证失败: {error_msg}")
failed_codes.append(f"{single_code} ({error_msg})")
else:
error_msg = response_data.get("msg", "未知错误")
logging.error(f"激活码 {single_code} 验证失败: {error_msg}")
failed_codes.append(f"{single_code} ({error_msg})")
except requests.RequestException as e:
logging.error(f"激活码 {single_code} 请求失败: {str(e)}")
failed_codes.append(f"{single_code} (网络请求失败)")
except Exception as e:
logging.error(f"激活码 {single_code} 处理失败: {str(e)}")
failed_codes.append(f"{single_code} (处理失败)")
if not success_codes:
failed_msg = "\n".join(failed_codes)
return False, f"激活失败:\n{failed_msg}", None
try:
# 使用最后一次激活的结果作为最终状态
final_result = activation_results[-1]
# 保存会员信息
member_info = {
"hardware_id": final_result.get("machine_id", self.hardware_id),
"expire_time": final_result.get("expire_time", ""),
"days_left": final_result.get("days_left", 0), # 使用days_left
"total_days": final_result.get("total_days", 0), # 使用total_days
"status": final_result.get("status", "inactive"),
"device_info": final_result.get("device_info", device_info),
"activation_time": final_result.get("activation_time", ""),
"activation_records": final_result.get("activation_records", []) # 保存激活记录
}
self.config.save_member_info(member_info)
# 生成结果消息
message = f"激活成功\n"
# 显示每个成功激活码的信息
for i, result in enumerate(activation_results, 1):
message += f"\n{i}个激活码:\n"
message += f"- 新增天数: {result.get('added_days', 0)}\n" # 使用added_days显示本次新增天数
# 格式化时间显示
activation_time = result.get('activation_time', '')
if activation_time:
try:
from datetime import datetime
dt = datetime.strptime(activation_time, "%Y-%m-%d %H:%M:%S")
activation_time = dt.strftime("%Y-%m-%d %H:%M:%S")
except:
pass
message += f"- 激活时间: {activation_time}\n"
message += f"\n最终状态:"
message += f"\n- 总天数: {final_result.get('total_days', 0)}" # 累计总天数
message += f"\n- 剩余天数: {final_result.get('days_left', 0)}" # 剩余天数
# 格式化到期时间显示
expire_time = final_result.get('expire_time', '')
if expire_time:
try:
dt = datetime.strptime(expire_time, "%Y-%m-%d %H:%M:%S")
expire_time = dt.strftime("%Y-%m-%d %H:%M:%S")
except:
pass
message += f"\n- 到期时间: {expire_time}"
# 显示激活记录历史
message += "\n\n历史激活记录:"
for record in final_result.get('activation_records', []):
activation_time = record.get('activation_time', '')
if activation_time:
try:
dt = datetime.strptime(activation_time, "%Y-%m-%d %H:%M:%S")
activation_time = dt.strftime("%Y-%m-%d %H:%M:%S")
except:
pass
message += f"\n- 激活码: {record.get('code', '')}"
message += f"\n 天数: {record.get('days', 0)}"
message += f"\n 时间: {activation_time}\n"
if failed_codes:
message += f"\n\n以下激活码验证失败:\n" + "\n".join(failed_codes)
return True, message, member_info
except Exception as e:
logging.error(f"处理激活结果时出错: {str(e)}")
return False, f"处理激活结果失败: {str(e)}", None
except Exception as e:
logging.error(f"激活码验证过程出错: {str(e)}")
return False, f"激活失败: {str(e)}", None
def reset_machine_id(self) -> bool:
"""重置机器码"""
try:
# 读取package.json
with open(self.package_json, "r", encoding="utf-8") as f:
data = json.load(f)
# 删除machineId
if "machineId" in data:
del data["machineId"]
# 保存文件
with open(self.package_json, "w", encoding="utf-8") as f:
json.dump(data, f, indent=2)
logging.info("机器码重置完成")
return True
except Exception as e:
logging.error(f"重置机器码失败: {str(e)}")
return False
def activate_and_switch(self, activation_code: str) -> Tuple[bool, str]:
"""激活并切换账号
Returns:
Tuple[bool, str]: (是否成功, 提示消息)
"""
try:
# 验证激活码
success, message, account_info = self.check_activation_code(activation_code)
return success, message
except Exception as e:
logging.error(f"激活过程出错: {str(e)}")
return False, f"激活失败: {str(e)}"
def get_member_status(self) -> Optional[Dict]:
"""获取会员状态
Returns:
Optional[Dict]: 会员状态信息
"""
try:
# 读取保存的会员信息
member_info = self.config.load_member_info()
# 构造状态检查请求
endpoint = "https://cursorapi.nosqli.com/admin/api.member/status"
data = {
"machine_id": self.hardware_id
}
headers = {
"Content-Type": "application/json"
}
try:
response = requests.post(endpoint, json=data, headers=headers, timeout=10)
response_data = response.json()
if response_data.get("code") == 200:
# 正常状态
data = response_data.get("data", {})
status = data.get("status", "inactive")
# 构造会员信息
member_info = {
"hardware_id": data.get("machine_id", self.hardware_id),
"expire_time": data.get("expire_time", ""),
"days_left": data.get("days_left", 0), # 使用days_left
"total_days": data.get("total_days", 0), # 使用total_days
"status": status,
"activation_records": data.get("activation_records", []) # 保存激活记录
}
# 打印调试信息
logging.info(f"API返回数据: {data}")
logging.info(f"处理后的会员信息: {member_info}")
self.config.save_member_info(member_info)
return member_info
elif response_data.get("code") == 401:
# 未激活或已过期
logging.warning("会员未激活或已过期")
return self._get_inactive_status()
elif response_data.get("code") == 400:
# 参数错误
error_msg = response_data.get("msg", "参数错误")
logging.error(f"获取会员状态失败: {error_msg}")
return self._get_inactive_status()
elif response_data.get("code") == 500:
# 系统错误
error_msg = response_data.get("msg", "系统错误")
logging.error(f"获取会员状态失败: {error_msg}")
return self._get_inactive_status()
else:
# 其他未知错误
error_msg = response_data.get("msg", "未知错误")
logging.error(f"获取会员状态失败: {error_msg}")
return self._get_inactive_status()
except requests.RequestException as e:
logging.error(f"请求会员状态失败: {str(e)}")
return self._get_inactive_status()
except Exception as e:
logging.error(f"获取会员状态出错: {str(e)}")
return self._get_inactive_status()
def _get_inactive_status(self) -> Dict:
"""获取未激活状态的默认信息"""
return {
"hardware_id": self.hardware_id,
"expire_time": "",
"days": 0,
"total_days": 0,
"status": "inactive",
"last_activation": {},
"activation_records": []
}
def main():
"""主函数"""
try:
switcher = AccountSwitcher()
print("\n=== Cursor账号切换工具 ===")
print("1. 激活并切换账号")
print("2. 仅重置机器码")
while True:
try:
choice = int(input("\n请选择操作 (1 或 2): ").strip())
if choice in [1, 2]:
break
else:
print("无效的选项,请重新输入")
except ValueError:
print("请输入有效的数字")
if choice == 1:
activation_code = input("请输入激活码: ").strip()
if switcher.activate_and_switch(activation_code):
print("\n账号激活成功!")
else:
print("\n账号激活失败,请查看日志了解详细信息")
else:
if switcher.reset_machine_id():
print("\n机器码重置成功!")
else:
print("\n机器码重置失败,请查看日志了解详细信息")
except Exception as e:
logging.error(f"程序执行出错: {str(e)}")
print("\n程序执行出错,请查看日志了解详细信息")
finally:
input("\n按回车键退出...")
if __name__ == "__main__":
main()

View File

@@ -0,0 +1,303 @@
import sys
import tkinter as tk
from tkinter import ttk, messagebox
from pathlib import Path
import logging
import os
from PIL import Image, ImageTk
sys.path.append(str(Path(__file__).parent.parent))
from utils.config import Config
from account_switcher import AccountSwitcher
class MainWindow:
def __init__(self):
self.config = Config()
self.switcher = AccountSwitcher()
self.root = tk.Tk()
self.root.title("听泉Cursor助手 v2.2.2 (本机Cursor版本: 0.45.11)")
self.root.geometry("600x500") # 调整窗口大小
self.root.resizable(True, True) # 允许调整窗口大小
# 设置窗口最小尺寸
self.root.minsize(600, 500)
try:
# 设置图标 - 使用PIL
current_dir = os.path.dirname(os.path.abspath(__file__))
icon_path = os.path.join(os.path.dirname(current_dir), "icon", "th.jpg")
if os.path.exists(icon_path):
# 打开并调整图片大小
img = Image.open(icon_path)
img = img.resize((32, 32), Image.Resampling.LANCZOS)
# 转换为PhotoImage
self.icon = ImageTk.PhotoImage(img)
self.root.iconphoto(True, self.icon)
logging.info(f"成功加载图标: {icon_path}")
else:
logging.error(f"图标文件不存在: {icon_path}")
except Exception as e:
logging.error(f"设置图标失败: {str(e)}")
# 设置关闭窗口处理
self.root.protocol("WM_DELETE_WINDOW", self.on_closing)
# 初始化状态变量
self.status_var = tk.StringVar(value="")
# 设置样式
self.style = ttk.Style()
self.style.configure("TButton", padding=5, font=("Microsoft YaHei UI", 9))
self.style.configure("TLabelframe", padding=10, font=("Microsoft YaHei UI", 9))
self.style.configure("TLabel", padding=2, font=("Microsoft YaHei UI", 9))
self.style.configure("Custom.TButton", padding=10, font=("Microsoft YaHei UI", 9))
self.style.configure("Action.TButton", padding=8, font=("Microsoft YaHei UI", 9))
self.setup_ui()
# 启动时检查一次状态
self.check_status()
def setup_ui(self):
"""设置UI界面"""
# 主框架
main_frame = ttk.Frame(self.root, padding=10)
main_frame.pack(fill="both", expand=True)
# 功能菜单
menu_frame = ttk.Frame(main_frame)
menu_frame.pack(fill="x", pady=(0, 10))
ttk.Label(menu_frame, text="功能(F)").pack(side="left")
# 设备ID区域
device_frame = ttk.Frame(main_frame)
device_frame.pack(fill="x", pady=(0, 10))
ttk.Label(device_frame, text="设备识别码(勿动):").pack(side="left")
self.hardware_id_var = tk.StringVar(value=self.switcher.hardware_id)
device_id_entry = ttk.Entry(device_frame, textvariable=self.hardware_id_var, width=35, state="readonly")
device_id_entry.pack(side="left", padx=5)
copy_btn = ttk.Button(device_frame, text="复制ID", command=self.copy_device_id, width=8)
copy_btn.pack(side="left")
# 会员状态区域
status_frame = ttk.LabelFrame(main_frame, text="会员状态")
status_frame.pack(fill="x", pady=(0, 10))
self.status_text = tk.Text(status_frame, height=5, width=40, font=("Microsoft YaHei UI", 9))
self.status_text.pack(fill="both", padx=5, pady=5)
self.status_text.config(state="disabled")
# 激活区域
activation_frame = ttk.LabelFrame(main_frame, text="激活(叠加)会员,多个激活码可叠加整体时长")
activation_frame.pack(fill="x", pady=(0, 10))
input_frame = ttk.Frame(activation_frame)
input_frame.pack(fill="x", padx=5, pady=5)
ttk.Label(input_frame, text="激活码:").pack(side="left")
self.activation_var = tk.StringVar()
activation_entry = ttk.Entry(input_frame, textvariable=self.activation_var, width=35)
activation_entry.pack(side="left", padx=5)
activate_btn = ttk.Button(input_frame, text="激活", command=self.activate_account, width=8)
activate_btn.pack(side="left")
# 操作按钮区域
btn_frame = ttk.Frame(main_frame)
btn_frame.pack(fill="x", pady=5)
self.style.configure("Action.TButton", padding=8)
ttk.Button(btn_frame, text="刷新Cursor授权", command=self.reset_machine_id, style="Action.TButton").pack(fill="x", pady=2)
ttk.Button(btn_frame, text="实现Cursor0.45.x限制", command=self.dummy_function, style="Action.TButton").pack(fill="x", pady=2)
ttk.Button(btn_frame, text="禁用Cursor版本更新", command=self.dummy_function, style="Action.TButton").pack(fill="x", pady=2)
def copy_device_id(self):
"""复制设备ID到剪贴板"""
# 先检查状态
if not self.check_status():
return
self.root.clipboard_clear()
self.root.clipboard_append(self.hardware_id_var.get())
def activate_account(self):
"""激活账号"""
code = self.activation_var.get().strip()
if not code:
messagebox.showwarning("提示", "请输入激活码")
return
self.status_var.set("正在激活...")
self.root.update()
try:
success, message, account_info = self.switcher.check_activation_code(code)
if success:
# 更新会员信息显示
self.update_status_display(account_info)
messagebox.showinfo("激活成功", "激活成功!\n" + message)
self.status_var.set("激活成功")
# 清空激活码输入框
self.activation_var.set("")
else:
messagebox.showerror("激活失败", message)
self.status_var.set("激活失败")
# 激活后检查一次状态
self.check_status()
except Exception as e:
messagebox.showerror("错误", f"激活失败: {str(e)}")
self.status_var.set("发生错误")
# 出错后也检查状态
self.check_status()
def update_status_display(self, status_info: dict):
"""更新状态显示"""
# 打印API返回的原始数据
logging.info("=== API返回数据 ===")
logging.info(f"状态信息: {status_info}")
if 'activation_records' in status_info:
logging.info("激活记录:")
for record in status_info['activation_records']:
logging.info(f"- 记录: {record}")
# 启用文本框编辑
self.status_text.config(state="normal")
# 清空当前内容
self.status_text.delete(1.0, tk.END)
# 更新状态文本
status_map = {
"active": "正常",
"inactive": "未激活",
"expired": "已过期"
}
status_text = status_map.get(status_info.get('status', 'inactive'), "未知")
# 构建状态文本
status_lines = [
f"会员状态:{status_text}",
f"到期时间:{status_info.get('expire_time', '')}",
f"总天数:{status_info.get('total_days', 0)}",
f"剩余天数:{status_info.get('days_left', 0)}"
]
# 如果有激活记录,显示最近一次激活信息
activation_records = status_info.get('activation_records', [])
if activation_records:
latest_record = activation_records[-1] # 获取最新的激活记录
device_info = latest_record.get('device_info', {})
status_lines.extend([
"",
"最近激活信息:",
f"激活码:{latest_record.get('code', '')}",
f"激活时间:{latest_record.get('activation_time', '')}",
f"增加天数:{latest_record.get('days', 0)}",
"",
"设备信息:",
f"系统:{device_info.get('os', '')}",
f"设备名:{device_info.get('device_name', '')}",
f"IP地址{device_info.get('ip', '')}",
f"地区:{device_info.get('location', '--')}"
])
# 写入状态信息
self.status_text.insert(tk.END, "\n".join(status_lines))
# 禁用文本框编辑
self.status_text.config(state="disabled")
def check_status(self):
"""检查会员状态
Returns:
bool: True 表示激活状态正常False 表示未激活或已过期
"""
try:
self.status_var.set("正在检查状态...")
self.root.update()
status = self.switcher.get_member_status()
if status:
self.update_status_display(status)
if status.get('status') == 'inactive':
messagebox.showwarning("警告", "当前设备未激活或激活已失效")
self.status_var.set("未激活")
return False
self.status_var.set("状态检查完成")
return True
else:
# 更新为未激活状态
inactive_status = {
"hardware_id": self.switcher.hardware_id,
"expire_time": "",
"days": 0,
"total_days": 0,
"status": "inactive",
"last_activation": {}
}
self.update_status_display(inactive_status)
self.status_var.set("未激活")
messagebox.showwarning("警告", "当前设备未激活")
return False
except Exception as e:
logging.error(f"检查状态失败: {str(e)}")
self.status_var.set("状态检查失败")
# 显示错误消息
messagebox.showerror("错误", f"检查状态失败: {str(e)}")
return False
def minimize_window(self):
"""最小化窗口"""
self.root.iconify()
def maximize_window(self):
"""最大化/还原窗口"""
if self.root.state() == 'zoomed':
self.root.state('normal')
else:
self.root.state('zoomed')
def on_closing(self):
"""窗口关闭处理"""
try:
logging.info("正在关闭程序...")
self.root.quit()
except Exception as e:
logging.error(f"关闭程序时出错: {str(e)}")
finally:
self.root.destroy()
def run(self):
"""运行程序"""
self.root.mainloop()
def dummy_function(self):
"""占位函数"""
# 先检查状态
if not self.check_status():
return
messagebox.showinfo("提示", "此功能暂未实现")
def reset_machine_id(self):
"""重置机器码"""
# 先检查状态
if not self.check_status():
return
try:
if self.switcher.reset_machine_id():
messagebox.showinfo("成功", "机器码重置成功")
# 重置后检查一次状态
self.check_status()
else:
messagebox.showerror("错误", "机器码重置失败,请查看日志")
# 失败后也检查状态
self.check_status()
except Exception as e:
messagebox.showerror("错误", f"重置失败: {str(e)}")
# 出错后也检查状态
self.check_status()