first commit: 初始化项目,添加基本功能和打包脚本
Some checks failed
Remove old artifacts / remove-old-artifacts (push) Has been cancelled

This commit is contained in:
huangzhenpc
2025-02-17 18:58:01 +08:00
parent 351b12a039
commit 96604f7139
9 changed files with 1015 additions and 1 deletions

View File

@@ -1,3 +1,69 @@
# 听泉助手
一个用于管理Cursor编辑器授权的跨平台桌面应用。
## 功能特点
- 设备ID管理
- 会员状态显示
- 激活码管理
- Cursor编辑器授权刷新
- 版本限制实现
- 更新控制
## 开发环境要求
- Python 3.9+
- PyQt5
- 其他依赖见 requirements.txt
## 安装依赖
```bash
# 创建虚拟环境
python -m venv venv
# 激活虚拟环境
# Windows:
venv\Scripts\activate
# Mac/Linux:
source venv/bin/activate
# 安装依赖
pip install -r requirements.txt
```
## 打包说明
### Windows版本
```bash
# 运行打包脚本
build_win.bat
```
### Mac版本
```bash
# 添加执行权限
chmod +x build_mac.command
# 运行打包脚本
./build_mac.command
```
## 项目结构
```
.
├── README.md
├── requirements.txt
├── build_mac.command # Mac打包脚本
├── build_win.bat # Windows打包脚本
├── gui/
│ └── main_window.py # 主窗口界面
├── logger.py # 日志模块
├── update_cursor_token.py # 授权更新模块
└── ... # 其他模块
```
# Cursor Pro 自动化工具使用说明

61
build_mac.command Normal file
View File

@@ -0,0 +1,61 @@
#!/bin/bash
# 获取脚本所在目录
cd "$(dirname "$0")"
# 检查虚拟环境
if [ -f "venv/bin/activate" ]; then
echo "激活虚拟环境..."
source venv/bin/activate
else
echo "警告: 未找到虚拟环境,创建新的虚拟环境..."
python3 -m venv venv
source venv/bin/activate
echo "安装依赖..."
pip3 install --upgrade pip
pip3 install pyinstaller
pip3 install PyQt5
pip3 install requests
pip3 install urllib3
fi
# 确保依赖已安装
echo "检查依赖..."
pip3 list | grep -E "pyinstaller|PyQt5|requests|urllib3" || {
echo "安装缺失的依赖..."
pip3 install pyinstaller PyQt5 requests urllib3
}
# 执行打包
echo "开始打包..."
python3 -m PyInstaller \
--name="听泉助手" \
--windowed \
--clean \
--noconfirm \
--add-data="logger.py:." \
--add-data="update_cursor_token.py:." \
--add-data="cursor_auth_manager.py:." \
--add-data="reset_machine.py:." \
--add-data="patch_cursor_get_machine_id.py:." \
--add-data="exit_cursor.py:." \
--add-data="go_cursor_help.py:." \
--add-data="logo.py:." \
--hidden-import=PyQt5 \
--hidden-import=requests \
--hidden-import=urllib3 \
--target-architecture=universal2 \
--codesign-identity=- \
--osx-bundle-identifier=com.tingquan.helper \
gui/main_window.py
# 检查打包结果
if [ -d "dist/听泉助手.app" ]; then
echo "打包成功!应用程序包已生成在 dist/听泉助手.app"
else
echo "打包失败,请检查错误信息"
fi
# 退出虚拟环境
deactivate

35
build_mac.py Normal file
View File

@@ -0,0 +1,35 @@
import PyInstaller.__main__
import os
import sys
# 获取当前脚本所在目录
current_dir = os.path.dirname(os.path.abspath(__file__))
# 打包参数
params = [
'gui/main_window.py', # 主程序入口
'--name=听泉助手', # 应用名称
'--onefile', # 打包成单个文件
'--windowed', # 不显示控制台窗口
'--clean', # 清理临时文件
'--noconfirm', # 不确认覆盖
f'--distpath={os.path.join(current_dir, "dist")}', # 输出目录
f'--workpath={os.path.join(current_dir, "build")}', # 工作目录
f'--specpath={current_dir}', # spec文件目录
'--add-data=logger.py;.', # 添加额外文件
'--add-data=update_cursor_token.py;.',
'--add-data=cursor_auth_manager.py;.',
'--add-data=reset_machine.py;.',
'--add-data=patch_cursor_get_machine_id.py;.',
'--add-data=exit_cursor.py;.',
'--add-data=go_cursor_help.py;.',
'--add-data=logo.py;.',
'--hidden-import=PyQt5',
'--hidden-import=requests',
'--hidden-import=urllib3',
]
# 执行打包
PyInstaller.__main__.run(params)
print("打包完成可执行文件已生成在dist目录下。")

