195 lines
7.3 KiB
Python
195 lines
7.3 KiB
Python
from PySide6.QtWidgets import (
|
|
QApplication,
|
|
QFileDialog,
|
|
QMainWindow,
|
|
QPushButton,
|
|
QVBoxLayout,
|
|
QWidget,
|
|
QListView,
|
|
QAbstractItemView,
|
|
QTreeView,
|
|
QTableView,
|
|
QLabel,
|
|
QLineEdit,
|
|
QHBoxLayout,
|
|
QMessageBox,
|
|
QProgressBar, # 导入 QProgressBar 类
|
|
)
|
|
from PySide6.QtWidgets import QHeaderView
|
|
from PySide6.QtGui import QStandardItemModel, QStandardItem
|
|
from PySide6.QtCore import Slot # 导入 Slot 装饰器
|
|
import core
|
|
|
|
|
|
class FolderSelectorApp(QMainWindow):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.setWindowTitle("照片整理")
|
|
self.setGeometry(100, 100, 600, 400)
|
|
|
|
# 创建按钮
|
|
# TODO: 暂时不用选择文件夹按钮
|
|
self.select_folders_button = QPushButton("选择多个文件夹", None)
|
|
self.select_folders_button.clicked.connect(self.select_folders)
|
|
|
|
self.select_files_button = QPushButton("选择多个文件", self)
|
|
self.select_files_button.clicked.connect(self.select_files)
|
|
|
|
# 整理照片按钮
|
|
self.arrage_button = QPushButton("开始整理", self)
|
|
self.arrage_button.clicked.connect(self.arrage_pic)
|
|
|
|
# 创建文件开始计数
|
|
self.start_index_label = QLabel("目录开始基数：", self)
|
|
self.start_index_text = QLineEdit(self)
|
|
self.start_index_text.setText("0")
|
|
|
|
# 创建QTableView
|
|
self.table_view = QTableView(self)
|
|
self.model = QStandardItemModel(0, 2, self)
|
|
self.model.setHorizontalHeaderLabels(["文件夹名称", "路径"])
|
|
self.table_view.setModel(self.model)
|
|
self.table_view.horizontalHeader().setSectionResizeMode(0, QHeaderView.Fixed)
|
|
self.table_view.setColumnWidth(0, int(self.width() * 0.3))
|
|
self.table_view.horizontalHeader().setSectionResizeMode(1, QHeaderView.Stretch)
|
|
|
|
# 创建进度条
|
|
self.progress_bar = QProgressBar(self)
|
|
self.progress_bar.setValue(0)
|
|
|
|
# 布局
|
|
# 设置计数的布局
|
|
start_index_layout = QHBoxLayout()
|
|
start_index_layout.addWidget(self.start_index_label)
|
|
start_index_layout.addWidget(self.start_index_text)
|
|
start_index_layout.addStretch(2)
|
|
# 选择文件的布局
|
|
layout = QVBoxLayout()
|
|
# layout.addWidget(self.select_folders_button)
|
|
layout.addLayout(start_index_layout)
|
|
layout.addWidget(self.select_files_button)
|
|
layout.addWidget(self.arrage_button)
|
|
layout.addWidget(self.table_view)
|
|
layout.addWidget(self.progress_bar) # 添加进度条到布局
|
|
|
|
container = QWidget(self)
|
|
# container.setLayout(start_index_layout)
|
|
container.setLayout(layout)
|
|
|
|
# container.layout.addLayout(start_index_layout)
|
|
self.setCentralWidget(container)
|
|
|
|
self.worker = None # 初始化 worker 为 None
|
|
|
|
def select_folders(self):
|
|
# 打开文件夹选择对话框，允许多选
|
|
dialog = QFileDialog(self)
|
|
dialog.setWindowTitle("请选择多个文件夹")
|
|
dialog.setFileMode(QFileDialog.Directory) # 设置为目录选择模式
|
|
dialog.setOption(
|
|
QFileDialog.DontUseNativeDialog, True
|
|
) # 使用Qt对话框以支持多选
|
|
dialog.setOption(QFileDialog.ShowDirsOnly, True) # 只显示文件夹
|
|
dialog.setOption(QFileDialog.DontResolveSymlinks, True) # 不解析符号链接
|
|
|
|
# 设置列表视图和树视图为多选模式
|
|
for view in dialog.findChildren(QListView) + dialog.findChildren(QTreeView):
|
|
view.setSelectionMode(QAbstractItemView.ExtendedSelection)
|
|
# 设置样式表，使选中项的背景色为蓝色
|
|
view.setStyleSheet(
|
|
"""QAbstractItemView::item:selected {
|
|
background-color: #0078D7;
|
|
color: white;
|
|
}
|
|
QAbstractItemView::item:selected:!active {
|
|
background-color: #CCE8FF;
|
|
color: black;
|
|
}"""
|
|
)
|
|
|
|
if dialog.exec():
|
|
folder_paths = dialog.selectedFiles() # 获取选择的文件夹路径
|
|
print("选择的文件夹路径:", folder_paths)
|
|
|
|
# 清空当前模型
|
|
self.model.removeRows(0, self.model.rowCount())
|
|
|
|
# 添加新的文件夹路径到模型
|
|
for path in folder_paths:
|
|
folder_name = (
|
|
path.split("/")[-1] if "/" in path else path.split("\\")[-1]
|
|
)
|
|
name_item = QStandardItem(folder_name)
|
|
path_item = QStandardItem(path)
|
|
self.model.appendRow([name_item, path_item])
|
|
|
|
@Slot(int) # 使用 Slot 装饰器
|
|
def update_progress(self, value):
|
|
"""更新进度条的值"""
|
|
self.progress_bar.setValue(value)
|
|
|
|
def arrage_pic(self):
|
|
# 获取table_view中第一列的数据
|
|
name_list = []
|
|
for row in range(self.model.rowCount()):
|
|
item = self.model.item(row, 1)
|
|
if item is not None:
|
|
name_list.append(item.text())
|
|
if len(name_list) == 0:
|
|
QMessageBox.warning(self, "警告", "请选择文件!")
|
|
return
|
|
|
|
# 设置进度条的最大值
|
|
self.progress_bar.setMaximum(len(name_list))
|
|
self.progress_bar.setValue(0)
|
|
|
|
# 创建 Worker 实例
|
|
self.worker = core.Worker(
|
|
name_list, start_index=int(self.start_index_text.text())
|
|
)
|
|
self.worker.progress.connect(self.update_progress) # 连接信号到槽函数
|
|
self.worker.finished.connect(self.worker.deleteLater) # 任务完成后删除 worker
|
|
self.worker.start() # 启动线程
|
|
|
|
def resizeEvent(self, event):
|
|
super().resizeEvent(event)
|
|
self.table_view.setColumnWidth(0, int(self.width() * 0.3))
|
|
|
|
def select_files(self):
|
|
# 打开文件选择对话框，允许多选
|
|
dialog = QFileDialog(self)
|
|
dialog.setWindowTitle("请选择多个文件")
|
|
dialog.setFileMode(QFileDialog.ExistingFiles) # 设置为文件选择模式
|
|
dialog.setOption(
|
|
QFileDialog.DontUseNativeDialog, False
|
|
) # 使用Qt对话框以支持多选
|
|
|
|
# 设置列表视图和树视图为多选模式
|
|
for view in dialog.findChildren(QListView) + dialog.findChildren(QTreeView):
|
|
view.setSelectionMode(QAbstractItemView.ExtendedSelection)
|
|
# 设置样式表，使选中项的背景色为蓝色
|
|
view.setStyleSheet(
|
|
"""QAbstractItemView::item:selected {
|
|
background-color: #0078D7;
|
|
color: white;
|
|
}
|
|
QAbstractItemView::item:selected:!active {
|
|
background-color: #CCE8FF;
|
|
color: black;
|
|
}"""
|
|
)
|
|
|
|
if dialog.exec():
|
|
file_paths = dialog.selectedFiles() # 获取选择的文件路径
|
|
print("选择的文件路径:", file_paths)
|
|
|
|
# 清空当前模型
|
|
self.model.removeRows(0, self.model.rowCount())
|
|
|
|
# 添加新的文件路径到模型
|
|
for path in file_paths:
|
|
file_name = path.split("/")[-1] if "/" in path else path.split("\\")[-1]
|
|
name_item = QStandardItem(file_name)
|
|
path_item = QStandardItem(path)
|
|
self.model.appendRow([name_item, path_item])
|