import sys import json import requests from PySide6.QtWidgets import ( QApplication, QMainWindow, QWidget, QVBoxLayout, QHBoxLayout, QTabWidget, QGroupBox, QLabel, QLineEdit, QPushButton, QTextEdit, QTableWidget, QTableWidgetItem, QComboBox, QSpinBox, QFileDialog, QMessageBox, QStatusBar, QHeaderView, QFormLayout, QCheckBox ) from PySide6.QtCore import Qt, QTimer, QThread, Signal from PySide6.QtGui import QIcon # API通信类 class ApiClient: def __init__(self): self.base_url = "http://127.0.0.1:8000/api" self.token = "" def set_server(self, base_url): self.base_url = base_url.rstrip('/') def set_token(self, token): self.token = token def _get_headers(self): return { "Authorization": f"Token {self.token}", "Content-Type": "application/json" } def get_tasks(self): try: response = requests.get(f"{self.base_url}/tasks/", headers=self._get_headers()) response.raise_for_status() return response.json() except requests.exceptions.RequestException as e: raise Exception(f"获取任务失败: {str(e)}") def create_task(self, name, client_name, script, timeout_seconds): try: data = { "name": name, "client_name": client_name or None, "script": script or None, "timeout_seconds": timeout_seconds } response = requests.post(f"{self.base_url}/tasks/", headers=self._get_headers(), json=data) response.raise_for_status() return response.json() except requests.exceptions.RequestException as e: raise Exception(f"创建任务失败: {str(e)}") def claim_task(self, client_name): try: data = {"client_name": client_name} response = requests.post(f"{self.base_url}/tasks/claim/", headers=self._get_headers(), json=data) response.raise_for_status() return response.json() except requests.exceptions.RequestException as e: raise Exception(f"认领任务失败: {str(e)}") def start_task(self, task_id): try: response = requests.post(f"{self.base_url}/tasks/{task_id}/start/", headers=self._get_headers()) response.raise_for_status() return response.json() except requests.exceptions.RequestException as e: raise Exception(f"开始任务失败: {str(e)}") def complete_task(self, task_id, status, message): try: data = { "status": status, "message": message } response = requests.post(f"{self.base_url}/tasks/{task_id}/complete/", headers=self._get_headers(), json=data) response.raise_for_status() return response.json() except requests.exceptions.RequestException as e: raise Exception(f"完成任务失败: {str(e)}") def upload_result(self, task_id, client_id, status, message, result_file=None): try: headers = { "Authorization": f"Token {self.token}" } data = { "task": task_id, "client": client_id, "status": status, "message": message } files = {} if result_file: files["result_file"] = open(result_file, "rb") response = requests.post(f"{self.base_url}/task_results/", headers=headers, data=data, files=files) response.raise_for_status() return response.json() except requests.exceptions.RequestException as e: raise Exception(f"上传结果失败: {str(e)}") # 任务列表更新线程 class TaskUpdateThread(QThread): update_signal = Signal(list) error_signal = Signal(str) def __init__(self, api_client): super().__init__() self.api_client = api_client self.running = True def run(self): while self.running: try: tasks = self.api_client.get_tasks() self.update_signal.emit(tasks) except Exception as e: self.error_signal.emit(str(e)) self.sleep(5) # 每5秒更新一次 def stop(self): self.running = False self.wait() # 主应用窗口 class TaskCenterClient(QMainWindow): def __init__(self): super().__init__() self.api_client = ApiClient() self.task_update_thread = None self.init_ui() def init_ui(self): self.setWindowTitle("任务中心客户端") self.setGeometry(100, 100, 1000, 700) # 创建中心部件 central_widget = QWidget() self.setCentralWidget(central_widget) # 主布局 main_layout = QVBoxLayout(central_widget) # 创建标签页 self.tabs = QTabWidget() main_layout.addWidget(self.tabs) # 添加配置标签页 self.create_config_tab() # 添加任务列表标签页 self.create_task_list_tab() # 添加任务创建标签页 self.create_task_create_tab() # 添加任务操作标签页 self.create_task_operation_tab() # 创建状态栏 self.status_bar = QStatusBar() self.setStatusBar(self.status_bar) self.status_bar.showMessage("已连接到服务器") def create_config_tab(self): tab = QWidget() layout = QVBoxLayout(tab) # 服务器配置 server_group = QGroupBox("服务器配置") server_layout = QFormLayout() self.server_url_edit = QLineEdit("http://127.0.0.1:8000/api") server_layout.addRow(QLabel("服务器地址:"), self.server_url_edit) self.token_edit = QLineEdit() self.token_edit.setEchoMode(QLineEdit.Password) server_layout.addRow(QLabel("API Token:"), self.token_edit) # 显示/隐藏Token按钮 self.token_visibility_btn = QPushButton("显示") self.token_visibility_btn.clicked.connect(self.toggle_token_visibility) server_layout.addRow(QLabel(" "), self.token_visibility_btn) self.save_config_btn = QPushButton("保存配置") self.save_config_btn.clicked.connect(self.save_config) server_layout.addRow(QLabel(" "), self.save_config_btn) server_group.setLayout(server_layout) layout.addWidget(server_group) # 测试连接按钮 self.test_connection_btn = QPushButton("测试连接") self.test_connection_btn.clicked.connect(self.test_connection) layout.addWidget(self.test_connection_btn) # 连接状态 self.connection_status = QLabel("未连接") self.connection_status.setStyleSheet("color: red;") layout.addWidget(self.connection_status, alignment=Qt.AlignCenter) self.tabs.addTab(tab, "配置") def create_task_list_tab(self): tab = QWidget() layout = QVBoxLayout(tab) # 刷新按钮 refresh_layout = QHBoxLayout() self.refresh_tasks_btn = QPushButton("刷新任务列表") self.refresh_tasks_btn.clicked.connect(self.refresh_tasks) refresh_layout.addWidget(self.refresh_tasks_btn) self.auto_refresh_checkbox = QCheckBox("自动刷新") self.auto_refresh_checkbox.stateChanged.connect(self.toggle_auto_refresh) refresh_layout.addWidget(self.auto_refresh_checkbox) layout.addLayout(refresh_layout) # 任务列表 self.task_table = QTableWidget() self.task_table.setColumnCount(8) self.task_table.setHorizontalHeaderLabels([ "ID", "名称", "指定客户端", "实际客户端", "状态", "创建时间", "开始时间", "完成时间" ]) self.task_table.horizontalHeader().setSectionResizeMode(QHeaderView.ResizeToContents) self.task_table.horizontalHeader().setStretchLastSection(True) self.task_table.cellClicked.connect(self.on_task_selected) layout.addWidget(self.task_table) # 任务详情 self.task_detail = QTextEdit() self.task_detail.setReadOnly(True) self.task_detail.setPlaceholderText("点击任务查看详情") layout.addWidget(QLabel("任务详情:")) layout.addWidget(self.task_detail) self.tabs.addTab(tab, "任务列表") def create_task_create_tab(self): tab = QWidget() layout = QVBoxLayout(tab) # 任务创建表单 form_layout = QFormLayout() self.task_name_edit = QLineEdit() form_layout.addRow(QLabel("任务名称:"), self.task_name_edit) self.task_client_edit = QLineEdit() form_layout.addRow(QLabel("指定客户端:"), self.task_client_edit) self.task_script_edit = QTextEdit() self.task_script_edit.setPlaceholderText("输入执行脚本（可选）") form_layout.addRow(QLabel("执行脚本:"), self.task_script_edit) self.task_timeout_spin = QSpinBox() self.task_timeout_spin.setMinimum(60) self.task_timeout_spin.setMaximum(86400) self.task_timeout_spin.setValue(3600) form_layout.addRow(QLabel("超时时间（秒）:"), self.task_timeout_spin) self.create_task_btn = QPushButton("创建任务") self.create_task_btn.clicked.connect(self.create_task) form_layout.addRow(QLabel(" "), self.create_task_btn) layout.addLayout(form_layout) # 创建结果 self.create_result = QTextEdit() self.create_result.setReadOnly(True) self.create_result.setPlaceholderText("创建任务结果将显示在这里") layout.addWidget(QLabel("创建结果:")) layout.addWidget(self.create_result) self.tabs.addTab(tab, "创建任务") def create_task_operation_tab(self): tab = QWidget() layout = QVBoxLayout(tab) # 任务选择 task_select_layout = QHBoxLayout() self.selected_task_id = QLineEdit() self.selected_task_id.setPlaceholderText("输入任务ID") task_select_layout.addWidget(QLabel("任务ID:")) task_select_layout.addWidget(self.selected_task_id) self.select_from_list_btn = QPushButton("从列表选择") self.select_from_list_btn.clicked.connect(self.select_task_from_list) task_select_layout.addWidget(self.select_from_list_btn) layout.addLayout(task_select_layout) # 操作按钮组 operations_layout = QHBoxLayout() self.claim_task_btn = QPushButton("认领任务") self.claim_task_btn.clicked.connect(self.claim_task) operations_layout.addWidget(self.claim_task_btn) self.start_task_btn = QPushButton("开始任务") self.start_task_btn.clicked.connect(self.start_task) operations_layout.addWidget(self.start_task_btn) self.complete_task_btn = QPushButton("完成任务") self.complete_task_btn.clicked.connect(self.complete_task) operations_layout.addWidget(self.complete_task_btn) layout.addLayout(operations_layout) # 完成任务表单 complete_layout = QFormLayout() self.complete_status_combo = QComboBox() self.complete_status_combo.addItems(["success", "failed"]) complete_layout.addRow(QLabel("执行状态:"), self.complete_status_combo) self.complete_message_edit = QTextEdit() self.complete_message_edit.setPlaceholderText("输入执行结果消息") complete_layout.addRow(QLabel("执行消息:"), self.complete_message_edit) # 文件上传 self.result_file_edit = QLineEdit() self.result_file_edit.setReadOnly(True) browse_btn = QPushButton("浏览") browse_btn.clicked.connect(self.browse_result_file) complete_layout.addRow(QLabel("结果文件:"), self.result_file_edit) complete_layout.addRow(QLabel(" "), browse_btn) layout.addLayout(complete_layout) # 操作结果 self.operation_result = QTextEdit() self.operation_result.setReadOnly(True) self.operation_result.setPlaceholderText("操作结果将显示在这里") layout.addWidget(QLabel("操作结果:")) layout.addWidget(self.operation_result) self.tabs.addTab(tab, "任务操作") def toggle_token_visibility(self): if self.token_edit.echoMode() == QLineEdit.Password: self.token_edit.setEchoMode(QLineEdit.Normal) self.token_visibility_btn.setText("隐藏") else: self.token_edit.setEchoMode(QLineEdit.Password) self.token_visibility_btn.setText("显示") def save_config(self): server_url = self.server_url_edit.text() token = self.token_edit.text() if not server_url or not token: QMessageBox.warning(self, "警告", "服务器地址和API Token不能为空！") return # 保存配置到文件 try: config = { "server_url": server_url, "token": token } with open("client_config.json", "w") as f: json.dump(config, f) QMessageBox.information(self, "成功", "配置已保存！") self.api_client.set_server(server_url) self.api_client.set_token(token) except Exception as e: QMessageBox.critical(self, "错误", f"保存配置失败: {str(e)}") def test_connection(self): server_url = self.server_url_edit.text() token = self.token_edit.text() if not server_url or not token: QMessageBox.warning(self, "警告", "服务器地址和API Token不能为空！") return try: # 测试连接 temp_client = ApiClient() temp_client.set_server(server_url) temp_client.set_token(token) tasks = temp_client.get_tasks() QMessageBox.information(self, "成功", f"连接成功！找到 {len(tasks)} 个任务") self.connection_status.setText("已连接") self.connection_status.setStyleSheet("color: green;") except Exception as e: QMessageBox.critical(self, "错误", f"连接失败: {str(e)}") self.connection_status.setText(f"连接失败: {str(e)}") self.connection_status.setStyleSheet("color: red;") def refresh_tasks(self): try: tasks = self.api_client.get_tasks() self.update_task_table(tasks) self.status_bar.showMessage(f"刷新完成，共 {len(tasks)} 个任务") except Exception as e: QMessageBox.critical(self, "错误", str(e)) self.status_bar.showMessage(f"刷新失败: {str(e)}") def update_task_table(self, tasks): self.task_table.setRowCount(len(tasks)) for row, task in enumerate(tasks): self.task_table.setItem(row, 0, QTableWidgetItem(str(task["id"]))) self.task_table.setItem(row, 1, QTableWidgetItem(task["name"])) self.task_table.setItem(row, 2, QTableWidgetItem(task["client_name"] or "未指定")) self.task_table.setItem(row, 3, QTableWidgetItem(task["assigned_to"] or "未分配")) self.task_table.setItem(row, 4, QTableWidgetItem(task["status"])) self.task_table.setItem(row, 5, QTableWidgetItem(task["created_at"])) self.task_table.setItem(row, 6, QTableWidgetItem(task["started_at"] or "-")) self.task_table.setItem(row, 7, QTableWidgetItem(task["completed_at"] or "-")) def toggle_auto_refresh(self, state): if state == Qt.Checked: # 启动自动刷新线程 self.task_update_thread = TaskUpdateThread(self.api_client) self.task_update_thread.update_signal.connect(self.update_task_table) self.task_update_thread.error_signal.connect(self.show_thread_error) self.task_update_thread.start() self.status_bar.showMessage("已启动自动刷新") else: # 停止自动刷新线程 if self.task_update_thread: self.task_update_thread.stop() self.task_update_thread = None self.status_bar.showMessage("已停止自动刷新") def show_thread_error(self, error_msg): self.status_bar.showMessage(f"自动刷新失败: {error_msg}") def on_task_selected(self, row, column): task_id = self.task_table.item(row, 0).text() self.selected_task_id.setText(task_id) # 显示任务详情 task_name = self.task_table.item(row, 1).text() client_name = self.task_table.item(row, 2).text() assigned_to = self.task_table.item(row, 3).text() status = self.task_table.item(row, 4).text() created_at = self.task_table.item(row, 5).text() started_at = self.task_table.item(row, 6).text() completed_at = self.task_table.item(row, 7).text() detail = f"任务ID: {task_id}\n" detail += f"任务名称: {task_name}\n" detail += f"指定客户端: {client_name}\n" detail += f"实际客户端: {assigned_to}\n" detail += f"状态: {status}\n" detail += f"创建时间: {created_at}\n" detail += f"开始时间: {started_at}\n" detail += f"完成时间: {completed_at}\n" self.task_detail.setPlainText(detail) def toggle_token_visibility(self): if self.token_edit.echoMode() == QLineEdit.Password: self.token_edit.setEchoMode(QLineEdit.Normal) self.token_visibility_btn.setText("隐藏") else: self.token_edit.setEchoMode(QLineEdit.Password) self.token_visibility_btn.setText("显示") def create_task(self): name = self.task_name_edit.text() client_name = self.task_client_edit.text() script = self.task_script_edit.toPlainText() timeout = self.task_timeout_spin.value() if not name: QMessageBox.warning(self, "警告", "任务名称不能为空！") return try: result = self.api_client.create_task(name, client_name, script, timeout) self.create_result.setPlainText(json.dumps(result, indent=2, ensure_ascii=False)) QMessageBox.information(self, "成功", "任务创建成功！") # 清空表单 self.task_name_edit.clear() self.task_client_edit.clear() self.task_script_edit.clear() self.task_timeout_spin.setValue(3600) except Exception as e: self.create_result.setPlainText(str(e)) QMessageBox.critical(self, "错误", str(e)) def claim_task(self): task_id = self.selected_task_id.text() if not task_id: QMessageBox.warning(self, "警告", "请输入任务ID！") return try: result = self.api_client.claim_task(self.task_client_edit.text() or "default_client") self.operation_result.setPlainText(json.dumps(result, indent=2, ensure_ascii=False)) QMessageBox.information(self, "成功", "任务认领成功！") except Exception as e: self.operation_result.setPlainText(str(e)) QMessageBox.critical(self, "错误", str(e)) def start_task(self): task_id = self.selected_task_id.text() if not task_id: QMessageBox.warning(self, "警告", "请输入任务ID！") return try: result = self.api_client.start_task(task_id) self.operation_result.setPlainText(json.dumps(result, indent=2, ensure_ascii=False)) QMessageBox.information(self, "成功", "任务开始成功！") except Exception as e: self.operation_result.setPlainText(str(e)) QMessageBox.critical(self, "错误", str(e)) def complete_task(self): task_id = self.selected_task_id.text() if not task_id: QMessageBox.warning(self, "警告", "请输入任务ID！") return status = self.complete_status_combo.currentText() message = self.complete_message_edit.toPlainText() try: result = self.api_client.complete_task(task_id, status, message) self.operation_result.setPlainText(json.dumps(result, indent=2, ensure_ascii=False)) QMessageBox.information(self, "成功", "任务完成成功！") except Exception as e: self.operation_result.setPlainText(str(e)) QMessageBox.critical(self, "错误", str(e)) def select_task_from_list(self): selected_rows = self.task_table.selectedItems() if selected_rows: task_id = selected_rows[0].text() self.selected_task_id.setText(task_id) def browse_result_file(self): file_path, _ = QFileDialog.getOpenFileName(self, "选择结果文件") if file_path: self.result_file_edit.setText(file_path) def load_config(self): try: with open("client_config.json", "r") as f: config = json.load(f) self.server_url_edit.setText(config.get("server_url", "http://127.0.0.1:8000/api")) self.token_edit.setText(config.get("token", "")) self.api_client.set_server(config.get("server_url", "http://127.0.0.1:8000/api")) self.api_client.set_token(config.get("token", "")) except FileNotFoundError: pass # 配置文件不存在，使用默认值 except Exception as e: QMessageBox.warning(self, "警告", f"加载配置失败: {str(e)}") if __name__ == "__main__": app = QApplication(sys.argv) window = TaskCenterClient() window.load_config() window.show() sys.exit(app.exec())