73 lines
2.5 KiB
Python
73 lines
2.5 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Export all folders to zip files in output directory
|
|
"""
|
|
|
|
import os
|
|
import zipfile
|
|
import shutil
|
|
from datetime import datetime
|
|
|
|
|
|
def export_folders_to_zip(base_dir: str = ".", output_dir: str = "output"):
|
|
"""Export all subdirectories to individual zip files"""
|
|
|
|
# Create output directory
|
|
os.makedirs(output_dir, exist_ok=True)
|
|
|
|
# Get all subdirectories
|
|
items = os.listdir(base_dir)
|
|
folders = [item for item in items if os.path.isdir(item) and not item.startswith('.') and item != output_dir]
|
|
|
|
if not folders:
|
|
print("No folders found to export.")
|
|
return
|
|
|
|
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
|
exported_count = 0
|
|
|
|
print(f"{'='*60}")
|
|
print(f"Exporting folders to ZIP")
|
|
print(f"{'='*60}")
|
|
print(f"Base directory: {os.path.abspath(base_dir)}")
|
|
print(f"Output directory: {os.path.abspath(output_dir)}")
|
|
print(f"Folders to export: {len(folders)}")
|
|
print(f"{'='*60}\n")
|
|
|
|
for folder in folders:
|
|
folder_path = os.path.join(base_dir, folder)
|
|
zip_filename = f"{folder}_{timestamp}.zip"
|
|
zip_path = os.path.join(output_dir, zip_filename)
|
|
|
|
try:
|
|
print(f" Creating: {zip_filename}")
|
|
with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
|
|
for root, dirs, files in os.walk(folder_path):
|
|
# Skip hidden directories
|
|
dirs[:] = [d for d in dirs if not d.startswith('.')]
|
|
for file in files:
|
|
if not file.startswith('.'):
|
|
file_path = os.path.join(root, file)
|
|
arcname = os.path.relpath(file_path, base_dir)
|
|
zipf.write(file_path, arcname)
|
|
print(f" [OK] {zip_filename}")
|
|
exported_count += 1
|
|
except Exception as e:
|
|
print(f" [FAIL] {folder}: {e}")
|
|
|
|
print(f"\n{'='*60}")
|
|
print(f"Done! Exported {exported_count}/{len(folders)} folders")
|
|
print(f"Output directory: {os.path.abspath(output_dir)}")
|
|
print(f"{'='*60}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import argparse
|
|
|
|
parser = argparse.ArgumentParser(description="Export all folders to zip files")
|
|
parser.add_argument("--base-dir", type=str, default=".", help="Base directory to scan")
|
|
parser.add_argument("--output-dir", type=str, default="output", help="Output directory for zip files")
|
|
|
|
args = parser.parse_args()
|
|
export_folders_to_zip(args.base_dir, args.output_dir)
|