How to Mock CloudFront Invalidation Paths in Python

import argparse

def build_invalidation_paths(files, include_index=True):
    """
    Create CloudFront invalidation paths from a list of files.
    Converts file names to root-relative paths and optionally adds /index.html.
    """
    paths = []
    for f in files:
        f = f.strip()
        if not f:
            continue
        # Normalize: remove leading slash, make path start with /
        normalized = f if f.startswith("/") else f"/{f}"
        # Avoid duplicate index paths
        if include_index and normalized.endswith(".html"):
            paths.append(f"{normalized}")
            if normalized.endswith("index.html"):
                paths.append(normalized[:-len("index.html")] or "/")
            else:
                # Add implicit index fragment for directories
                paths.append(f"{normalized.replace('.html', '')}/")
        else:
            paths.append(normalized)
    return sorted(set(paths))

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("files", nargs="+", help="List of file paths")
    args = parser.parse_args()
    for path in build_invalidation_paths(args.files):
        print(path)