from PyQt5.QtCore import Qt, QThread, pyqtSignal from PyQt5.QtWidgets import QVBoxLayout, QHBoxLayout, QWidget from qfluentwidgets import SubtitleLabel, BodyLabel, HorizontalSeparator, PushButton, TreeWidget, ProgressBar, Dialog, InfoBar, InfoBarPosition, FluentIcon import requests import shutil import os from urllib.parse import quote class DownloadThread(QThread): progressChanged = pyqtSignal(int) finished = pyqtSignal(list, list) # success, fail def __init__(self, files, server_url, secret_key, local_dir, parent=None): super().__init__(parent) self.files = files self.server_url = server_url self.secret_key = secret_key self.local_dir = local_dir def run(self): success, fail = [], [] total = len(self.files) max_retry = 3 for idx, rel_path in enumerate(self.files): retry = 0 while retry < max_retry: try: rel_path_unix = rel_path.replace("\\", "/") encoded_path = quote(rel_path_unix) url = f"{self.server_url}/download/{encoded_path}" params = {"key": self.secret_key} resp = requests.get(url, params=params, stream=True, timeout=10) if resp.status_code == 200: local_path = os.path.join(self.local_dir, rel_path) os.makedirs(os.path.dirname(local_path), exist_ok=True) with open(local_path, "wb") as f: shutil.copyfileobj(resp.raw, f) success.append(rel_path) break else: retry += 1 except Exception: retry += 1 else: fail.append(rel_path) self.progressChanged.emit(int((idx + 1) / total * 100)) self.finished.emit(success, fail) class PartLibraryInterface(QWidget): SERVER_URL = "http://154.37.215.220:5000" SECRET_KEY = "MRobot_Download" LOCAL_LIB_DIR = "mech_lib" def __init__(self, parent=None): super().__init__(parent=parent) self.setObjectName("partLibraryInterface") layout = QVBoxLayout(self) layout.setSpacing(16) layout.addWidget(SubtitleLabel("零件库（在线bate版）")) layout.addWidget(HorizontalSeparator()) layout.addWidget(BodyLabel("感谢重庆邮电大学整理的零件库，选择需要的文件下载到本地。（如无法使用或者下载失败，请尝试重新下载或检查网络连接）")) btn_layout = QHBoxLayout() refresh_btn = PushButton(FluentIcon.SYNC, "刷新列表") refresh_btn.clicked.connect(self.refresh_list) btn_layout.addWidget(refresh_btn) open_local_btn = PushButton(FluentIcon.FOLDER, "打开本地零件库") open_local_btn.clicked.connect(self.open_local_lib) btn_layout.addWidget(open_local_btn) btn_layout.addStretch() layout.addLayout(btn_layout) self.tree = TreeWidget(self) self.tree.setHeaderLabels(["名称", "类型"]) self.tree.setSelectionMode(self.tree.ExtendedSelection) self.tree.header().setSectionResizeMode(0, self.tree.header().Stretch) self.tree.header().setSectionResizeMode(1, self.tree.header().ResizeToContents) self.tree.setCheckedColor("#0078d4", "#2d7d9a") self.tree.setBorderRadius(8) self.tree.setBorderVisible(True) layout.addWidget(self.tree, stretch=1) download_btn = PushButton(FluentIcon.DOWNLOAD, "下载选中文件") download_btn.clicked.connect(self.download_selected_files) layout.addWidget(download_btn) self.refresh_list(first=True) def refresh_list(self, first=False): self.tree.clear() try: resp = requests.get( f"{self.SERVER_URL}/list", params={"key": self.SECRET_KEY}, timeout=5 ) resp.raise_for_status() tree = resp.json() self.populate_tree(self.tree, tree, "") if not first: InfoBar.success( title="刷新成功", content="零件库已经是最新的！", parent=self, position=InfoBarPosition.TOP, duration=2000 ) except Exception as e: InfoBar.error( title="刷新失败", content=f"获取零件库失败: {e}", parent=self, position=InfoBarPosition.TOP, duration=3000 ) def populate_tree(self, parent, node, path_prefix): from PyQt5.QtWidgets import QTreeWidgetItem for dname, dnode in node.get("dirs", {}).items(): item = QTreeWidgetItem([dname, "文件夹"]) if isinstance(parent, TreeWidget): parent.addTopLevelItem(item) else: parent.addChild(item) self.populate_tree(item, dnode, os.path.join(path_prefix, dname)) for fname in node.get("files", []): item = QTreeWidgetItem([fname, "文件"]) item.setFlags(item.flags() | Qt.ItemIsUserCheckable) item.setCheckState(0, Qt.Unchecked) item.setData(0, Qt.UserRole, os.path.join(path_prefix, fname)) if isinstance(parent, TreeWidget): parent.addTopLevelItem(item) else: parent.addChild(item) def get_checked_files(self): files = [] def _traverse(item): for i in range(item.childCount()): child = item.child(i) if child.text(1) == "文件" and child.checkState(0) == Qt.Checked: files.append(child.data(0, Qt.UserRole)) _traverse(child) root = self.tree.invisibleRootItem() for i in range(root.childCount()): _traverse(root.child(i)) return files def download_selected_files(self): files = self.get_checked_files() if not files: InfoBar.info( title="提示", content="请先勾选要下载的文件。", parent=self, position=InfoBarPosition.TOP, duration=2000 ) return self.progress_dialog = Dialog( title="正在下载", content="正在下载选中文件，请稍候...", parent=self ) self.progress_bar = ProgressBar() self.progress_bar.setValue(0) self.progress_dialog.textLayout.addWidget(self.progress_bar) self.progress_dialog.show() self.download_thread = DownloadThread( files, self.SERVER_URL, self.SECRET_KEY, self.LOCAL_LIB_DIR ) self.download_thread.progressChanged.connect(self.progress_bar.setValue) self.download_thread.finished.connect(self.on_download_finished) self.download_thread.finished.connect(self.download_thread.deleteLater) self.download_thread.start() def on_download_finished(self, success, fail): self.progress_dialog.close() msg = f"成功下载: {len(success)} 个文件\n失败: {len(fail)} 个文件" dialog = Dialog( title="下载结果", content=msg, parent=self ) open_btn = PushButton("打开文件夹") def open_folder(): folder = os.path.abspath(self.LOCAL_LIB_DIR) import platform, subprocess if platform.system() == "Darwin": subprocess.call(["open", folder]) elif platform.system() == "Windows": subprocess.call(["explorer", folder]) else: subprocess.call(["xdg-open", folder]) dialog.close() open_btn.clicked.connect(open_folder) dialog.textLayout.addWidget(open_btn) dialog.exec() def open_local_lib(self): folder = os.path.abspath(self.LOCAL_LIB_DIR) import platform, subprocess if platform.system() == "Darwin": subprocess.call(["open", folder]) elif platform.system() == "Windows": subprocess.call(["explorer", folder]) else: subprocess.call(["xdg-open", folder])