122 lines
4.3 KiB
Python
122 lines
4.3 KiB
Python
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 文件。"
|
||
)
|