增加python 界面。

.gitignore

@@ -2,4 +2,10 @@ bin
 Packages
 Windows API Code Pack 1.1
 obj
-Windows API Code Pack 1.1.zip
+Windows API Code Pack 1.1.zip
+.venv
+.vscode
+__pycache__
+*.whl
+pyvenv.cfg
+*suo

自动归类/core.py

@@ -0,0 +1,118 @@
+import os
+import glob
+from typing import List
+from exif import Image
+from datetime import datetime
+import os
+from PySide6.QtCore import QThread, Signal  # 导入 QThread 和 Signal 类
+
+
+# import shutil
+def get_jpg_files(file_path: str):
+    """获取当前目录下所有.jpg文件的完整路径"""
+    current_dir = os.path.abspath(file_path)
+    return glob.glob(os.path.join(current_dir, "*.jpg"))
+
+
+def get_datetime_from_image_exif(image_path: str):
+    with open(image_path, "rb") as image_f:
+        my_image = Image(image_f)
+        if hasattr(my_image, "datetime_original"):
+            return my_image.datetime_original
+        else:
+            print(my_image.list_all())
+            raise Exception(f"{image_path}该图片没有时间信息")
+
+
+def convert_to_unix_timestamp(date_time_str):
+    # 将字符串解析为 datetime 对象
+    date_time_obj = datetime.strptime(date_time_str, "%Y:%m:%d %H:%M:%S")
+
+    # 将 datetime 对象转换为 Unix 时间戳
+    unix_timestamp = date_time_obj.timestamp()
+
+    return unix_timestamp
+
+
+# 把时间戳相差在2分钟以内的分为一组
+def group_datetime(image_with_datetime: list):
+    """把时间戳相差在2分钟以内的分为一组"""
+    image_with_datetime.sort(key=lambda x: x[1])
+    grouped_images = []
+    current_group = []
+
+    for i in range(len(image_with_datetime)):
+        if i == 0:
+            current_group.append(image_with_datetime[i])
+        else:
+            # 计算当前图像与前一个图像的时间差
+            time_diff = image_with_datetime[i][1] - image_with_datetime[i - 1][1]
+            if time_diff <= 120:  # 2分钟 = 120秒
+                current_group.append(image_with_datetime[i])
+            else:
+                grouped_images.append(current_group)
+                current_group = [image_with_datetime[i]]
+
+    # 添加最后一组
+    grouped_images.append(current_group)
+    return grouped_images
+
+
+def get_file_name_before_bracket(file_path):
+    # 获取文件名
+    file_name = os.path.basename(file_path)
+
+    # 查找括号的位置
+    bracket_index = file_name.find("(")
+
+    # 如果找到了括号，返回括号前的部分
+    if bracket_index != -1:
+        return file_name[:bracket_index]
+    else:
+        # 如果没有找到括号，返回整个文件名
+        return file_name
+
+
+class Worker(QThread):
+    progress = Signal(int)  # 定义一个信号，用于传递进度
+
+    def __init__(self, images_path: List[str], start_index: int = 0):
+        super().__init__()
+        self.images_path = images_path
+        self.start_index = start_index
+
+    def run(self):
+        if len(self.images_path) == 0:
+            return
+        # 以下是接需要设置的参数
+        # image_path = r"D:/现场查看/"
+        # 提取目录路径
+        new_dir_location = os.path.join(os.path.dirname(self.images_path[0]), "整理后")
+        # start_index = 156  # 目录编号
+        # 以下是实际代码
+        # all_images = get_jpg_files(image_path)
+        all_images = self.images_path
+        image_with_datetime = []
+        for image_path in all_images:
+            image_datetime = get_datetime_from_image_exif(image_path)
+            image_with_datetime.append(
+                (image_path, convert_to_unix_timestamp(image_datetime))
+            )  # 第一个是图片路径，第二个是时间戳
+        grouped_image = group_datetime(image_with_datetime)
+        # total_files = sum(len(group) for group in grouped_image)
+        processed_files = 0
+        for group in grouped_image:
+            dir_name = get_file_name_before_bracket(group[0][0])
+            new_dir_name = f"{new_dir_location}/{self.start_index}.{dir_name}"
+            os.makedirs(new_dir_name, exist_ok=True)
+            for image in group:
+                processed_files += 1
+                self.progress.emit(processed_files)  # 发射进度信号
+                new_file_path_name = new_dir_name.replace("/", "//")
+                os.system(
+                    f'copy "{os.path.normpath(image[0])}" "{os.path.normpath(new_file_path_name)}" /Y'
+                )
+                # shutil.copy(image[0], destination_path)
+                print(f"copy {image[0]} {new_file_path_name}")
+            # print(group)
+            self.start_index += 1

自动归类/main.py

@@ -0,0 +1,12 @@
+from ui import FolderSelectorApp
+
+from PySide6.QtWidgets import (
+    QApplication,
+)
+import sys
+
+if __name__ == "__main__":
+    app = QApplication(sys.argv)
+    window = FolderSelectorApp()
+    window.show()
+    app.exec()

自动归类/ui.py

@@ -0,0 +1,194 @@
+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])