17
build_mac_docker.bat Normal file
View File

@@ -0,0 +1,17 @@
@echo off
setlocal
echo 构建Docker镜像...
docker build -t tingquan-mac-builder .
echo 运行Docker容器进行打包...
docker run --rm -v "%cd%":/app tingquan-mac-builder
echo 检查打包结果...
if exist "dist\听泉助手.app" (
echo 打包成功Mac应用程序包已生成在 dist\听泉助手.app
) else (
echo 打包失败,请检查错误信息
)
endlocal

54
build_win.bat Normal file
View File

@@ -0,0 +1,54 @@
@echo off
setlocal
:: 获取脚本所在目录
cd /d "%~dp0"
:: 检查虚拟环境
if exist "venv\Scripts\activate.bat" (
echo Activating virtual environment...
call venv\Scripts\activate.bat
) else (
echo Creating new virtual environment...
python -m venv venv
call venv\Scripts\activate.bat
echo Installing dependencies...
python -m pip install --upgrade pip
pip install pyinstaller
pip install PyQt5
pip install requests
pip install urllib3
)
:: 执行打包
echo Starting build process...
python -m PyInstaller ^
--name="TingquanHelper" ^
--onefile ^
--windowed ^
--clean ^
--noconfirm ^
--add-data="logger.py;." ^
--add-data="update_cursor_token.py;." ^
--add-data="cursor_auth_manager.py;." ^
--add-data="reset_machine.py;." ^
--add-data="patch_cursor_get_machine_id.py;." ^
--add-data="exit_cursor.py;." ^
--add-data="go_cursor_help.py;." ^
--add-data="logo.py;." ^
--hidden-import=PyQt5 ^
--hidden-import=requests ^
--hidden-import=urllib3 ^
gui/main_window.py
:: 检查打包结果
if exist "dist\TingquanHelper.exe" (
echo Build successful! Executable created at dist\TingquanHelper.exe
) else (
echo Build failed. Please check the error messages.
)
:: 退出虚拟环境
deactivate
endlocal

35
build_win.py Normal file
View File

@@ -0,0 +1,35 @@
import PyInstaller.__main__
import os
import sys
# 获取当前脚本所在目录
current_dir = os.path.dirname(os.path.abspath(__file__))
# 打包参数
params = [
'gui/main_window.py', # 主程序入口
'--name=听泉助手', # 应用名称
'--onefile', # 打包成单个文件
'--windowed', # 不显示控制台窗口
'--clean', # 清理临时文件
'--noconfirm', # 不确认覆盖
f'--distpath={os.path.join(current_dir, "dist")}', # 输出目录
f'--workpath={os.path.join(current_dir, "build")}', # 工作目录
f'--specpath={current_dir}', # spec文件目录
'--add-data=logger.py;.', # 添加额外文件
'--add-data=update_cursor_token.py;.',
'--add-data=cursor_auth_manager.py;.',
'--add-data=reset_machine.py;.',
'--add-data=patch_cursor_get_machine_id.py;.',
'--add-data=exit_cursor.py;.',
'--add-data=go_cursor_help.py;.',
'--add-data=logo.py;.',
'--hidden-import=PyQt5',
'--hidden-import=requests',
'--hidden-import=urllib3',
]
# 执行打包
PyInstaller.__main__.run(params)
print("打包完成可执行文件已生成在dist目录下。")

405
gui/main_window.py Normal file
View File

