最初一个版本，已经初步实现各种功能
主页连接 远程命令 django Gunicorn操作
This commit is contained in:
858
threads.py
Normal file
858
threads.py
Normal file
@@ -0,0 +1,858 @@
|
||||
import os
|
||||
import time
|
||||
from loguru import logger
|
||||
from PySide6.QtCore import QThread, Signal
|
||||
import paramiko
|
||||
|
||||
|
||||
class PasswordDialog:
|
||||
def __init__(self, parent=None):
|
||||
self.parent = parent
|
||||
self.password = None
|
||||
|
||||
def get_password(self, prompt="请输入sudo密码:"):
|
||||
# 这里简化实现，实际应该使用QInputDialog
|
||||
if self.parent:
|
||||
from PySide6.QtWidgets import QInputDialog
|
||||
password, ok = QInputDialog.getText(self.parent, "sudo密码", prompt,
|
||||
echo=QInputDialog.Password)
|
||||
if ok:
|
||||
self.password = password
|
||||
return password
|
||||
return None
|
||||
|
||||
|
||||
class SSHConnectionThread(QThread):
|
||||
connection_status = Signal(bool, str)
|
||||
|
||||
def __init__(self, host, username, password, port=22):
|
||||
super().__init__()
|
||||
self.host = host
|
||||
self.username = username
|
||||
self.password = password
|
||||
self.port = port
|
||||
|
||||
def run(self):
|
||||
try:
|
||||
ssh = paramiko.SSHClient()
|
||||
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
||||
ssh.connect(self.host, port=self.port, username=self.username, password=self.password, timeout=10)
|
||||
|
||||
# 测试连接
|
||||
stdin, stdout, stderr = ssh.exec_command("echo 'Connection test'")
|
||||
output = stdout.read().decode().strip()
|
||||
|
||||
if output == "Connection test":
|
||||
self.connection_status.emit(True, f"成功连接到 {self.host}")
|
||||
logger.info(f"SSH连接成功: {self.host}")
|
||||
else:
|
||||
self.connection_status.emit(False, "连接测试失败")
|
||||
logger.error("SSH连接测试失败")
|
||||
|
||||
ssh.close()
|
||||
|
||||
except Exception as e:
|
||||
error_msg = str(e)
|
||||
self.connection_status.emit(False, error_msg)
|
||||
logger.error(f"SSH连接失败: {error_msg}")
|
||||
|
||||
|
||||
class GitInstallThread(QThread):
|
||||
result_ready = Signal(bool, str)
|
||||
|
||||
def __init__(self, ssh_client):
|
||||
super().__init__()
|
||||
self.ssh_client = ssh_client
|
||||
|
||||
def run(self):
|
||||
try:
|
||||
# 检查Git是否已安装
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command("git --version")
|
||||
git_version = stdout.read().decode().strip()
|
||||
|
||||
if git_version:
|
||||
self.result_ready.emit(True, f"Git已安装: {git_version}")
|
||||
logger.info(f"Git已安装: {git_version}")
|
||||
return
|
||||
|
||||
# 安装Git
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command("sudo apt update && sudo apt install -y git")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status == 0:
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command("git --version")
|
||||
git_version = stdout.read().decode().strip()
|
||||
self.result_ready.emit(True, f"Git安装成功: {git_version}")
|
||||
logger.info(f"Git安装成功: {git_version}")
|
||||
else:
|
||||
error = stderr.read().decode()
|
||||
self.result_ready.emit(False, f"Git安装失败: {error}")
|
||||
logger.error(f"Git安装失败: {error}")
|
||||
|
||||
except Exception as e:
|
||||
error_msg = str(e)
|
||||
self.result_ready.emit(False, error_msg)
|
||||
logger.error(f"Git安装异常: {error_msg}")
|
||||
|
||||
|
||||
class GitCloneThread(QThread):
|
||||
result_ready = Signal(bool, str)
|
||||
|
||||
def __init__(self, ssh_client, git_url, project_path):
|
||||
super().__init__()
|
||||
self.ssh_client = ssh_client
|
||||
self.git_url = git_url
|
||||
self.project_path = project_path
|
||||
|
||||
def run(self):
|
||||
try:
|
||||
# 检查目录是否存在
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"ls -la {self.project_path}")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status == 0:
|
||||
self.result_ready.emit(False, f"目录 {self.project_path} 已存在")
|
||||
logger.warning(f"目录已存在: {self.project_path}")
|
||||
return
|
||||
|
||||
# 创建父目录
|
||||
parent_dir = os.path.dirname(self.project_path)
|
||||
if parent_dir:
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"mkdir -p {parent_dir}")
|
||||
|
||||
# 克隆仓库
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"cd {parent_dir} && git clone {self.git_url} {os.path.basename(self.project_path)}")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status == 0:
|
||||
self.result_ready.emit(True, f"Git克隆成功: {self.project_path}")
|
||||
logger.info(f"Git克隆成功: {self.project_path}")
|
||||
else:
|
||||
error = stderr.read().decode()
|
||||
self.result_ready.emit(False, f"Git克隆失败: {error}")
|
||||
logger.error(f"Git克隆失败: {error}")
|
||||
|
||||
except Exception as e:
|
||||
error_msg = str(e)
|
||||
self.result_ready.emit(False, error_msg)
|
||||
logger.error(f"Git克隆异常: {error_msg}")
|
||||
|
||||
|
||||
class ListDirectoryThread(QThread):
|
||||
result_ready = Signal(bool, str)
|
||||
|
||||
def __init__(self, ssh_client, path):
|
||||
super().__init__()
|
||||
self.ssh_client = ssh_client
|
||||
self.path = path
|
||||
|
||||
def run(self):
|
||||
try:
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"ls -la {self.path}")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status == 0:
|
||||
output = stdout.read().decode()
|
||||
self.result_ready.emit(True, output)
|
||||
logger.info(f"目录列表成功: {self.path}")
|
||||
else:
|
||||
error = stderr.read().decode()
|
||||
self.result_ready.emit(False, error)
|
||||
logger.error(f"目录列表失败: {error}")
|
||||
|
||||
except Exception as e:
|
||||
error_msg = str(e)
|
||||
self.result_ready.emit(False, error_msg)
|
||||
logger.error(f"目录列表异常: {error_msg}")
|
||||
|
||||
|
||||
class DeleteDirectoryThread(QThread):
|
||||
result_ready = Signal(bool, str)
|
||||
|
||||
def __init__(self, ssh_client, path):
|
||||
super().__init__()
|
||||
self.ssh_client = ssh_client
|
||||
self.path = path
|
||||
|
||||
def run(self):
|
||||
try:
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"rm -rf {self.path}")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status == 0:
|
||||
self.result_ready.emit(True, f"目录删除成功: {self.path}")
|
||||
logger.info(f"目录删除成功: {self.path}")
|
||||
else:
|
||||
error = stderr.read().decode()
|
||||
self.result_ready.emit(False, error)
|
||||
logger.error(f"目录删除失败: {error}")
|
||||
|
||||
except Exception as e:
|
||||
error_msg = str(e)
|
||||
self.result_ready.emit(False, error_msg)
|
||||
logger.error(f"目录删除异常: {error_msg}")
|
||||
|
||||
|
||||
class SetTimezoneAndRestartThread(QThread):
|
||||
result_ready = Signal(bool, str)
|
||||
|
||||
def __init__(self, ssh_client):
|
||||
super().__init__()
|
||||
self.ssh_client = ssh_client
|
||||
|
||||
def run(self):
|
||||
try:
|
||||
# 设置时区
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command("sudo timedatectl set-timezone Asia/Shanghai")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status != 0:
|
||||
error = stderr.read().decode()
|
||||
self.result_ready.emit(False, f"设置时区失败: {error}")
|
||||
logger.error(f"设置时区失败: {error}")
|
||||
return
|
||||
|
||||
# 重启服务器
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command("sudo reboot")
|
||||
|
||||
self.result_ready.emit(True, "时区设置成功，服务器正在重启")
|
||||
logger.info("时区设置成功，服务器正在重启")
|
||||
|
||||
except Exception as e:
|
||||
error_msg = str(e)
|
||||
self.result_ready.emit(False, error_msg)
|
||||
logger.error(f"时区设置异常: {error_msg}")
|
||||
|
||||
|
||||
class CheckFirewallThread(QThread):
|
||||
result_ready = Signal(bool, str)
|
||||
|
||||
def __init__(self, ssh_client):
|
||||
super().__init__()
|
||||
self.ssh_client = ssh_client
|
||||
|
||||
def run(self):
|
||||
try:
|
||||
# 检查UFW状态
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command("sudo ufw status")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status == 0:
|
||||
output = stdout.read().decode()
|
||||
self.result_ready.emit(True, output)
|
||||
logger.info("防火墙状态检查成功")
|
||||
else:
|
||||
error = stderr.read().decode()
|
||||
self.result_ready.emit(False, error)
|
||||
logger.error(f"防火墙状态检查失败: {error}")
|
||||
|
||||
except Exception as e:
|
||||
error_msg = str(e)
|
||||
self.result_ready.emit(False, error_msg)
|
||||
logger.error(f"防火墙状态检查异常: {error_msg}")
|
||||
|
||||
|
||||
class OpenPortThread(QThread):
|
||||
result_ready = Signal(bool, str)
|
||||
|
||||
def __init__(self, ssh_client, port):
|
||||
super().__init__()
|
||||
self.ssh_client = ssh_client
|
||||
self.port = port
|
||||
|
||||
def run(self):
|
||||
try:
|
||||
# 开放端口
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"sudo ufw allow {self.port}")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status == 0:
|
||||
# 重新加载防火墙
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command("sudo ufw reload")
|
||||
|
||||
self.result_ready.emit(True, f"端口 {self.port} 开放成功")
|
||||
logger.info(f"端口 {self.port} 开放成功")
|
||||
else:
|
||||
error = stderr.read().decode()
|
||||
self.result_ready.emit(False, error)
|
||||
logger.error(f"端口开放失败: {error}")
|
||||
|
||||
except Exception as e:
|
||||
error_msg = str(e)
|
||||
self.result_ready.emit(False, error_msg)
|
||||
logger.error(f"端口开放异常: {error_msg}")
|
||||
|
||||
|
||||
class DjangoInstallThread(QThread):
|
||||
result_ready = Signal(bool, str)
|
||||
progress_updated = Signal(int)
|
||||
|
||||
def __init__(self, ssh_client):
|
||||
super().__init__()
|
||||
self.ssh_client = ssh_client
|
||||
|
||||
def run(self):
|
||||
try:
|
||||
self.progress_updated.emit(10)
|
||||
|
||||
# 检查Django是否已安装
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command("python3 -m django --version")
|
||||
django_version = stdout.read().decode().strip()
|
||||
|
||||
if django_version:
|
||||
self.result_ready.emit(True, f"Django已安装: {django_version}")
|
||||
logger.info(f"Django已安装: {django_version}")
|
||||
return
|
||||
|
||||
self.progress_updated.emit(30)
|
||||
|
||||
# 尝试使用pip安装
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command("pip3 install django")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status == 0:
|
||||
self.progress_updated.emit(90)
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command("python3 -m django --version")
|
||||
django_version = stdout.read().decode().strip()
|
||||
self.result_ready.emit(True, f"Django安装成功: {django_version}")
|
||||
logger.info(f"Django安装成功: {django_version}")
|
||||
return
|
||||
|
||||
self.progress_updated.emit(50)
|
||||
|
||||
# 如果pip安装失败，尝试使用apt安装
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command("sudo apt update && sudo apt install -y python3-django")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status == 0:
|
||||
self.progress_updated.emit(90)
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command("python3 -m django --version")
|
||||
django_version = stdout.read().decode().strip()
|
||||
self.result_ready.emit(True, f"Django安装成功: {django_version}")
|
||||
logger.info(f"Django安装成功: {django_version}")
|
||||
else:
|
||||
error = stderr.read().decode()
|
||||
self.result_ready.emit(False, f"Django安装失败: {error}")
|
||||
logger.error(f"Django安装失败: {error}")
|
||||
|
||||
except Exception as e:
|
||||
error_msg = str(e)
|
||||
self.result_ready.emit(False, error_msg)
|
||||
logger.error(f"Django安装异常: {error_msg}")
|
||||
|
||||
|
||||
class DjangoTestThread(QThread):
|
||||
result_ready = Signal(bool, str)
|
||||
progress_updated = Signal(int)
|
||||
|
||||
def __init__(self, ssh_client, django_path):
|
||||
super().__init__()
|
||||
self.ssh_client = ssh_client
|
||||
self.django_path = django_path
|
||||
|
||||
def run(self):
|
||||
try:
|
||||
self.progress_updated.emit(10)
|
||||
|
||||
# 检查Django项目是否存在
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"ls -la {self.django_path}/manage.py")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status != 0:
|
||||
self.result_ready.emit(False, f"Django项目不存在: {self.django_path}")
|
||||
logger.error(f"Django项目不存在: {self.django_path}")
|
||||
return
|
||||
|
||||
self.progress_updated.emit(30)
|
||||
|
||||
# 检查requirements.txt
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"ls -la {self.django_path}/requirements.txt")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status == 0:
|
||||
self.progress_updated.emit(50)
|
||||
# 安装依赖
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"cd {self.django_path} && pip3 install -r requirements.txt")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status != 0:
|
||||
error = stderr.read().decode()
|
||||
self.result_ready.emit(False, f"依赖安装失败: {error}")
|
||||
logger.error(f"依赖安装失败: {error}")
|
||||
return
|
||||
|
||||
self.progress_updated.emit(70)
|
||||
|
||||
# 运行Django测试服务器
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"cd {self.django_path} && python3 manage.py runserver 0.0.0.0:8000 --noreload &")
|
||||
time.sleep(3) # 等待服务器启动
|
||||
|
||||
# 检查服务器是否运行
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command("ps aux | grep 'manage.py runserver'")
|
||||
output = stdout.read().decode()
|
||||
|
||||
if "manage.py runserver" in output:
|
||||
self.progress_updated.emit(100)
|
||||
self.result_ready.emit(True, "Django测试服务器启动成功")
|
||||
logger.info("Django测试服务器启动成功")
|
||||
else:
|
||||
self.result_ready.emit(False, "Django测试服务器启动失败")
|
||||
logger.error("Django测试服务器启动失败")
|
||||
|
||||
except Exception as e:
|
||||
error_msg = str(e)
|
||||
self.result_ready.emit(False, error_msg)
|
||||
logger.error(f"Django测试异常: {error_msg}")
|
||||
|
||||
|
||||
class DownloadSettingsThread(QThread):
|
||||
result_ready = Signal(bool, str)
|
||||
|
||||
def __init__(self, ssh_client, django_path):
|
||||
super().__init__()
|
||||
self.ssh_client = ssh_client
|
||||
self.django_path = django_path
|
||||
|
||||
def run(self):
|
||||
try:
|
||||
# 查找settings.py文件
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"find {self.django_path} -name settings.py")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status != 0:
|
||||
self.result_ready.emit(False, "未找到settings.py文件")
|
||||
logger.error("未找到settings.py文件")
|
||||
return
|
||||
|
||||
settings_path = stdout.read().decode().strip()
|
||||
|
||||
# 下载settings.py内容
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"cat {settings_path}")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status == 0:
|
||||
settings_content = stdout.read().decode()
|
||||
self.result_ready.emit(True, settings_content)
|
||||
logger.info("settings.py下载成功")
|
||||
else:
|
||||
error = stderr.read().decode()
|
||||
self.result_ready.emit(False, error)
|
||||
logger.error(f"settings.py下载失败: {error}")
|
||||
|
||||
except Exception as e:
|
||||
error_msg = str(e)
|
||||
self.result_ready.emit(False, error_msg)
|
||||
logger.error(f"settings.py下载异常: {error_msg}")
|
||||
|
||||
|
||||
class UploadSettingsThread(QThread):
|
||||
result_ready = Signal(bool, str)
|
||||
|
||||
def __init__(self, ssh_client, django_path, settings_content):
|
||||
super().__init__()
|
||||
self.ssh_client = ssh_client
|
||||
self.django_path = django_path
|
||||
self.settings_content = settings_content
|
||||
|
||||
def run(self):
|
||||
try:
|
||||
# 查找settings.py文件
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"find {self.django_path} -name settings.py")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status != 0:
|
||||
self.result_ready.emit(False, "未找到settings.py文件")
|
||||
logger.error("未找到settings.py文件")
|
||||
return
|
||||
|
||||
settings_path = stdout.read().decode().strip()
|
||||
|
||||
# 创建临时文件
|
||||
temp_file = "/tmp/settings_upload.py"
|
||||
sftp = self.ssh_client.open_sftp()
|
||||
|
||||
with sftp.file(temp_file, 'w') as f:
|
||||
f.write(self.settings_content)
|
||||
|
||||
# 移动到目标位置
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"sudo mv {temp_file} {settings_path}")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status == 0:
|
||||
self.result_ready.emit(True, "settings.py上传成功")
|
||||
logger.info("settings.py上传成功")
|
||||
else:
|
||||
error = stderr.read().decode()
|
||||
self.result_ready.emit(False, error)
|
||||
logger.error(f"settings.py上传失败: {error}")
|
||||
|
||||
sftp.close()
|
||||
|
||||
except Exception as e:
|
||||
error_msg = str(e)
|
||||
self.result_ready.emit(False, error_msg)
|
||||
logger.error(f"settings.py上传异常: {error_msg}")
|
||||
|
||||
|
||||
class CollectStaticThread(QThread):
|
||||
result_ready = Signal(bool, str)
|
||||
progress_updated = Signal(int)
|
||||
|
||||
def __init__(self, ssh_client, django_path):
|
||||
super().__init__()
|
||||
self.ssh_client = ssh_client
|
||||
self.django_path = django_path
|
||||
|
||||
def run(self):
|
||||
try:
|
||||
self.progress_updated.emit(10)
|
||||
|
||||
# 检查Django项目是否存在
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"ls -la {self.django_path}/manage.py")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status != 0:
|
||||
self.result_ready.emit(False, f"Django项目不存在: {self.django_path}")
|
||||
logger.error(f"Django项目不存在: {self.django_path}")
|
||||
return
|
||||
|
||||
self.progress_updated.emit(30)
|
||||
|
||||
# 收集静态文件
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"cd {self.django_path} && python3 manage.py collectstatic --noinput")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status == 0:
|
||||
self.progress_updated.emit(100)
|
||||
output = stdout.read().decode()
|
||||
self.result_ready.emit(True, f"静态文件收集成功: {output}")
|
||||
logger.info("静态文件收集成功")
|
||||
else:
|
||||
error = stderr.read().decode()
|
||||
self.result_ready.emit(False, error)
|
||||
logger.error(f"静态文件收集失败: {error}")
|
||||
|
||||
except Exception as e:
|
||||
error_msg = str(e)
|
||||
self.result_ready.emit(False, error_msg)
|
||||
logger.error(f"静态文件收集异常: {error_msg}")
|
||||
|
||||
|
||||
class CheckDjangoStatusThread(QThread):
|
||||
result_ready = Signal(bool, str)
|
||||
|
||||
def __init__(self, ssh_client, django_path):
|
||||
super().__init__()
|
||||
self.ssh_client = ssh_client
|
||||
self.django_path = django_path
|
||||
|
||||
def run(self):
|
||||
try:
|
||||
status_info = []
|
||||
|
||||
# 检查Django项目是否存在
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"ls -la {self.django_path}/manage.py")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status != 0:
|
||||
self.result_ready.emit(False, f"Django项目不存在: {self.django_path}")
|
||||
logger.error(f"Django项目不存在: {self.django_path}")
|
||||
return
|
||||
|
||||
status_info.append("✓ Django项目存在")
|
||||
|
||||
# 检查Django版本
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command("python3 -m django --version")
|
||||
django_version = stdout.read().decode().strip()
|
||||
status_info.append(f"✓ Django版本: {django_version}")
|
||||
|
||||
# 检查requirements.txt
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"ls -la {self.django_path}/requirements.txt")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status == 0:
|
||||
status_info.append("✓ requirements.txt存在")
|
||||
else:
|
||||
status_info.append("✗ requirements.txt不存在")
|
||||
|
||||
# 检查settings.py
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"find {self.django_path} -name settings.py")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status == 0:
|
||||
status_info.append("✓ settings.py存在")
|
||||
else:
|
||||
status_info.append("✗ settings.py不存在")
|
||||
|
||||
# 检查静态文件目录
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"ls -la {self.django_path}/static")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status == 0:
|
||||
status_info.append("✓ static目录存在")
|
||||
else:
|
||||
status_info.append("✗ static目录不存在")
|
||||
|
||||
status_text = "\n".join(status_info)
|
||||
self.result_ready.emit(True, status_text)
|
||||
logger.info("Django状态检查完成")
|
||||
|
||||
except Exception as e:
|
||||
error_msg = str(e)
|
||||
self.result_ready.emit(False, error_msg)
|
||||
logger.error(f"Django状态检查异常: {error_msg}")
|
||||
|
||||
|
||||
class GunicornInstallThread(QThread):
|
||||
result_ready = Signal(bool, str)
|
||||
progress_updated = Signal(int)
|
||||
|
||||
def __init__(self, ssh_client, password=None):
|
||||
super().__init__()
|
||||
self.ssh_client = ssh_client
|
||||
self.password = password
|
||||
|
||||
def run(self):
|
||||
try:
|
||||
self.progress_updated.emit(10)
|
||||
|
||||
# 检查Gunicorn是否已安装
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command("gunicorn --version")
|
||||
gunicorn_version = stdout.read().decode().strip()
|
||||
|
||||
if gunicorn_version:
|
||||
self.result_ready.emit(True, f"Gunicorn已安装: {gunicorn_version}")
|
||||
logger.info(f"Gunicorn已安装: {gunicorn_version}")
|
||||
return
|
||||
|
||||
self.progress_updated.emit(30)
|
||||
|
||||
# 尝试使用pip安装
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command("pip3 install gunicorn")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status == 0:
|
||||
self.progress_updated.emit(90)
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command("gunicorn --version")
|
||||
gunicorn_version = stdout.read().decode().strip()
|
||||
self.result_ready.emit(True, f"Gunicorn安装成功: {gunicorn_version}")
|
||||
logger.info(f"Gunicorn安装成功: {gunicorn_version}")
|
||||
return
|
||||
|
||||
self.progress_updated.emit(50)
|
||||
|
||||
# 如果pip安装失败，尝试使用apt安装
|
||||
if self.password:
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command("sudo -S apt update && sudo -S apt install -y gunicorn")
|
||||
stdin.write(f"{self.password}\n")
|
||||
stdin.flush()
|
||||
else:
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command("sudo apt update && sudo apt install -y gunicorn")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status == 0:
|
||||
self.progress_updated.emit(90)
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command("gunicorn --version")
|
||||
gunicorn_version = stdout.read().decode().strip()
|
||||
self.result_ready.emit(True, f"Gunicorn安装成功: {gunicorn_version}")
|
||||
logger.info(f"Gunicorn安装成功: {gunicorn_version}")
|
||||
else:
|
||||
error = stderr.read().decode()
|
||||
self.result_ready.emit(False, f"Gunicorn安装失败: {error}")
|
||||
logger.error(f"Gunicorn安装失败: {error}")
|
||||
|
||||
except Exception as e:
|
||||
error_msg = str(e)
|
||||
self.result_ready.emit(False, error_msg)
|
||||
logger.error(f"Gunicorn安装异常: {error_msg}")
|
||||
|
||||
|
||||
class GunicornTestThread(QThread):
|
||||
result_ready = Signal(bool, str)
|
||||
progress_updated = Signal(int)
|
||||
|
||||
def __init__(self, ssh_client, django_path):
|
||||
super().__init__()
|
||||
self.ssh_client = ssh_client
|
||||
self.django_path = django_path
|
||||
|
||||
def run(self):
|
||||
try:
|
||||
self.progress_updated.emit(10)
|
||||
|
||||
# 检查Django项目是否存在
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"ls -la {self.django_path}/manage.py")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status != 0:
|
||||
self.result_ready.emit(False, f"Django项目不存在: {self.django_path}")
|
||||
logger.error(f"Django项目不存在: {self.django_path}")
|
||||
return
|
||||
|
||||
self.progress_updated.emit(30)
|
||||
|
||||
# 检查wsgi.py文件
|
||||
project_name = os.path.basename(self.django_path)
|
||||
wsgi_path = f"{self.django_path}/{project_name}/wsgi.py"
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"ls -la {wsgi_path}")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status != 0:
|
||||
self.result_ready.emit(False, f"wsgi.py文件不存在: {wsgi_path}")
|
||||
logger.error(f"wsgi.py文件不存在: {wsgi_path}")
|
||||
return
|
||||
|
||||
self.progress_updated.emit(50)
|
||||
|
||||
# 测试Gunicorn启动
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"cd {self.django_path} && gunicorn --workers 2 --bind 0.0.0.0:8001 {project_name}.wsgi:application --timeout 10 &")
|
||||
time.sleep(5) # 等待Gunicorn启动
|
||||
|
||||
# 检查Gunicorn是否运行
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command("ps aux | grep gunicorn")
|
||||
output = stdout.read().decode()
|
||||
|
||||
if "gunicorn" in output and "--bind 0.0.0.0:8001" in output:
|
||||
# 停止测试进程
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command("pkill -f 'gunicorn --bind 0.0.0.0:8001'")
|
||||
|
||||
self.progress_updated.emit(100)
|
||||
self.result_ready.emit(True, "Gunicorn测试成功")
|
||||
logger.info("Gunicorn测试成功")
|
||||
else:
|
||||
self.result_ready.emit(False, "Gunicorn测试失败")
|
||||
logger.error("Gunicorn测试失败")
|
||||
|
||||
except Exception as e:
|
||||
error_msg = str(e)
|
||||
self.result_ready.emit(False, error_msg)
|
||||
logger.error(f"Gunicorn测试异常: {error_msg}")
|
||||
|
||||
|
||||
class UploadGunicornServiceThread(QThread):
|
||||
result_ready = Signal(bool, str)
|
||||
|
||||
def __init__(self, ssh_client, service_name, service_content, password=None):
|
||||
super().__init__()
|
||||
self.ssh_client = ssh_client
|
||||
self.service_name = service_name
|
||||
self.service_content = service_content
|
||||
self.password = password
|
||||
|
||||
def run(self):
|
||||
try:
|
||||
# 创建服务文件
|
||||
service_file = f"/etc/systemd/system/{self.service_name}.service"
|
||||
temp_file = f"/tmp/{self.service_name}.service"
|
||||
|
||||
sftp = self.ssh_client.open_sftp()
|
||||
|
||||
with sftp.file(temp_file, 'w') as f:
|
||||
f.write(self.service_content)
|
||||
|
||||
# 移动到systemd目录
|
||||
if self.password:
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"sudo -S mv {temp_file} {service_file}")
|
||||
stdin.write(f"{self.password}\n")
|
||||
stdin.flush()
|
||||
else:
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"sudo mv {temp_file} {service_file}")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status != 0:
|
||||
error = stderr.read().decode()
|
||||
self.result_ready.emit(False, f"服务文件移动失败: {error}")
|
||||
logger.error(f"服务文件移动失败: {error}")
|
||||
sftp.close()
|
||||
return
|
||||
|
||||
# 设置权限
|
||||
if self.password:
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"sudo -S chmod 644 {service_file}")
|
||||
stdin.write(f"{self.password}\n")
|
||||
stdin.flush()
|
||||
else:
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"sudo chmod 644 {service_file}")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status != 0:
|
||||
error = stderr.read().decode()
|
||||
self.result_ready.emit(False, f"权限设置失败: {error}")
|
||||
logger.error(f"权限设置失败: {error}")
|
||||
sftp.close()
|
||||
return
|
||||
|
||||
# 重新加载systemd
|
||||
if self.password:
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command("sudo -S systemctl daemon-reload")
|
||||
stdin.write(f"{self.password}\n")
|
||||
stdin.flush()
|
||||
else:
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command("sudo systemctl daemon-reload")
|
||||
|
||||
sftp.close()
|
||||
|
||||
self.result_ready.emit(True, f"服务文件上传成功: {service_file}")
|
||||
logger.info(f"服务文件上传成功: {service_file}")
|
||||
|
||||
except Exception as e:
|
||||
error_msg = str(e)
|
||||
self.result_ready.emit(False, error_msg)
|
||||
logger.error(f"服务文件上传异常: {error_msg}")
|
||||
|
||||
|
||||
class ManageGunicornServiceThread(QThread):
|
||||
result_ready = Signal(bool, str)
|
||||
|
||||
def __init__(self, ssh_client, service_name, action, password=None):
|
||||
super().__init__()
|
||||
self.ssh_client = ssh_client
|
||||
self.service_name = service_name
|
||||
self.action = action
|
||||
self.password = password
|
||||
|
||||
def run(self):
|
||||
try:
|
||||
# 构建systemd命令
|
||||
cmd = f"systemctl {self.action} {self.service_name}"
|
||||
|
||||
if self.action == "status":
|
||||
# 状态命令需要特殊处理
|
||||
if self.password:
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"sudo -S {cmd}")
|
||||
stdin.write(f"{self.password}\n")
|
||||
stdin.flush()
|
||||
else:
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"sudo {cmd}")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status == 0:
|
||||
output = stdout.read().decode()
|
||||
self.result_ready.emit(True, output)
|
||||
logger.info(f"服务状态查询成功: {self.service_name}")
|
||||
else:
|
||||
error = stderr.read().decode()
|
||||
self.result_ready.emit(False, error)
|
||||
logger.error(f"服务状态查询失败: {error}")
|
||||
else:
|
||||
# 其他操作（start, stop, restart, enable, disable）
|
||||
if self.password:
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"sudo -S {cmd}")
|
||||
stdin.write(f"{self.password}\n")
|
||||
stdin.flush()
|
||||
else:
|
||||
stdin, stdout, stderr = self.ssh_client.exec_command(f"sudo {cmd}")
|
||||
exit_status = stdout.channel.recv_exit_status()
|
||||
|
||||
if exit_status == 0:
|
||||
self.result_ready.emit(True, f"服务{self.action}成功: {self.service_name}")
|
||||
logger.info(f"服务{self.action}成功: {self.service_name}")
|
||||
else:
|
||||
error = stderr.read().decode()
|
||||
self.result_ready.emit(False, error)
|
||||
logger.error(f"服务{self.action}失败: {error}")
|
||||
|
||||
except Exception as e:
|
||||
error_msg = str(e)
|
||||
self.result_ready.emit(False, error_msg)
|
||||
logger.error(f"服务管理异常: {error_msg}")
|
||||
Reference in New Issue
Block a user