feat: 添加图形界面和自动化重置功能 - 新增 PyQt6 图形界面，优化密码输入和 sudo 权限处理，改进重置机器码流程，添加应用图标和打包配置，更新依赖项
This commit is contained in:
ruisu
@@ -13,6 +13,13 @@ import go_cursor_help
|
||||
import patch_cursor_get_machine_id
|
||||
from reset_machine import MachineIDResetter
|
||||
from logo import print_logo
|
||||
from typing import Optional, Tuple, Dict, Any
|
||||
from urllib3.exceptions import InsecureRequestWarning
|
||||
import urllib3
|
||||
import time
|
||||
|
||||
# 禁用不安全请求警告
|
||||
requests.packages.urllib3.disable_warnings(InsecureRequestWarning)
|
||||
|
||||
# 定义 EMOJI 字典
|
||||
EMOJI = {"ERROR": "❌", "WARNING": "⚠️", "INFO": "ℹ️"}
|
||||
@@ -152,6 +159,268 @@ def get_mac_unique_id() -> str:
|
||||
return hashlib.md5(unique_string.encode()).hexdigest()
|
||||
|
||||
|
||||
class CursorAccountManager:
|
||||
def __init__(self):
|
||||
self.base_url = "https://cursorapi.nosqli.com"
|
||||
self.api_endpoints = {
|
||||
"activate": f"{self.base_url}/admin/api.member/activate",
|
||||
"status": f"{self.base_url}/admin/api.member/status",
|
||||
"get_unused": f"{self.base_url}/admin/api.account/getUnused",
|
||||
"heartbeat": f"{self.base_url}/admin/api.member/heartbeat"
|
||||
}
|
||||
self.hardware_id = get_mac_unique_id()
|
||||
|
||||
def get_device_info(self) -> dict:
|
||||
"""获取设备信息"""
|
||||
return {
|
||||
"system": platform.system(),
|
||||
"device_name": platform.node(),
|
||||
"ip": self._get_ip_address(),
|
||||
"location": self._get_location()
|
||||
}
|
||||
|
||||
def _get_ip_address(self) -> str:
|
||||
"""获取IP地址"""
|
||||
try:
|
||||
response = requests.get('https://api.ipify.org?format=json', timeout=5)
|
||||
return response.json()['ip']
|
||||
except:
|
||||
return "未知"
|
||||
|
||||
def _get_location(self) -> str:
|
||||
"""获取地理位置"""
|
||||
try:
|
||||
ip = self._get_ip_address()
|
||||
if ip != "未知":
|
||||
response = requests.get(f'http://ip-api.com/json/{ip}', timeout=5)
|
||||
data = response.json()
|
||||
if data.get('status') == 'success':
|
||||
return f"{data.get('country', '')} {data.get('city', '')}"
|
||||
except:
|
||||
pass
|
||||
return "未知"
|
||||
|
||||
def check_member_status(self) -> tuple[bool, dict]:
|
||||
"""检查会员状态
|
||||
|
||||
Returns:
|
||||
tuple[bool, dict]: (是否成功, 状态信息)
|
||||
"""
|
||||
try:
|
||||
data = {
|
||||
"machine_id": self.hardware_id
|
||||
}
|
||||
|
||||
api_url = self.api_endpoints["status"]
|
||||
logging.info(f"正在检查会员状态...")
|
||||
|
||||
request_kwargs = {
|
||||
"json": data,
|
||||
"headers": {"Content-Type": "application/json"},
|
||||
"timeout": 2,
|
||||
"verify": False
|
||||
}
|
||||
|
||||
session = requests.Session()
|
||||
session.verify = False
|
||||
|
||||
try:
|
||||
response = session.post(api_url, **request_kwargs)
|
||||
except requests.exceptions.Timeout:
|
||||
logging.warning("首次请求超时，正在重试...")
|
||||
response = session.post(api_url, **request_kwargs)
|
||||
|
||||
result = response.json()
|
||||
logging.info(f"状态检查响应: {result}")
|
||||
|
||||
if result.get("code") in [1, 200]:
|
||||
api_data = result.get("data", {})
|
||||
status_data = {
|
||||
"is_active": api_data.get("status") == "active",
|
||||
"expire_time": api_data.get("expire_time", ""),
|
||||
"total_days": api_data.get("total_days", 0),
|
||||
"days_left": api_data.get("days_left", 0),
|
||||
"device_info": self.get_device_info()
|
||||
}
|
||||
return True, status_data
|
||||
else:
|
||||
error_msg = result.get("msg", "未知错误")
|
||||
logging.error(f"获取状态失败: {error_msg}")
|
||||
return False, {
|
||||
"is_active": False,
|
||||
"expire_time": "",
|
||||
"total_days": 0,
|
||||
"days_left": 0,
|
||||
"device_info": self.get_device_info()
|
||||
}
|
||||
|
||||
except Exception as e:
|
||||
logging.error(f"获取会员状态失败: {str(e)}")
|
||||
return False, {
|
||||
"is_active": False,
|
||||
"expire_time": "",
|
||||
"total_days": 0,
|
||||
"days_left": 0,
|
||||
"device_info": self.get_device_info()
|
||||
}
|
||||
|
||||
def check_activation_code(self, code: str) -> tuple[bool, str, dict | None]:
|
||||
"""检查激活码
|
||||
|
||||
Args:
|
||||
code: 激活码
|
||||
|
||||
Returns:
|
||||
tuple: (成功标志, 消息, 账号信息)
|
||||
"""
|
||||
max_retries = 3 # 最大重试次数
|
||||
retry_delay = 1 # 重试间隔(秒)
|
||||
|
||||
for attempt in range(max_retries):
|
||||
try:
|
||||
data = {
|
||||
"machine_id": self.hardware_id,
|
||||
"code": code
|
||||
}
|
||||
|
||||
# 设置请求参数
|
||||
request_kwargs = {
|
||||
"json": data,
|
||||
"headers": {
|
||||
"Content-Type": "application/json",
|
||||
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/121.0.0.0 Safari/537.36",
|
||||
"Accept": "*/*",
|
||||
"Connection": "keep-alive"
|
||||
},
|
||||
"timeout": 10,
|
||||
"verify": False
|
||||
}
|
||||
|
||||
# 创建session
|
||||
session = requests.Session()
|
||||
session.verify = False
|
||||
|
||||
# 设置重试策略
|
||||
retry_strategy = urllib3.Retry(
|
||||
total=3,
|
||||
backoff_factor=0.5,
|
||||
status_forcelist=[500, 502, 503, 504]
|
||||
)
|
||||
adapter = requests.adapters.HTTPAdapter(max_retries=retry_strategy)
|
||||
session.mount("http://", adapter)
|
||||
session.mount("https://", adapter)
|
||||
|
||||
try:
|
||||
# 尝试发送请求
|
||||
response = session.post(
|
||||
self.api_endpoints["activate"],
|
||||
**request_kwargs
|
||||
)
|
||||
response.raise_for_status()
|
||||
|
||||
result = response.json()
|
||||
logging.info(f"激活响应: {result}")
|
||||
|
||||
# 激活成功
|
||||
if result["code"] == 200:
|
||||
api_data = result["data"]
|
||||
account_info = {
|
||||
"status": "active",
|
||||
"expire_time": api_data.get("expire_time", ""),
|
||||
"total_days": api_data.get("total_days", 0),
|
||||
"days_left": api_data.get("days_left", 0),
|
||||
"device_info": self.get_device_info()
|
||||
}
|
||||
return True, result["msg"], account_info
|
||||
# 激活码无效或已被使用
|
||||
elif result["code"] == 400:
|
||||
logging.warning(f"激活码无效或已被使用: {result.get('msg', '未知错误')}")
|
||||
return False, result.get("msg", "激活码无效或已被使用"), None
|
||||
# 其他错误情况
|
||||
else:
|
||||
error_msg = result.get("msg", "未知错误")
|
||||
if attempt < max_retries - 1:
|
||||
logging.warning(f"第{attempt + 1}次尝试失败: {error_msg}, 准备重试...")
|
||||
time.sleep(retry_delay)
|
||||
continue
|
||||
logging.error(f"激活失败: {error_msg}")
|
||||
return False, error_msg, None
|
||||
|
||||
except requests.exceptions.RequestException as e:
|
||||
if attempt < max_retries - 1:
|
||||
logging.warning(f"第{attempt + 1}次网络请求失败: {str(e)}, 准备重试...")
|
||||
time.sleep(retry_delay)
|
||||
continue
|
||||
logging.error(f"网络请求失败: {str(e)}")
|
||||
return False, f"网络连接失败: {str(e)}", None
|
||||
|
||||
except Exception as e:
|
||||
if attempt < max_retries - 1:
|
||||
logging.warning(f"第{attempt + 1}次请求发生错误: {str(e)}, 准备重试...")
|
||||
time.sleep(retry_delay)
|
||||
continue
|
||||
logging.error(f"激活失败: {str(e)}")
|
||||
return False, f"激活失败: {str(e)}", None
|
||||
|
||||
return False, "多次尝试后激活失败，请检查网络连接或稍后重试", None
|
||||
|
||||
|
||||
def reset_auth_with_password(password: str = None) -> tuple[bool, str]:
|
||||
"""
|
||||
封装重置授权的完整流程
|
||||
|
||||
Args:
|
||||
password: 系统密码（可选）
|
||||
|
||||
Returns:
|
||||
tuple[bool, str]: (是否成功, 消息)
|
||||
"""
|
||||
try:
|
||||
logging.info("\n=== 初始化重置流程 ===")
|
||||
greater_than_0_45 = check_cursor_version()
|
||||
|
||||
logging.info("正在从API获取账号信息...")
|
||||
success, account_data = get_account_from_api()
|
||||
|
||||
if not success:
|
||||
return False, "获取账号信息失败"
|
||||
|
||||
email = account_data.get("email", "")
|
||||
access_token = account_data.get("access_token", "")
|
||||
refresh_token = account_data.get("refresh_token", "")
|
||||
|
||||
if not all([email, access_token, refresh_token]):
|
||||
return False, "账号信息不完整"
|
||||
|
||||
logging.info(f"获取到账号信息：\n邮箱: {email}")
|
||||
|
||||
# 更新认证信息
|
||||
logging.info("正在更新认证信息...")
|
||||
auth_manager = CursorAuthManager()
|
||||
if auth_manager.update_auth(email, access_token, refresh_token):
|
||||
logging.info("认证信息更新成功")
|
||||
|
||||
# 重置机器码
|
||||
logging.info("正在重置机器码...")
|
||||
|
||||
# 如果提供了密码，设置环境变量
|
||||
if password:
|
||||
import os
|
||||
os.environ['SUDO_PASSWORD'] = password
|
||||
|
||||
reset_machine_id(greater_than_0_45)
|
||||
logging.info("重置完成")
|
||||
return True, "重置成功"
|
||||
else:
|
||||
return False, "更新认证信息失败"
|
||||
|
||||
except Exception as e:
|
||||
logging.error(f"重置过程出错: {str(e)}")
|
||||
import traceback
|
||||
logging.error(traceback.format_exc())
|
||||
return False, f"重置失败: {str(e)}"
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
print_logo()
|
||||
greater_than_0_45 = check_cursor_version()
|
||||
|
||||
Reference in New Issue
Block a user