import os
import zipfile

def zip_directory(folder_path, zip_path, exclude_files):
    with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
        for root, dirs, files in os.walk(folder_path):
            for file in files:
                if file in exclude_files:
                    continue  # 跳过虚拟内存文件

                file_path = os.path.join(root, file)
                arcname = os.path.relpath(file_path, folder_path)
                
                try:
                    zipf.write(file_path, arcname)
                except PermissionError:
                    print(f"Permission denied: {file_path}, skipping.")
                except Exception as e:
                    print(f"Error compressing {file_path}: {e}, skipping.")

def main():
    # 需要压缩的盘符路径
    drive_to_backup = "G:\\"
    
    # 固定ZIP文件的路径和名称
    backup_dir = "H:\\Documents\\cache_backup\\"  # 使用原始字符串
    if not os.path.exists(backup_dir):
        os.makedirs(backup_dir)
    
    zip_filename = "weekly_backup.zip"  # 固定文件名
    zip_path = os.path.join(backup_dir, zip_filename)

    # 要排除的文件列表
    exclude_files = ["pagefile.sys", "swapfile.sys", "hiberfil.sys", "DumpStack.log.tmp", "TEMP_ZJG"]

    # 开始压缩
    zip_directory(drive_to_backup, zip_path, exclude_files)

if __name__ == "__main__":
    main()