保存现有功能 增加域名和添加时间关联
This commit is contained in:
huangzhenpc
283
DEPLOYMENT_README.md
Normal file
283
DEPLOYMENT_README.md
Normal file
@@ -0,0 +1,283 @@
|
||||
# Cursor自动化服务部署指南
|
||||
|
||||
本文档详细说明如何在新服务器上从零开始部署Cursor自动化服务，包括环境准备、数据库配置、服务启动等各个步骤。
|
||||
|
||||
## 系统要求
|
||||
|
||||
- **操作系统**: Linux (推荐Ubuntu/Debian)，也支持Windows和macOS
|
||||
- **Python版本**: 3.7+
|
||||
- **数据库**: MySQL 5.7+ 或 MariaDB 10+
|
||||
- **缓存系统**: Redis 5+ (可选但推荐)
|
||||
- **内存**: 至少2GB RAM
|
||||
- **存储**: 至少1GB可用空间
|
||||
- **网络**: 稳定的互联网连接
|
||||
|
||||
## 部署流程概述
|
||||
|
||||
1. 安装必要的系统依赖
|
||||
2. 准备Python环境
|
||||
3. 下载代码并安装依赖
|
||||
4. 配置和初始化数据库
|
||||
5. 导入邮箱账号
|
||||
6. 启动服务
|
||||
7. 设置为系统服务(可选)
|
||||
|
||||
## 详细部署步骤
|
||||
|
||||
### 1. 安装必要的系统依赖
|
||||
|
||||
#### Ubuntu/Debian系统:
|
||||
|
||||
```bash
|
||||
# 更新系统包
|
||||
sudo apt update
|
||||
|
||||
# 安装Python、MySQL和Redis
|
||||
sudo apt install -y python3 python3-venv python3-full mysql-server redis-server
|
||||
|
||||
# 安装pip和开发库
|
||||
sudo apt install -y python3-pip python3-dev default-libmysqlclient-dev build-essential
|
||||
```
|
||||
|
||||
#### CentOS/RHEL系统:
|
||||
|
||||
```bash
|
||||
# 安装Python
|
||||
sudo yum install -y python3 python3-devel
|
||||
|
||||
# 安装MySQL
|
||||
sudo yum install -y mariadb-server mariadb-devel
|
||||
|
||||
# 安装Redis
|
||||
sudo yum install -y redis
|
||||
|
||||
# 启动服务
|
||||
sudo systemctl start mariadb
|
||||
sudo systemctl start redis
|
||||
sudo systemctl enable mariadb
|
||||
sudo systemctl enable redis
|
||||
```
|
||||
|
||||
### 2. 准备Python环境
|
||||
|
||||
```bash
|
||||
# 创建项目目录
|
||||
mkdir -p /opt/cursor-service
|
||||
cd /opt/cursor-service
|
||||
|
||||
# 创建并激活虚拟环境
|
||||
python3 -m venv venv
|
||||
source venv/bin/activate
|
||||
```
|
||||
|
||||
### 3. 下载代码并安装依赖
|
||||
|
||||
```bash
|
||||
# 下载代码(示例使用git，也可以直接上传文件)
|
||||
git clone https://your-repo-url.git .
|
||||
# 或者直接解压上传的zip文件
|
||||
|
||||
# 安装依赖
|
||||
pip install -r requirements.txt
|
||||
pip install cryptography # 确保安装cryptography包
|
||||
```
|
||||
|
||||
### 4. 配置和初始化数据库
|
||||
|
||||
我们提供了两种初始化数据库的方式：
|
||||
|
||||
#### 方式一：使用交互式设置向导(推荐)
|
||||
|
||||
```bash
|
||||
# 修复可能的MySQL格式化问题
|
||||
python fix_setup.py
|
||||
|
||||
# 运行设置向导
|
||||
python setup_environment.py
|
||||
```
|
||||
|
||||
设置向导将引导您完成：
|
||||
- 配置服务器标识
|
||||
- 设置MySQL数据库连接信息
|
||||
- 配置Redis(可选)
|
||||
- 创建数据库和表结构
|
||||
|
||||
#### 方式二：手动执行初始化脚本
|
||||
|
||||
```bash
|
||||
# 直接运行数据库初始化脚本
|
||||
python init_database.py
|
||||
```
|
||||
|
||||
脚本会交互式地询问：
|
||||
- MySQL root用户名和密码
|
||||
- 应用数据库名和用户信息
|
||||
- Redis配置(可选)
|
||||
|
||||
### 5. 导入邮箱账号
|
||||
|
||||
```bash
|
||||
# 准备email.txt文件，格式为：
|
||||
# email@example.com----密码----client_id----refresh_token
|
||||
|
||||
# 运行导入脚本
|
||||
python import_emails.py
|
||||
```
|
||||
|
||||
### 6. 启动服务
|
||||
|
||||
有多种方式可以启动服务：
|
||||
|
||||
#### 方式一：使用统一管理脚本(推荐)
|
||||
|
||||
```bash
|
||||
# 启用并启动服务
|
||||
python start.py --enable
|
||||
|
||||
# 或者直接启动(会根据数据库中的启用状态决定是否启动服务)
|
||||
python start.py
|
||||
```
|
||||
|
||||
#### 方式二：直接启动自动服务
|
||||
|
||||
```bash
|
||||
# 直接启动自动化服务
|
||||
python auto_cursor_service.py
|
||||
```
|
||||
|
||||
#### 方式三：仅运行注册进程
|
||||
|
||||
```bash
|
||||
# 仅启动注册进程
|
||||
python main.py
|
||||
```
|
||||
|
||||
### 7. 设置为系统服务(可选但推荐)
|
||||
|
||||
#### 创建系统服务文件:
|
||||
|
||||
```bash
|
||||
sudo nano /etc/systemd/system/cursor-service.service
|
||||
```
|
||||
|
||||
添加以下内容:
|
||||
|
||||
```ini
|
||||
[Unit]
|
||||
Description=Cursor Auto Registration Service
|
||||
After=network.target mysql.service redis.service
|
||||
|
||||
[Service]
|
||||
Type=simple
|
||||
User=your_username
|
||||
WorkingDirectory=/opt/cursor-service
|
||||
ExecStart=/opt/cursor-service/venv/bin/python start.py
|
||||
Restart=on-failure
|
||||
RestartSec=10
|
||||
|
||||
[Install]
|
||||
WantedBy=multi-user.target
|
||||
```
|
||||
|
||||
#### 启动并设置开机自启:
|
||||
|
||||
```bash
|
||||
sudo systemctl daemon-reload
|
||||
sudo systemctl start cursor-service
|
||||
sudo systemctl enable cursor-service
|
||||
```
|
||||
|
||||
## 管理服务
|
||||
|
||||
### 检查服务状态
|
||||
|
||||
```bash
|
||||
# 查看服务状态
|
||||
python start.py --status
|
||||
|
||||
# 如果设置为系统服务，可以使用
|
||||
sudo systemctl status cursor-service
|
||||
```
|
||||
|
||||
### 启用/禁用服务
|
||||
|
||||
```bash
|
||||
# 启用服务
|
||||
python start.py --enable
|
||||
|
||||
# 禁用服务
|
||||
python start.py --disable
|
||||
```
|
||||
|
||||
### 手动上传账号
|
||||
|
||||
```bash
|
||||
python start.py --upload
|
||||
```
|
||||
|
||||
## 数据库升级
|
||||
|
||||
如果您从旧版本升级，需要运行数据库升级脚本：
|
||||
|
||||
```bash
|
||||
python upgrade_database.py
|
||||
```
|
||||
|
||||
这会添加新的`extracted`字段和相关索引，以支持账号上传功能。
|
||||
|
||||
## 监控与故障排除
|
||||
|
||||
### 日志文件
|
||||
|
||||
主要日志文件：
|
||||
- `auto_cursor_service.log` - 自动服务日志
|
||||
- `upload_accounts.log` - 账号上传日志
|
||||
- `register.log` - 注册进程日志
|
||||
|
||||
### 常见问题解决
|
||||
|
||||
1. **数据库连接失败**
|
||||
- 检查MySQL服务是否运行: `sudo systemctl status mysql`
|
||||
- 验证数据库用户名和密码是否正确
|
||||
|
||||
2. **Redis连接失败**
|
||||
- 检查Redis服务是否运行: `sudo systemctl status redis`
|
||||
- 如不需要Redis，可在config.yaml中设置`use_redis: false`
|
||||
|
||||
3. **服务无法启动**
|
||||
- 检查日志文件中的具体错误信息
|
||||
- 确保所有依赖包都已安装: `pip install -r requirements.txt`
|
||||
|
||||
4. **账号注册失败率高**
|
||||
- 检查网络连接和代理设置
|
||||
- 查看注册日志中的具体错误信息
|
||||
|
||||
5. **邮箱账号不足**
|
||||
- 手动导入更多邮箱账号: `python import_emails.py`
|
||||
- 检查API获取邮箱是否正常
|
||||
|
||||
## 备份与维护
|
||||
|
||||
建议定期备份数据库：
|
||||
|
||||
```bash
|
||||
# MySQL备份
|
||||
mysqldump -u [用户名] -p [数据库名] > backup_$(date +%Y%m%d).sql
|
||||
```
|
||||
|
||||
## 安全考虑
|
||||
|
||||
1. 使用强密码保护数据库
|
||||
2. 限制MySQL和Redis只接受本地连接
|
||||
3. 保护config.yaml文件，不要泄露API密钥和数据库密码
|
||||
4. 定期更新系统和依赖包
|
||||
|
||||
## 更新服务
|
||||
|
||||
更新服务代码后，请按以下步骤操作：
|
||||
|
||||
1. 停止服务: `python start.py --disable` 或 `sudo systemctl stop cursor-service`
|
||||
2. 更新代码
|
||||
3. 如有新的依赖，安装: `pip install -r requirements.txt`
|
||||
4. 如有数据库结构变更，运行升级脚本: `python upgrade_database.py`
|
||||
5. 重新启动服务: `python start.py --enable` 或 `sudo systemctl start cursor-service`
|
||||
186
fix_setup.py
Normal file
186
fix_setup.py
Normal file
@@ -0,0 +1,186 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Cursor自动化服务 - 修复脚本
|
||||
添加虚拟环境支持并修复MySQL连接问题
|
||||
"""
|
||||
import os
|
||||
import sys
|
||||
import subprocess
|
||||
import shutil
|
||||
import time
|
||||
|
||||
# 彩色输出函数
|
||||
class Colors:
|
||||
GREEN = '\033[92m'
|
||||
YELLOW = '\033[93m'
|
||||
RED = '\033[91m'
|
||||
BLUE = '\033[94m'
|
||||
ENDC = '\033[0m'
|
||||
BOLD = '\033[1m'
|
||||
|
||||
def print_success(msg):
|
||||
print(f"{Colors.GREEN}✓ {msg}{Colors.ENDC}")
|
||||
|
||||
def print_warning(msg):
|
||||
print(f"{Colors.YELLOW}⚠ {msg}{Colors.ENDC}")
|
||||
|
||||
def print_error(msg):
|
||||
print(f"{Colors.RED}✗ {msg}{Colors.ENDC}")
|
||||
|
||||
def print_title(title):
|
||||
print("\n" + "=" * 60)
|
||||
print(f"{Colors.BOLD}{Colors.BLUE}{title.center(60)}{Colors.ENDC}")
|
||||
print("=" * 60 + "\n")
|
||||
|
||||
def check_virtual_env():
|
||||
"""检查是否在虚拟环境中运行"""
|
||||
return hasattr(sys, 'real_prefix') or (hasattr(sys, 'base_prefix') and sys.base_prefix != sys.prefix)
|
||||
|
||||
def setup_virtual_env():
|
||||
"""设置Python虚拟环境"""
|
||||
print_title("虚拟环境设置")
|
||||
|
||||
if check_virtual_env():
|
||||
print_success("当前已在虚拟环境中运行")
|
||||
return True
|
||||
|
||||
print_warning("当前未在虚拟环境中运行")
|
||||
create_venv = input("是否创建虚拟环境? (推荐) (y/n, 默认: y): ").strip().lower()
|
||||
if create_venv == 'n':
|
||||
print_warning("跳过虚拟环境创建")
|
||||
return True
|
||||
|
||||
venv_path = input("请输入虚拟环境路径 (默认: ./venv): ").strip() or "./venv"
|
||||
|
||||
try:
|
||||
# 创建虚拟环境
|
||||
print(f"创建虚拟环境: {venv_path}")
|
||||
subprocess.check_call([sys.executable, "-m", "venv", venv_path])
|
||||
|
||||
# 获取激活脚本路径
|
||||
if sys.platform.startswith('win'):
|
||||
activate_script = os.path.join(venv_path, "Scripts", "activate.bat")
|
||||
python_exe = os.path.join(venv_path, "Scripts", "python.exe")
|
||||
else:
|
||||
activate_script = os.path.join(venv_path, "bin", "activate")
|
||||
python_exe = os.path.join(venv_path, "bin", "python")
|
||||
|
||||
# 创建安装脚本
|
||||
print("创建安装脚本...")
|
||||
setup_script = "setup_venv.sh"
|
||||
with open(setup_script, "w") as f:
|
||||
if sys.platform.startswith('win'):
|
||||
f.write(f"@echo off\n")
|
||||
f.write(f"call {activate_script}\n")
|
||||
f.write(f"pip install -r requirements.txt\n")
|
||||
f.write(f"pip install cryptography\n")
|
||||
f.write(f"python setup_environment.py\n")
|
||||
f.write(f"pause\n")
|
||||
else:
|
||||
f.write(f"#!/bin/bash\n")
|
||||
f.write(f"source {activate_script}\n")
|
||||
f.write(f"pip install -r requirements.txt\n")
|
||||
f.write(f"pip install cryptography\n")
|
||||
f.write(f"python setup_environment.py\n")
|
||||
|
||||
if not sys.platform.startswith('win'):
|
||||
os.chmod(setup_script, 0o755)
|
||||
|
||||
print_success(f"虚拟环境创建成功: {venv_path}")
|
||||
print_success(f"安装脚本已创建: {setup_script}")
|
||||
|
||||
print("\n请运行以下命令完成安装:")
|
||||
if sys.platform.startswith('win'):
|
||||
print(f" {setup_script}")
|
||||
else:
|
||||
print(f" ./{setup_script}")
|
||||
|
||||
return False
|
||||
except Exception as e:
|
||||
print_error(f"创建虚拟环境失败: {str(e)}")
|
||||
return True
|
||||
|
||||
def fix_mysql_format_issue():
|
||||
"""修复MySQL格式化问题"""
|
||||
print_title("修复MySQL连接问题")
|
||||
|
||||
if not os.path.exists("setup_environment.py"):
|
||||
print_error("未找到setup_environment.py文件")
|
||||
return False
|
||||
|
||||
backup_file = f"setup_environment.py.bak.{int(time.time())}"
|
||||
try:
|
||||
# 创建备份
|
||||
shutil.copy2("setup_environment.py", backup_file)
|
||||
print_success(f"已创建备份: {backup_file}")
|
||||
|
||||
# 读取文件内容
|
||||
with open("setup_environment.py", "r", encoding="utf-8") as f:
|
||||
content = f.read()
|
||||
|
||||
# 修复格式化问题
|
||||
fixed_content = content
|
||||
|
||||
# 1. 添加cryptography到required_packages
|
||||
if "cryptography" not in content:
|
||||
fixed_content = fixed_content.replace(
|
||||
'required_packages = [\n "loguru",\n "pymysql",\n "aiomysql",\n "redis",\n "pyyaml"',
|
||||
'required_packages = [\n "loguru",\n "pymysql",\n "aiomysql",\n "redis",\n "pyyaml",\n "cryptography"'
|
||||
)
|
||||
print_success("已添加cryptography到依赖包列表")
|
||||
|
||||
# 2. 修复CREATE USER语句的格式化问题
|
||||
if "cursor.execute(f\"CREATE USER '{db_user}'@'%'" in content:
|
||||
fixed_content = fixed_content.replace(
|
||||
"cursor.execute(f\"CREATE USER '{db_user}'@'%' IDENTIFIED BY %s\", (db_password,))",
|
||||
"cursor.execute(\"CREATE USER %s@'%%' IDENTIFIED BY %s\", (db_user, db_password))"
|
||||
)
|
||||
print_success("已修复CREATE USER语句")
|
||||
|
||||
# 3. 修复ALTER USER语句的格式化问题
|
||||
if "cursor.execute(f\"ALTER USER '{db_user}'@'%'" in content:
|
||||
fixed_content = fixed_content.replace(
|
||||
"cursor.execute(f\"ALTER USER '{db_user}'@'%' IDENTIFIED BY %s\", (db_password,))",
|
||||
"cursor.execute(\"ALTER USER %s@'%%' IDENTIFIED BY %s\", (db_user, db_password))"
|
||||
)
|
||||
print_success("已修复ALTER USER语句")
|
||||
|
||||
# 写入修复后的内容
|
||||
with open("setup_environment.py", "w", encoding="utf-8") as f:
|
||||
f.write(fixed_content)
|
||||
|
||||
print_success("MySQL连接问题修复完成")
|
||||
return True
|
||||
except Exception as e:
|
||||
print_error(f"修复过程出错: {str(e)}")
|
||||
try:
|
||||
if os.path.exists(backup_file):
|
||||
shutil.copy2(backup_file, "setup_environment.py")
|
||||
print_warning("已恢复原始文件")
|
||||
except:
|
||||
pass
|
||||
return False
|
||||
|
||||
def main():
|
||||
print_title("Cursor自动化服务 - 修复工具")
|
||||
|
||||
# 设置虚拟环境
|
||||
if not setup_virtual_env():
|
||||
# 如果创建了虚拟环境并生成了安装脚本，直接退出
|
||||
sys.exit(0)
|
||||
|
||||
# 修复MySQL格式化问题
|
||||
if not fix_mysql_format_issue():
|
||||
print_error("修复失败，请手动检查setup_environment.py文件")
|
||||
sys.exit(1)
|
||||
|
||||
print_title("修复完成")
|
||||
print("现在可以安全地运行设置向导:")
|
||||
print(" python setup_environment.py")
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
main()
|
||||
except KeyboardInterrupt:
|
||||
print("\n\n操作已取消")
|
||||
sys.exit(1)
|
||||
@@ -14,15 +14,6 @@ import random
|
||||
import string
|
||||
from typing import Dict, Any, Optional, Tuple
|
||||
|
||||
# 尝试导入必要的库，如果不存在则安装
|
||||
required_packages = [
|
||||
"loguru",
|
||||
"pymysql",
|
||||
"aiomysql",
|
||||
"redis",
|
||||
"pyyaml"
|
||||
]
|
||||
|
||||
# 设置颜色输出
|
||||
class Colors:
|
||||
HEADER = '\033[95m'
|
||||
@@ -61,6 +52,97 @@ def print_error(message):
|
||||
"""打印错误信息"""
|
||||
print(f"{Colors.RED}✗ {message}{Colors.ENDC}")
|
||||
|
||||
# 检测虚拟环境
|
||||
def is_in_virtualenv():
|
||||
"""检查是否在虚拟环境中运行"""
|
||||
return hasattr(sys, 'real_prefix') or (hasattr(sys, 'base_prefix') and sys.base_prefix != sys.prefix)
|
||||
|
||||
def setup_virtualenv():
|
||||
"""设置虚拟环境"""
|
||||
print_title("虚拟环境检测")
|
||||
|
||||
if is_in_virtualenv():
|
||||
print_success("已在虚拟环境中运行")
|
||||
return True
|
||||
|
||||
print_warning("当前不在虚拟环境中运行")
|
||||
create_venv = input("是否创建并使用虚拟环境? (推荐) (y/n, 默认: y): ").strip().lower()
|
||||
|
||||
if create_venv == 'n':
|
||||
print_warning("跳过虚拟环境创建，将直接在系统Python环境中安装依赖")
|
||||
return True
|
||||
|
||||
venv_path = input("请输入虚拟环境路径 (默认: ./venv): ").strip() or "./venv"
|
||||
|
||||
try:
|
||||
# 检查venv模块
|
||||
try:
|
||||
import venv
|
||||
has_venv = True
|
||||
except ImportError:
|
||||
has_venv = False
|
||||
|
||||
if not has_venv:
|
||||
print_warning("Python venv模块不可用，尝试安装...")
|
||||
if sys.platform.startswith('linux'):
|
||||
print("在Linux上安装venv模块...")
|
||||
try:
|
||||
subprocess.check_call(["sudo", "apt", "install", "-y", "python3-venv", "python3-full"])
|
||||
except:
|
||||
try:
|
||||
subprocess.check_call(["sudo", "yum", "install", "-y", "python3-venv"])
|
||||
except:
|
||||
print_error("无法自动安装venv模块，请手动安装后重试")
|
||||
print_warning("Ubuntu/Debian: sudo apt install python3-venv python3-full")
|
||||
print_warning("CentOS/RHEL: sudo yum install python3-venv")
|
||||
return False
|
||||
else:
|
||||
print_error("请安装Python venv模块后重试")
|
||||
return False
|
||||
|
||||
# 创建虚拟环境
|
||||
print_step("创建", f"正在创建虚拟环境: {venv_path}")
|
||||
if sys.platform.startswith('win'):
|
||||
subprocess.check_call([sys.executable, "-m", "venv", venv_path])
|
||||
else:
|
||||
subprocess.check_call([sys.executable, "-m", "venv", venv_path])
|
||||
|
||||
# 计算激活脚本路径
|
||||
if sys.platform.startswith('win'):
|
||||
activate_script = os.path.join(venv_path, "Scripts", "activate.bat")
|
||||
python_path = os.path.join(venv_path, "Scripts", "python.exe")
|
||||
else:
|
||||
activate_script = os.path.join(venv_path, "bin", "activate")
|
||||
python_path = os.path.join(venv_path, "bin", "python")
|
||||
|
||||
print_success(f"虚拟环境创建成功: {venv_path}")
|
||||
print_warning("请退出当前程序，然后执行以下命令激活虚拟环境并重新运行:")
|
||||
|
||||
if sys.platform.startswith('win'):
|
||||
print(f"\n{Colors.BOLD}Windows:{Colors.ENDC}")
|
||||
print(f" {activate_script}")
|
||||
print(f" python setup_environment.py")
|
||||
else:
|
||||
print(f"\n{Colors.BOLD}Linux/Mac:{Colors.ENDC}")
|
||||
print(f" source {activate_script}")
|
||||
print(f" python setup_environment.py")
|
||||
|
||||
return False
|
||||
|
||||
except Exception as e:
|
||||
print_error(f"创建虚拟环境失败: {str(e)}")
|
||||
return False
|
||||
|
||||
# 尝试导入必要的库，如果不存在则安装
|
||||
required_packages = [
|
||||
"loguru",
|
||||
"pymysql",
|
||||
"aiomysql",
|
||||
"redis",
|
||||
"pyyaml",
|
||||
"cryptography" # 添加这一行
|
||||
]
|
||||
|
||||
def check_and_install_packages():
|
||||
"""检查并安装必要的Python包"""
|
||||
print_step(1, "检查并安装必要的Python包")
|
||||
|
||||
225
setup_patch.py
Normal file
225
setup_patch.py
Normal file
@@ -0,0 +1,225 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Cursor自动化服务 - 设置脚本补丁
|
||||
修复MySQL连接和添加虚拟环境支持
|
||||
"""
|
||||
import os
|
||||
import sys
|
||||
import subprocess
|
||||
import shutil
|
||||
|
||||
# 彩色输出
|
||||
class Colors:
|
||||
GREEN = '\033[92m'
|
||||
YELLOW = '\033[93m'
|
||||
RED = '\033[91m'
|
||||
ENDC = '\033[0m'
|
||||
BOLD = '\033[1m'
|
||||
|
||||
def print_title(title):
|
||||
"""打印标题"""
|
||||
print(f"\n{Colors.BOLD}{Colors.GREEN}{'='*60}{Colors.ENDC}")
|
||||
print(f"{Colors.BOLD}{Colors.GREEN}{title.center(60)}{Colors.ENDC}")
|
||||
print(f"{Colors.BOLD}{Colors.GREEN}{'='*60}{Colors.ENDC}\n")
|
||||
|
||||
def print_success(message):
|
||||
"""打印成功信息"""
|
||||
print(f"{Colors.GREEN}✓ {message}{Colors.ENDC}")
|
||||
|
||||
def print_warning(message):
|
||||
"""打印警告信息"""
|
||||
print(f"{Colors.YELLOW}⚠ {message}{Colors.ENDC}")
|
||||
|
||||
def print_error(message):
|
||||
"""打印错误信息"""
|
||||
print(f"{Colors.RED}✗ {message}{Colors.ENDC}")
|
||||
|
||||
def is_in_virtualenv():
|
||||
"""检查是否在虚拟环境中运行"""
|
||||
return hasattr(sys, 'real_prefix') or (hasattr(sys, 'base_prefix') and sys.base_prefix != sys.prefix)
|
||||
|
||||
def setup_virtualenv():
|
||||
"""设置虚拟环境"""
|
||||
print_title("虚拟环境设置")
|
||||
|
||||
if is_in_virtualenv():
|
||||
print_success("已在虚拟环境中运行")
|
||||
return True
|
||||
|
||||
print_warning("当前不在虚拟环境中运行")
|
||||
create_venv = input("是否创建并使用虚拟环境? (推荐) (y/n, 默认: y): ").strip().lower()
|
||||
|
||||
if create_venv == 'n':
|
||||
print_warning("跳过虚拟环境创建，将直接在系统Python环境中安装依赖")
|
||||
return True
|
||||
|
||||
venv_path = input("请输入虚拟环境路径 (默认: ./venv): ").strip() or "./venv"
|
||||
|
||||
try:
|
||||
# 检查venv模块
|
||||
try:
|
||||
subprocess.check_call([sys.executable, "-m", "venv", "--help"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
||||
has_venv = True
|
||||
except (subprocess.SubprocessError, ImportError):
|
||||
has_venv = False
|
||||
|
||||
if not has_venv:
|
||||
print_warning("Python venv模块不可用，尝试安装...")
|
||||
if sys.platform.startswith('linux'):
|
||||
print("在Linux上安装venv模块...")
|
||||
try:
|
||||
subprocess.check_call(["sudo", "apt", "install", "-y", "python3-venv", "python3-full"], stdout=subprocess.DEVNULL)
|
||||
print_success("venv模块安装成功")
|
||||
except:
|
||||
try:
|
||||
subprocess.check_call(["sudo", "yum", "install", "-y", "python3-venv"], stdout=subprocess.DEVNULL)
|
||||
print_success("venv模块安装成功")
|
||||
except:
|
||||
print_error("无法自动安装venv模块，请手动安装后重试")
|
||||
print_warning("Ubuntu/Debian: sudo apt install python3-venv python3-full")
|
||||
print_warning("CentOS/RHEL: sudo yum install python3-venv")
|
||||
return False
|
||||
else:
|
||||
print_error("请安装Python venv模块后重试")
|
||||
return False
|
||||
|
||||
# 创建虚拟环境
|
||||
print("\n正在创建虚拟环境...")
|
||||
subprocess.check_call([sys.executable, "-m", "venv", venv_path])
|
||||
|
||||
# 计算激活脚本路径
|
||||
if sys.platform.startswith('win'):
|
||||
activate_script = os.path.join(venv_path, "Scripts", "activate.bat")
|
||||
pip_path = os.path.join(venv_path, "Scripts", "pip.exe")
|
||||
python_path = os.path.join(venv_path, "Scripts", "python.exe")
|
||||
else:
|
||||
activate_script = os.path.join(venv_path, "bin", "activate")
|
||||
pip_path = os.path.join(venv_path, "bin", "pip")
|
||||
python_path = os.path.join(venv_path, "bin", "python")
|
||||
|
||||
print_success(f"虚拟环境创建成功: {venv_path}")
|
||||
|
||||
# 安装依赖
|
||||
print("\n正在安装依赖包...")
|
||||
if os.path.exists("requirements.txt"):
|
||||
try:
|
||||
if sys.platform.startswith('win'):
|
||||
subprocess.check_call([pip_path, "install", "-r", "requirements.txt"])
|
||||
else:
|
||||
script = f"""#!/bin/bash
|
||||
source "{activate_script}"
|
||||
pip install -r requirements.txt
|
||||
pip install cryptography
|
||||
echo "依赖安装完成，请运行以下命令启动设置向导:"
|
||||
echo "source {activate_script}"
|
||||
echo "python setup_environment.py"
|
||||
"""
|
||||
with open("setup_venv.sh", "w") as f:
|
||||
f.write(script)
|
||||
os.chmod("setup_venv.sh", 0o755)
|
||||
print_success("已创建安装脚本: setup_venv.sh")
|
||||
print("\n请运行以下命令完成安装并启动设置向导:")
|
||||
print(f" bash ./setup_venv.sh")
|
||||
except Exception as e:
|
||||
print_error(f"安装依赖失败: {str(e)}")
|
||||
print("\n请手动运行以下命令:")
|
||||
print(f" source {activate_script}")
|
||||
print(f" pip install -r requirements.txt")
|
||||
print(f" pip install cryptography")
|
||||
print(f" python setup_environment.py")
|
||||
else:
|
||||
print_warning("未找到requirements.txt文件")
|
||||
print("\n请手动运行以下命令:")
|
||||
print(f" source {activate_script}")
|
||||
print(f" pip install loguru pymysql aiomysql redis pyyaml cryptography")
|
||||
print(f" python setup_environment.py")
|
||||
|
||||
return False
|
||||
|
||||
except Exception as e:
|
||||
print_error(f"创建虚拟环境失败: {str(e)}")
|
||||
return False
|
||||
|
||||
def patch_setup_script():
|
||||
"""修补setup_environment.py脚本"""
|
||||
print_title("修补设置脚本")
|
||||
|
||||
if not os.path.exists("setup_environment.py"):
|
||||
print_error("未找到setup_environment.py文件")
|
||||
return False
|
||||
|
||||
# 备份原始文件
|
||||
backup_file = f"setup_environment.py.bak.{int(os.path.getmtime('setup_environment.py'))}"
|
||||
try:
|
||||
shutil.copy2("setup_environment.py", backup_file)
|
||||
print_success(f"已备份原始文件: {backup_file}")
|
||||
except Exception as e:
|
||||
print_error(f"备份文件失败: {str(e)}")
|
||||
return False
|
||||
|
||||
# 修改required_packages列表
|
||||
try:
|
||||
with open("setup_environment.py", "r", encoding="utf-8") as f:
|
||||
content = f.read()
|
||||
|
||||
# 添加cryptography包
|
||||
if "cryptography" not in content:
|
||||
content = content.replace(
|
||||
'required_packages = [\n "loguru",\n "pymysql",\n "aiomysql",\n "redis",\n "pyyaml"',
|
||||
'required_packages = [\n "loguru",\n "pymysql",\n "aiomysql",\n "redis",\n "pyyaml",\n "cryptography"'
|
||||
)
|
||||
print_success("已添加cryptography包到依赖列表")
|
||||
|
||||
# 修复MySQL字符串格式化问题
|
||||
if "CREATE USER '{db_user}'@'%'" in content:
|
||||
content = content.replace(
|
||||
"cursor.execute(f\"CREATE USER '{db_user}'@'%' IDENTIFIED BY %s\", (db_password,))",
|
||||
"cursor.execute(\"CREATE USER %s@'%%' IDENTIFIED BY %s\", (db_user, db_password))"
|
||||
)
|
||||
print_success("已修复MySQL用户创建语句的格式化问题")
|
||||
|
||||
if "ALTER USER '{db_user}'@'%'" in content:
|
||||
content = content.replace(
|
||||
"cursor.execute(f\"ALTER USER '{db_user}'@'%' IDENTIFIED BY %s\", (db_password,))",
|
||||
"cursor.execute(\"ALTER USER %s@'%%' IDENTIFIED BY %s\", (db_user, db_password))"
|
||||
)
|
||||
print_success("已修复MySQL用户更新语句的格式化问题")
|
||||
|
||||
# 写回文件
|
||||
with open("setup_environment.py", "w", encoding="utf-8") as f:
|
||||
f.write(content)
|
||||
|
||||
print_success("设置脚本修补完成")
|
||||
return True
|
||||
except Exception as e:
|
||||
print_error(f"修补文件失败: {str(e)}")
|
||||
try:
|
||||
# 恢复备份
|
||||
shutil.copy2(backup_file, "setup_environment.py")
|
||||
print_warning("已恢复原始文件")
|
||||
except:
|
||||
pass
|
||||
return False
|
||||
|
||||
def main():
|
||||
"""主函数"""
|
||||
print_title("Cursor自动化服务 - 设置补丁工具")
|
||||
|
||||
# 设置虚拟环境
|
||||
if not setup_virtualenv():
|
||||
return
|
||||
|
||||
# 修补设置脚本
|
||||
if not patch_setup_script():
|
||||
return
|
||||
|
||||
print_title("补丁应用完成")
|
||||
print("现在您可以安全地运行设置向导:")
|
||||
print(" python setup_environment.py")
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
main()
|
||||
except KeyboardInterrupt:
|
||||
print("\n\n操作已取消")
|
||||
sys.exit(1)
|
||||
Reference in New Issue
Block a user