import shutil import os import time from loguru import logger import zipfile import traceback def export_templates_folder(output_folder, stop_event, progress_callback=None): """ 将指定文件夹压缩为 zip 包 :param output_folder: 压缩包存放的文件夹路径 (默认 'backup') :param stop_event: 停止事件 :param progress_callback : 进度条回调 """ source_folder = "data" output_folder = output_folder if output_folder else "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 export_data(save_dir, root_dir=".", progress_callback=None): """ 导出 data 和 output 两个文件夹到同一个 zip 包中 :param save_dir: 用户在 GUI 弹窗中选择的保存目录 (例如: D:/Backup) :param root_dir: 项目根目录 (用于找到 data 和 output) :param progress_callback: 进度条回调函数，接收一个 float (0.0~1.0) """ # 1. 定义要打包的目标文件夹 targets = ["data", "output"] # 2. 检查保存目录 if not os.path.exists(save_dir): logger.error(f"保存目录不存在: {save_dir}") return None # --- 【新增步骤 A】预先计算文件总数 --- total_files = 0 for target in targets: target_abs_path = os.path.join(root_dir, target) if os.path.exists(target_abs_path): for _, _, files in os.walk(target_abs_path): total_files += len(files) logger.info(f"待压缩文件总数: {total_files}") # ------------------------------------ # 3. 生成压缩包路径 timestamp = time.strftime("%Y%m%d_%H%M%S") zip_filename = f"完整备份_{timestamp}.zip" zip_path = os.path.join(save_dir, zip_filename) logger.info(f"开始备份，目标文件: {zip_path}") try: # 4. 创建压缩包 with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf: processed_count = 0 # 当前处理的文件数 has_files = False # 标记是否真的压缩了文件 for target in targets: target_abs_path = os.path.join(root_dir, target) if not os.path.exists(target_abs_path): logger.warning(f"⚠️ 跳过: 找不到文件夹 '{target}'") continue logger.info(f"正在压缩: {target} ...") # 5. 遍历文件夹写入 ZIP for root, dirs, files in os.walk(target_abs_path): for file in files: # 获取文件的绝对路径 file_abs_path = os.path.join(root, file) # 计算相对路径 arcname = os.path.relpath(file_abs_path, root_dir) # 写入压缩包 zf.write(file_abs_path, arcname) has_files = True # 更新进度条 if progress_callback: progress_callback( processed_count + 1, total_files, "导出数据中..." ) if has_files: # 确保进度条最后能走到 100% if progress_callback: progress_callback(total_files, total_files, "导出数据成功") logger.success(f"✅ 备份成功! 文件已保存至:\n{zip_path}") return zip_path else: logger.error("❌ 备份失败: data 和 output 文件夹均为空或不存在。") if os.path.exists(zip_path): os.remove(zip_path) return None except Exception as e: logger.error(f"导出过程出错: {str(e)}") import traceback logger.error(traceback.format_exc()) return None def initialize_project(root_dir=".", progress_callback=None): """ 初始化项目：清空 data，重建目录，复制模板 :param root_dir: 项目根目录 :param progress_callback : 进度条回调 """ # 定义路径 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 文件。" ) def check_file_exists(file_path): """ 判断文件是否存在 """ return file_path and isinstance(file_path, str) and os.path.exists(file_path) def get_output_pptx_files(output_dir="output"): """ 获取 output 文件夹下所有的 pptx 文件 :param output_dir: output 文件夹路径 """ try: folder_path = os.path.abspath(output_dir) if not os.path.exists(folder_path): logger.error(f"文件夹不存在: {folder_path}") return # 获取所有 ppt/pptx 文件 files = [ f for f in os.listdir(folder_path) if f.lower().endswith((".ppt", ".pptx")) ] if not files: logger.warning("没有找到 PPT 文件") return total_count = len(files) logger.info(f"发现 {total_count} 个文件，准备开始转换...") return files except Exception as e: logger.error(f"发生未知错误: {e}") logger.error(traceback.format_exc()) def open_folder(folder_path): """ 打开指定文件夹 :param folder_path: 文件夹路径 """ try: if os.path.exists(folder_path): os.startfile(folder_path) else: logger.error(f"文件夹不存在: {folder_path}") except Exception as e: logger.error(f"打开文件夹失败: {e}")