"""One-shot read-only extraction of the municipal tables from the shared hasadna.duckdb into local parquet."""
import sys, duckdb
src, out = sys.argv[1], sys.argv[2]
con = duckdb.connect(src, read_only=True)
for t in ("muni_budgets", "lamas_muni"):
    con.execute(f"COPY (SELECT * FROM budget.{t}) TO '{out}/{t}.parquet' (FORMAT parquet)")
    print(t, con.execute(f"select count(*) from budget.{t}").fetchone())
# column docs for these
con.execute(f"COPY (SELECT * FROM meta.column_docs WHERE table_name in ('muni_budgets','lamas_muni')) TO '{out}/column_docs.parquet' (FORMAT parquet)")
con.close()