dump_lobbying_site_view.py

# TODO: Merge this with main.py

import argparse
import logging
from pathlib import Path

from src.lobbying_site_view import dump_lobbying_site_view

logging.basicConfig(level=logging.INFO)

if __name__ == "__main__":
    parser = argparse.ArgumentParser(
        description="Dump the lobbying site view to a file."
    )
    parser.add_argument("output", type=Path, help="Output file.")
    parser.add_argument(
        "--format",
        choices=["json", "parquet", "csv"],
        help="Output format",
        required=True,
    )

    args = parser.parse_args()

    dump_lobbying_site_view(args.output, args.format)