import os
import shutil
from zipfile import ZipFile


# 壓縮打包文件
def compress_pack_file(zip_path, compress_type="file", file_path_list=[], folder_path=""):
    """
    傳入的是文件路徑列表則壓縮指定文件，傳入的是文件夾則壓縮該文件夾
    :param zip_path: 需要寫入的zip文件路徑
    :param compress_type: 壓縮的類型："file":壓縮指定文件；"folder":壓縮文件夾
    :param file_path_list: 文件路徑列表
    :param folder_path: 文件夾路徑
    :return:
    """
    data_dict = {"data": "", "warning_info": "", "error_info": "", }
    try:
        if compress_type == "file":
            if not len(file_path_list):
                data_dict["error_info"] = "沒有文件需要壓縮打包"
                return data_dict

            with ZipFile(zip_path, 'w') as zip_file:
                for file_path in file_path_list:
                    if os.path.exists(file_path):
                        file_name = os.path.basename(file_path)
                        zip_file.write(file_path, file_name)
                    else:
                        data_dict["warning_info"] += f"【{file_path}】無法壓縮打包" + "\n"
        elif compress_type == "folder":
            if zip_path == folder_path:
                data_dict["error_info"] = "不能壓縮打包"
                return data_dict
            file_name_without_ext = os.path.splitext(zip_path)[0]
            shutil.make_archive(file_name_without_ext, 'zip', folder_path)
            
        else:
            data_dict["error_info"] = f"所選壓縮打包類型無效"

    except Exception as e:
        data_dict["error_info"] = f"執(zhí)行壓縮打包失敗"
        print(e)

    return data_dict


if __name__ == '__main__':
    f_path = r"D:\BusinessProject\結果文件\2023-12-01"
    z_p = r"D:\BusinessProject\結果文件\2023-12-01\壓縮文件.zip"
    f_l = [
        'D:\\BusinessProject\\結果文件\\2023-12-01\\2023-11-29-2023-11-29-官方旗艦店--2023-12-01_10：57.xlsx',
        'D:\\BusinessProject\\官方旗艦店--2023-12-01_11：00.xlsx',
        'D:\\BusinessProject\\官方旗艦店--2023-12-01_11：01.xlsx',
        'D:\\BusinessProject\\官方旗艦店--2023-12-01_11：02.xlsx',
        'D:\\BusinessProject\\臥室家具旗艦店--2023-12-01_11：01.xlsx',
        'D:\\BusinessProject\\臥室家具旗艦店--2023-12-01_11：02.xlsx']
    fl_path = 'D:\\BusinessProject\\結果文件\\2023-12-01\\2023-11-29-2023-11-29-官方旗艦店--2023-12-01_10：57.xlsx'

    # result_dict = compress_pack_file(z_p, "file", file_path_list=f_l)  # 壓縮打包文件
    result_dict = compress_pack_file(z_p, "folder", folder_path=fl_path)  # 壓縮打包文件夾
    print(result_dict)

import os


file_paths = []
f_path = r"D:\BusinessProject\結果文件\2023-12-01"
for root, dirs, files in os.walk(f_path):
    for file_name in files:
        file_path = os.path.join(root, file_name)
        file_paths.append(file_path)

print(file_paths)