134 lines
4.8 KiB
Python
134 lines
4.8 KiB
Python
import os
|
|
import shutil
|
|
from PyQt6.QtWidgets import (
|
|
QApplication, QWidget, QLabel, QLineEdit, QPushButton, QTextEdit,
|
|
QFileDialog, QVBoxLayout, QHBoxLayout, QMessageBox
|
|
)
|
|
from PyQt6.QtCore import Qt
|
|
import sys
|
|
|
|
|
|
class FileMoverGUI(QWidget):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.setWindowTitle("配对文件移动工具")
|
|
self.setGeometry(200, 200, 600, 400)
|
|
|
|
# --- 控件初始化 ---
|
|
self.folder_label = QLabel("原始文件夹:")
|
|
self.folder_input = QLineEdit()
|
|
self.folder_btn = QPushButton("浏览")
|
|
|
|
self.dest1_label = QLabel("目标文件夹1 (.tiff):")
|
|
self.dest1_input = QLineEdit()
|
|
self.dest1_btn = QPushButton("浏览")
|
|
|
|
self.dest2_label = QLabel("目标文件夹2 (.xml):")
|
|
self.dest2_input = QLineEdit()
|
|
self.dest2_btn = QPushButton("浏览")
|
|
|
|
self.ext1_label = QLabel("扩展名1:")
|
|
self.ext1_input = QLineEdit(".tiff")
|
|
self.ext2_label = QLabel("扩展名2:")
|
|
self.ext2_input = QLineEdit(".xml")
|
|
|
|
self.run_btn = QPushButton("执行移动")
|
|
self.log_box = QTextEdit()
|
|
self.log_box.setReadOnly(True)
|
|
|
|
# --- 布局 ---
|
|
layout = QVBoxLayout()
|
|
|
|
def hline(label, edit, btn):
|
|
box = QHBoxLayout()
|
|
box.addWidget(label)
|
|
box.addWidget(edit)
|
|
box.addWidget(btn)
|
|
return box
|
|
|
|
layout.addLayout(hline(self.folder_label, self.folder_input, self.folder_btn))
|
|
layout.addLayout(hline(self.dest1_label, self.dest1_input, self.dest1_btn))
|
|
layout.addLayout(hline(self.dest2_label, self.dest2_input, self.dest2_btn))
|
|
|
|
ext_box = QHBoxLayout()
|
|
ext_box.addWidget(self.ext1_label)
|
|
ext_box.addWidget(self.ext1_input)
|
|
ext_box.addWidget(self.ext2_label)
|
|
ext_box.addWidget(self.ext2_input)
|
|
layout.addLayout(ext_box)
|
|
|
|
layout.addWidget(self.run_btn)
|
|
layout.addWidget(QLabel("日志输出："))
|
|
layout.addWidget(self.log_box)
|
|
self.setLayout(layout)
|
|
|
|
# --- 信号连接 ---
|
|
self.folder_btn.clicked.connect(lambda: self.select_folder(self.folder_input))
|
|
self.dest1_btn.clicked.connect(lambda: self.select_folder(self.dest1_input))
|
|
self.dest2_btn.clicked.connect(lambda: self.select_folder(self.dest2_input))
|
|
self.run_btn.clicked.connect(self.run_script)
|
|
|
|
# 文件夹选择
|
|
def select_folder(self, target_input):
|
|
folder = QFileDialog.getExistingDirectory(self, "选择文件夹")
|
|
if folder:
|
|
target_input.setText(folder)
|
|
|
|
# 输出日志到窗口
|
|
def log(self, text):
|
|
self.log_box.append(text)
|
|
self.log_box.verticalScrollBar().setValue(self.log_box.verticalScrollBar().maximum())
|
|
QApplication.processEvents()
|
|
|
|
# 主逻辑（原 split_and_move_paired 函数的 GUI 版本）
|
|
def run_script(self):
|
|
folder = self.folder_input.text().strip()
|
|
dest1 = self.dest1_input.text().strip()
|
|
dest2 = self.dest2_input.text().strip()
|
|
ext1 = self.ext1_input.text().strip()
|
|
ext2 = self.ext2_input.text().strip()
|
|
|
|
if not all([folder, dest1, dest2, ext1, ext2]):
|
|
QMessageBox.warning(self, "输入错误", "请填写所有路径和扩展名。")
|
|
return
|
|
|
|
if not os.path.isdir(folder):
|
|
QMessageBox.critical(self, "错误", "原始文件夹无效。")
|
|
return
|
|
|
|
os.makedirs(dest1, exist_ok=True)
|
|
os.makedirs(dest2, exist_ok=True)
|
|
|
|
files_ext1 = {os.path.splitext(f)[0] for f in os.listdir(folder) if f.lower().endswith(ext1.lower())}
|
|
files_ext2 = {os.path.splitext(f)[0] for f in os.listdir(folder) if f.lower().endswith(ext2.lower())}
|
|
paired_files = files_ext1 & files_ext2
|
|
|
|
moved_count = 0
|
|
self.log_box.clear()
|
|
self.log("开始移动配对文件...\n")
|
|
|
|
for name in paired_files:
|
|
src1 = os.path.join(folder, name + ext1)
|
|
src2 = os.path.join(folder, name + ext2)
|
|
dst1 = os.path.join(dest1, name + ext1)
|
|
dst2 = os.path.join(dest2, name + ext2)
|
|
try:
|
|
shutil.move(src1, dst1)
|
|
shutil.move(src2, dst2)
|
|
moved_count += 1
|
|
self.log(f"✅ 已移动配对: {name}{ext1} 与 {name}{ext2}")
|
|
except Exception as e:
|
|
self.log(f"⚠️ 移动失败 {name}: {e}")
|
|
|
|
self.log(f"\n总计成功移动 {moved_count} 对文件。")
|
|
self.log("剩余未配对文件已留在原目录中。")
|
|
|
|
QMessageBox.information(self, "完成", f"成功移动 {moved_count} 对文件。")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app = QApplication(sys.argv)
|
|
window = FileMoverGUI()
|
|
window.show()
|
|
sys.exit(app.exec())
|