@@ -0,0 +1,405 @@
import sys
import os
# 添加父目录到系统路径
current_dir = os.path.dirname(os.path.abspath(__file__))
parent_dir = os.path.dirname(current_dir)
sys.path.append(parent_dir)
from PyQt5.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout,
QPushButton, QLabel, QLineEdit, QTextEdit, QMessageBox,
QHBoxLayout, QFrame)
from PyQt5.QtCore import Qt, QThread, pyqtSignal
from PyQt5.QtGui import QFont, QIcon
from update_cursor_token import CursorTokenUpdater
from logger import logging
class UpdateWorker(QThread):
"""后台更新线程"""
finished = pyqtSignal(bool, str)
progress = pyqtSignal(str)
def __init__(self, updater):
super().__init__()
self.updater = updater
def run(self):
try:
success = self.updater.full_update_process()
if success:
self.finished.emit(True, "更新成功!")
else:
self.finished.emit(False, "更新失败,请查看日志")
except Exception as e:
self.finished.emit(False, f"发生错误: {str(e)}")
class MainWindow(QMainWindow):
def __init__(self):
super().__init__()
try:
logging.info("正在初始化主窗口...")
self.updater = CursorTokenUpdater()
self.init_ui()
logging.info("主窗口初始化完成")
except Exception as e:
logging.error(f"初始化主窗口时发生错误: {str(e)}")
QMessageBox.critical(self, "错误", f"初始化失败: {str(e)}")
def init_ui(self):
# 设置窗口基本属性
self.setWindowTitle('听泉助手')
self.setMinimumSize(600, 500)
self.setStyleSheet("""
QMainWindow {
background-color: #f5f5f7;
}
QPushButton {
background-color: #0066cc;
color: white;
border: none;
border-radius: 6px;
padding: 10px;
font-size: 14px;
min-width: 120px;
}
QPushButton:hover {
background-color: #0052a3;
}
QPushButton:pressed {
background-color: #003d7a;
}
QLabel {
font-size: 14px;
color: #333333;
}
QTextEdit {
border: 1px solid #cccccc;
border-radius: 6px;
padding: 10px;
background-color: white;
font-family: Consolas, Monaco, monospace;
}
QLineEdit {
border: 1px solid #cccccc;
border-radius: 6px;
padding: 8px;
background-color: white;
}
QMessageBox {
background-color: #f5f5f7;
}
QMessageBox QPushButton {
min-width: 80px;
padding: 5px 15px;
}
""")
try:
# 创建主窗口部件
central_widget = QWidget()
self.setCentralWidget(central_widget)
layout = QVBoxLayout(central_widget)
layout.setSpacing(20)
layout.setContentsMargins(30, 30, 30, 30)
# 设备ID显示区域
device_frame = QFrame()
device_frame.setStyleSheet("""
QFrame {
background-color: white;
border-radius: 10px;
padding: 10px;
}
""")
device_layout = QHBoxLayout(device_frame)
device_layout.setContentsMargins(10, 5, 10, 5)
device_id_label = QLabel("设备标识:")
self.device_id_text = QLineEdit()
self.device_id_text.setReadOnly(True)
try:
hardware_id = self.updater.hardware_id
logging.info(f"获取到硬件ID: {hardware_id}")
self.device_id_text.setText(hardware_id)
except Exception as e:
logging.error(f"获取硬件ID失败: {str(e)}")
self.device_id_text.setText("获取失败")
copy_button = QPushButton("复制ID")
copy_button.setMaximumWidth(80)
copy_button.clicked.connect(self.copy_device_id)
device_layout.addWidget(device_id_label)
device_layout.addWidget(self.device_id_text)
device_layout.addWidget(copy_button)
layout.addWidget(device_frame)
# 会员状态区域
member_frame = QFrame()
member_frame.setStyleSheet("""
QFrame {
background-color: white;
border-radius: 10px;
padding: 20px;
}
""")
member_layout = QVBoxLayout(member_frame)
member_layout.setContentsMargins(15, 10, 15, 10)
member_layout.setSpacing(8)
# 会员信息显示区域
self.member_info = QTextEdit()
self.member_info.setReadOnly(True)
self.member_info.setFixedHeight(100) # 增加高度
self.member_info.setStyleSheet("""
QTextEdit {
border: none;
background-color: transparent;
font-family: Consolas, Monaco, monospace;
font-size: 14px;
line-height: 1.6;
}
""")
# 设置默认会员信息(简化版)
self.member_info.setText("会员状态: 正常\n到期时间: 2025-02-22 20:44:23")
member_layout.addWidget(self.member_info)
layout.addWidget(member_frame)
# 激活区域
activate_frame = QFrame()
activate_frame.setStyleSheet("""
QFrame {
background-color: white;
border-radius: 10px;
padding: 10px;
}
""")
activate_layout = QVBoxLayout(activate_frame)
activate_layout.setContentsMargins(10, 5, 10, 5)
activate_layout.setSpacing(5)
# 激活标题和说明
activate_title = QLabel("激活(益加)会员,多个激活码可益加整体时长")
activate_title.setStyleSheet("""
QLabel {
color: #333333;
}
""")
# 激活码输入区域
input_layout = QHBoxLayout()
input_layout.setSpacing(10)
self.activate_input = QLineEdit()
self.activate_input.setPlaceholderText("请输入激活码")
activate_button = QPushButton("激活")
activate_button.setMaximumWidth(80)
activate_button.clicked.connect(self.activate_license)
input_layout.addWidget(self.activate_input)
input_layout.addWidget(activate_button)
activate_layout.addWidget(activate_title)
activate_layout.addLayout(input_layout)
layout.addWidget(activate_frame)
# 功能按钮区域
button_frame = QFrame()
button_frame.setStyleSheet("""
QFrame {
background-color: white;
border-radius: 10px;
padding: 15px;
}
""")
button_layout = QVBoxLayout(button_frame)
button_layout.setSpacing(10)
# 第一行按钮
self.update_button = QPushButton("刷新Cursor编辑器授权")
self.update_button.clicked.connect(self.start_update)
# 第二行按钮
self.reset_button = QPushButton("实现Cursor0.45.+限制")
self.reset_button.clicked.connect(self.reset_machine)
# 第三行按钮
self.disable_update_button = QPushButton("禁用Cursor版本更新")
self.disable_update_button.clicked.connect(self.disable_cursor_update)
# 设置按钮样式
for button in [self.update_button, self.reset_button, self.disable_update_button]:
button.setStyleSheet("""
QPushButton {
background-color: #0066cc;
color: white;
border: none;
border-radius: 4px;
padding: 12px;
font-size: 14px;
}
QPushButton:hover {
background-color: #0052a3;
}
QPushButton:pressed {
background-color: #003d7a;
}
""")
button_layout.addWidget(self.update_button)
button_layout.addWidget(self.reset_button)
button_layout.addWidget(self.disable_update_button)
layout.addWidget(button_frame)
# 初始化工作线程
self.worker = None
logging.info("界面元素初始化完成")
except Exception as e:
logging.error(f"初始化界面元素时发生错误: {str(e)}")
raise
def copy_device_id(self):
"""复制设备ID到剪贴板"""
try:
clipboard = QApplication.clipboard()
clipboard.setText(self.device_id_text.text())
QMessageBox.information(self, "提示", "设备ID已复制到剪贴板")
logging.info("设备ID已复制到剪贴板")
except Exception as e:
logging.error(f"复制设备ID时发生错误: {str(e)}")
QMessageBox.warning(self, "错误", f"复制失败: {str(e)}")
def append_log(self, text):
"""添加日志到状态显示区域"""
try:
self.status_text.append(text)
self.status_text.verticalScrollBar().setValue(
self.status_text.verticalScrollBar().maximum()
)
logging.info(f"状态更新: {text}")
except Exception as e:
logging.error(f"更新状态显示时发生错误: {str(e)}")
def start_update(self):
"""开始更新流程"""
try:
self.update_button.setEnabled(False)
self.reset_button.setEnabled(False)
self.status_text.clear()
self.append_log("开始更新流程...")
self.worker = UpdateWorker(self.updater)
self.worker.finished.connect(self.update_finished)
self.worker.progress.connect(self.append_log)
self.worker.start()
logging.info("更新进程已启动")
except Exception as e:
logging.error(f"启动更新进程时发生错误: {str(e)}")
self.update_button.setEnabled(True)
self.reset_button.setEnabled(True)
QMessageBox.warning(self, "错误", f"启动更新失败: {str(e)}")
def update_finished(self, success, message):
"""更新完成的回调"""
try:
self.update_button.setEnabled(True)
self.reset_button.setEnabled(True)
if success:
QMessageBox.information(self, "成功", message)
else:
QMessageBox.warning(self, "失败", message)
self.append_log(message)
logging.info(f"更新完成: {message}")
except Exception as e:
logging.error(f"处理更新完成回调时发生错误: {str(e)}")
def reset_machine(self):
"""重置机器码"""
try:
reply = QMessageBox.question(
self,
"确认",
"确定要重置机器码吗?",
QMessageBox.Yes | QMessageBox.No,
QMessageBox.No
)
if reply == QMessageBox.Yes:
logging.info("开始重置机器码")
success = self.updater.reset_machine_id()
if success:
QMessageBox.information(self, "成功", "机器码重置成功")
self.device_id_text.setText(self.updater.hardware_id)
logging.info("机器码重置成功")
else:
QMessageBox.warning(self, "失败", "机器码重置失败")
logging.error("机器码重置失败")
except Exception as e:
logging.error(f"重置机器码时发生错误: {str(e)}")
QMessageBox.warning(self, "错误", f"重置失败: {str(e)}")
def activate_license(self):
"""激活许可证"""
try:
activation_code = self.activate_input.text().strip()
if not activation_code:
QMessageBox.warning(self, "提示", "请输入激活码")
return
# TODO: 实现激活逻辑
logging.info(f"正在处理激活请求,激活码: {activation_code}")
QMessageBox.information(self, "提示", "激活功能即将实现")
except Exception as e:
logging.error(f"激活过程中发生错误: {str(e)}")
QMessageBox.warning(self, "错误", f"激活失败: {str(e)}")
def disable_cursor_update(self):
"""禁用Cursor版本更新"""
try:
# TODO: 实现禁用更新逻辑
logging.info("正在禁用Cursor版本更新...")
QMessageBox.information(self, "提示", "禁用更新功能即将实现")
except Exception as e:
logging.error(f"禁用更新时发生错误: {str(e)}")
QMessageBox.warning(self, "错误", f"禁用失败: {str(e)}")
def main():
try:
logging.info("程序启动...")
app = QApplication(sys.argv)
# 设置应用程序图标
if getattr(sys, 'frozen', False):
# 如果是打包后的应用
application_path = sys._MEIPASS
logging.info(f"运行于打包环境: {application_path}")
else:
# 如果是开发环境
application_path = os.path.dirname(os.path.abspath(__file__))
logging.info(f"运行于开发环境: {application_path}")
# 创建并显示主窗口
window = MainWindow()
window.show()
logging.info("主窗口已显示")
sys.exit(app.exec_())
except Exception as e:
logging.error(f"程序运行时发生错误: {str(e)}")
QMessageBox.critical(None, "错误", f"程序启动失败: {str(e)}")
sys.exit(1)
if __name__ == "__main__":
main()

