fix:添加UI界面，完善功能

2025-12-11 11:16:09 +08:00
parent 81e3c40abb
commit f437842a81
17 changed files with 939 additions and 406 deletions
--- a/utils/file_utils.py
+++ b/utils/file_utils.py
@@ -0,0 +1,121 @@
+import shutil
+import os
+import time
+from loguru import logger
+
+
+def export_data_folder(source_folder="data", output_folder="backup"):
+    """
+    将指定文件夹压缩为 zip 包
+    :param source_folder: 要压缩的文件夹路径 (默认 'data')
+    :param output_folder: 压缩包存放的文件夹路径 (默认 'backup')
+    """
+    try:
+        # 1. 检查源文件夹是否存在
+        if not os.path.exists(source_folder):
+            logger.error(f"备份失败: 找不到源文件夹 '{source_folder}'")
+            return
+
+        # 2. 确保输出目录存在
+        if not os.path.exists(output_folder):
+            os.makedirs(output_folder)
+            logger.info(f"创建备份目录: {output_folder}")
+
+        # 3. 生成带时间戳的文件名 (例如: data_backup_20251211_103000)
+        timestamp = time.strftime("%Y%m%d_%H%M%S")
+        # 注意: make_archive 不需要写后缀 .zip，它会自动加
+        base_name = os.path.join(output_folder, f"data_备份_{timestamp}")
+
+        logger.info(f"正在压缩 '{source_folder}' ...")
+
+        # 4. 执行压缩
+        # format="zip": 压缩格式
+        # root_dir: 要压缩的根目录的上级目录 (通常填 None)
+        # base_dir: 要压缩的具体目录
+        zip_path = shutil.make_archive(
+            base_name=base_name,
+            format="zip",
+            root_dir=os.path.dirname(
+                os.path.abspath(source_folder)
+            ),  # 这里为了安全，定位到父级
+            base_dir=os.path.basename(
+                os.path.abspath(source_folder)
+            ),  # 只压缩 data 文件夹本身
+        )
+
+        logger.success(f"✅ 备份成功! 文件已保存至:\n{zip_path}")
+        return zip_path
+
+    except Exception as e:
+        logger.error(f"❌ 备份出错: {str(e)}")
+        import traceback
+
+        logger.error(traceback.format_exc())
+
+
+def initialize_project(root_dir="."):
+    """
+    初始化项目：清空 data，重建目录，复制模板
+    :param root_dir: 项目根目录
+    """
+    # 定义路径
+    data_dir = os.path.join(root_dir, "data")
+    images_dir = os.path.join(data_dir, "images")
+    output_dir = os.path.join(root_dir, "output")
+
+    # 模板源文件路径
+    template_dir = os.path.join(root_dir, "templates")
+    src_excel = os.path.join(template_dir, "names.xlsx")
+    dst_excel = os.path.join(data_dir, "names.xlsx")
+
+    logger.info("开始初始化/重置项目...")
+
+    # --- 1. 清空 data 文件夹 ---
+    if os.path.exists(data_dir):
+        try:
+            # 暴力删除整个 data 文件夹及其内容
+            shutil.rmtree(data_dir)
+            logger.info(f"已清理旧数据: {data_dir}")
+        except PermissionError:
+            logger.error(
+                f"❌ 删除失败！请检查 '{data_dir}' 下的文件（如 Excel）是否正在被打开。"
+            )
+            return
+        except Exception as e:
+            logger.error(f"❌ 删除出错: {e}")
+            return
+    # --- 2. 清空 output_dir 文件夹 ---
+    if os.path.exists(output_dir):
+        try:
+            # 暴力删除整个 output 文件夹及其内容
+            shutil.rmtree(output_dir)
+            logger.info(f"已清理旧数据: {output_dir}")
+        except PermissionError:
+            logger.error(f"❌ 删除失败！请检查 '{output_dir}' 下的文件是否正在被打开。")
+            return
+        except Exception as e:
+            logger.error(f"❌ 删除出错: {e}")
+            return
+
+    # --- 2. 新建目录结构 ---
+    try:
+        # makedirs 会自动创建父级目录 (data 和 data/images 都会被创建)
+        os.makedirs(images_dir, exist_ok=True)
+        os.makedirs(output_dir, exist_ok=True)
+        logger.info(f"✅ 已创建目录: {images_dir}")
+    except Exception as e:
+        logger.error(f"创建目录失败: {e}")
+        return
+
+    # --- 3. 复制模板文件 ---
+    if os.path.exists(src_excel):
+        try:
+            shutil.copy(src_excel, dst_excel)
+            logger.success(f"✅ 成功从模板恢复: {dst_excel}")
+            logger.success("🎉 初始化完成！一切已恢复出厂设置。")
+        except Exception as e:
+            logger.error(f"复制模板文件失败: {e}")
+    else:
+        logger.warning(
+            f"⚠️ 警告: 模板文件不存在 ({src_excel})，data 文件夹内将没有 Excel 文件。"
+        )