import pathlib
import zipfile
import os

def sanitize_path(path):
    # Remove invalid Windows characters and strip trailing spaces
    invalid_chars = '<>:"/\\|?*'
    for char in invalid_chars:
        path = path.replace(char, '_')
    return path.strip()

# Paths
source_dir = pathlib.Path(r"F:\google-takeout-orig - Copy")
unzip_root = pathlib.Path(r"F:\takeout-unzipped")
unzip_root.mkdir(parents=True, exist_ok=True)

# Process each zip file
for zip_file in source_dir.glob("*.zip"):
    try:
        dest_dir = unzip_root / zip_file.stem
        dest_dir.mkdir(parents=True, exist_ok=True)

        with zipfile.ZipFile(zip_file, 'r') as zip_ref:
            for member in zip_ref.infolist():
                try:
                    # Create sanitized path
                    original_path = pathlib.Path(member.filename)
                    sanitized_parts = [sanitize_path(part) for part in original_path.parts]
                    safe_path = dest_dir.joinpath(*sanitized_parts)

                    # Create parent folders
                    safe_path.parent.mkdir(parents=True, exist_ok=True)

                    if not member.is_dir():
                        # Extract the file
                        with zip_ref.open(member) as source, open(safe_path, "wb") as target:
                            target.write(source.read())
                except Exception as e:

                    print(f"Error extracting {member.filename}: {e}")

        # print(f"Unzipped: {zip_file.name} to {dest_dir}")
    except Exception as e:
        print(f"Error processing {zip_file.name}: {e}")
print("All zip files have been processed.")