View File

@@ -1,4 +1,7 @@
DrissionPage==4.1.0.9
colorama==0.4.6
python-dotenv==1.0.0
pyinstaller
pyinstaller==6.3.0
PyQt5==5.15.9
requests==2.31.0
urllib3==2.1.0

338
update_cursor_token.py Normal file
View File

@@ -0,0 +1,338 @@
import os
import json
import platform
import requests
import urllib3
import ssl
import sys
import subprocess
import hashlib
from cursor_auth_manager import CursorAuthManager
from logger import logging
from reset_machine import MachineIDResetter
import patch_cursor_get_machine_id
from exit_cursor import ExitCursor
import go_cursor_help
from logo import print_logo
from typing import Tuple, Dict, Optional
class CursorTokenUpdater:
def __init__(self):
self.auth_manager = CursorAuthManager()
self._hardware_id = None # 延迟初始化硬件ID
@property
def hardware_id(self) -> str:
"""获取硬件ID延迟初始化"""
if self._hardware_id is None:
self._hardware_id = self._get_hardware_id()
return self._hardware_id
def _get_hardware_id(self) -> str:
"""获取硬件唯一标识
方案1: CPU ID + 主板序列号 + BIOS序列号
方案2: 系统盘序列号 + Windows安装时间
方案3: 计算机名(最后的备选方案)
"""
try:
# 创建startupinfo对象来隐藏命令行窗口
startupinfo = None
if sys.platform == "win32":
startupinfo = subprocess.STARTUPINFO()
startupinfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW
startupinfo.wShowWindow = subprocess.SW_HIDE
# 方案1: 尝试获取硬件信息
try:
# 获取CPU ID
cpu_info = subprocess.check_output('wmic cpu get ProcessorId', startupinfo=startupinfo).decode()
cpu_id = cpu_info.split('\n')[1].strip()
# 获取主板序列号
board_info = subprocess.check_output('wmic baseboard get SerialNumber', startupinfo=startupinfo).decode()
board_id = board_info.split('\n')[1].strip()
# 获取BIOS序列号
bios_info = subprocess.check_output('wmic bios get SerialNumber', startupinfo=startupinfo).decode()
bios_id = bios_info.split('\n')[1].strip()
# 如果所有信息都获取成功且有效
if all([cpu_id, board_id, bios_id]) and not all(x in ['', '0', 'None', 'To be filled by O.E.M.'] for x in [cpu_id, board_id, bios_id]):
combined = f"{cpu_id}:{board_id}:{bios_id}"
hardware_id = hashlib.md5(combined.encode()).hexdigest()
logging.info("使用硬件信息生成ID成功")
return hardware_id
except Exception as e:
logging.warning(f"方案1失败: {str(e)}")
# 方案2: 系统盘序列号 + Windows安装时间
try:
backup_info = []
# 获取系统盘序列号
volume_info = subprocess.check_output('wmic logicaldisk where "DeviceID=\'C:\'" get VolumeSerialNumber', startupinfo=startupinfo).decode()
volume_serial = volume_info.split('\n')[1].strip()
if volume_serial and volume_serial not in ['', '0']:
backup_info.append(("volume", volume_serial))
# 获取Windows安装时间
os_info = subprocess.check_output('wmic os get InstallDate', startupinfo=startupinfo).decode()
install_date = os_info.split('\n')[1].strip()
if install_date:
backup_info.append(("install", install_date))
if backup_info:
combined = "|".join(f"{k}:{v}" for k, v in sorted(backup_info))
hardware_id = hashlib.md5(combined.encode()).hexdigest()
logging.info("使用系统信息生成ID成功")
return hardware_id
except Exception as e:
logging.warning(f"方案2失败: {str(e)}")
# 方案3: 使用计算机名(最后的备选方案)
computer_name = platform.node()
if computer_name:
hardware_id = hashlib.md5(computer_name.encode()).hexdigest()
logging.info("使用计算机名生成ID成功")
return hardware_id
raise ValueError("无法获取任何可用信息来生成硬件ID")
except Exception as e:
error_msg = f"生成硬件ID失败: {str(e)}"
logging.error(error_msg)
raise RuntimeError(error_msg)
def get_unused_account(self) -> Tuple[bool, str, Optional[Dict]]:
"""
从API获取未使用的账号
Returns:
Tuple[bool, str, Optional[Dict]]:
- 是否成功
- 错误信息
- 账号数据(如果成功)
"""
endpoint = "https://cursorapi.nosqli.com/admin/api.account/getUnused"
data = {
"machine_id": self.hardware_id
}
headers = {
"Content-Type": "application/json"
}
# 禁用SSL警告
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
request_kwargs = {
"json": data,
"headers": headers,
"timeout": 30,
"verify": False
}
try:
try:
response = requests.post(endpoint, **request_kwargs)
except requests.exceptions.SSLError:
# SSL错误时使用自定义SSL上下文
ssl_context = ssl.create_default_context()
ssl_context.check_hostname = False
ssl_context.verify_mode = ssl.CERT_NONE
session = requests.Session()
session.verify = False
response = session.post(endpoint, **request_kwargs)
response_data = response.json()
if response_data.get("code") == 200:
account_data = response_data.get("data", {})
# 获取账号信息
email = account_data.get("email", "")
access_token = account_data.get("access_token", "")
refresh_token = account_data.get("refresh_token", "")
expire_time = account_data.get("expire_time", "")
days_left = account_data.get("days_left", 0)
if not all([email, access_token, refresh_token]):
return False, "获取账号信息不完整", None
account_info = {
"email": email,
"access_token": access_token,
"refresh_token": refresh_token,
"expire_time": expire_time,
"days_left": days_left
}
logging.info(f"成功获取账号信息 - 邮箱: {email}, 剩余天数: {days_left}")
return True, "", account_info
else:
error_msg = response_data.get("msg", "未知错误")
return False, f"API返回错误: {error_msg}", None
except Exception as e:
error_msg = f"获取账号时发生错误: {str(e)}"
logging.error(error_msg)
return False, error_msg, None
def update_auth_info(self, email: str, access_token: str, refresh_token: str = None) -> bool:
"""
更新Cursor的认证信息
Args:
email: 用户邮箱
access_token: 访问令牌
refresh_token: 刷新令牌如果没有提供将使用access_token
Returns:
bool: 更新是否成功
"""
try:
# 如果没有提供refresh_token使用access_token
if refresh_token is None:
refresh_token = access_token
# 更新认证信息
result = self.auth_manager.update_auth(
email=email,
access_token=access_token,
refresh_token=refresh_token
)
if result:
logging.info(f"认证信息更新成功 - 邮箱: {email}")
return True
else:
logging.error("认证信息更新失败")
return False
except Exception as e:
logging.error(f"更新认证信息时发生错误: {str(e)}")
return False
def reset_machine_id(self, greater_than_0_45: bool = True) -> bool:
"""
重置机器码
Args:
greater_than_0_45: 是否大于0.45版本
Returns:
bool: 重置是否成功
"""
try:
logging.info("开始重置机器码...")
resetter = MachineIDResetter()
result = resetter.reset(greater_than_0_45)
if result:
logging.info("机器码重置成功")
# 重置后更新硬件ID缓存
self._hardware_id = None
return True
else:
logging.error("机器码重置失败")
return False
except Exception as e:
logging.error(f"重置机器码时发生错误: {str(e)}")
return False
def patch_machine_id(self) -> bool:
"""
修补机器码获取方法
Returns:
bool: 修补是否成功
"""
try:
logging.info("开始修补机器码获取方法...")
patch_cursor_get_machine_id.patch()
logging.info("机器码获取方法修补完成")
return True
except Exception as e:
logging.error(f"修补机器码获取方法时发生错误: {str(e)}")
return False
def exit_cursor(self) -> bool:
"""
退出Cursor进程
Returns:
bool: 退出是否成功
"""
try:
logging.info("正在退出Cursor进程...")
exit_handler = ExitCursor()
exit_handler.exit()
logging.info("Cursor进程已退出")
return True
except Exception as e:
logging.error(f"退出Cursor进程时发生错误: {str(e)}")
return False
def full_update_process(self, email: str = None, access_token: str = None, refresh_token: str = None) -> bool:
"""
执行完整的更新流程
Args:
email: 用户邮箱可选如果不提供则从API获取
access_token: 访问令牌可选如果不提供则从API获取
refresh_token: 刷新令牌可选如果不提供则从API获取
Returns:
bool: 更新流程是否全部成功
"""
print_logo()
logging.info("=== 开始完整更新流程 ===")
# 1. 退出Cursor进程
if not self.exit_cursor():
return False
# 2. 修补机器码获取方法
if not self.patch_machine_id():
return False
# 3. 重置机器码
if not self.reset_machine_id(greater_than_0_45=True):
return False
# 4. 如果没有提供认证信息从API获取
if not all([email, access_token]):
success, error_msg, account_info = self.get_unused_account()
if not success:
logging.error(f"无法获取账号信息: {error_msg}")
return False
email = account_info["email"]
access_token = account_info["access_token"]
refresh_token = account_info["refresh_token"]
# 5. 更新认证信息
if not self.update_auth_info(email, access_token, refresh_token):
return False
logging.info("=== 所有操作已完成 ===")
return True
def main():
updater = CursorTokenUpdater()
# 从环境变量获取认证信息(可选)
# 如果环境变量中有认证信息,使用环境变量中的信息
# 否则将从API获取新的账号信息
success = updater.full_update_process(
)
print("更新状态:", "成功" if success else "失败")
if __name__ == "__main__":
main()