import duckdb
from config.settings import RETI_CLUSTER_FILE, out_parquet


def run(con: duckdb.DuckDBPyConnection) -> int:
    src = RETI_CLUSTER_FILE.replace('\\', '/')
    out = out_parquet('reti_cluster').replace('\\', '/')

    con.execute(f"""
        COPY (
            SELECT
                codice_rete,
                rete_6chr,
                rete_2chr,
                rete_estesa,
                cluster_rete
            FROM read_csv(
                '{src}',
                delim=';',
                encoding='cp1252',
                header=true
            )
        ) TO '{out}' (FORMAT PARQUET, COMPRESSION ZSTD)
    """)
    return con.execute(f"SELECT COUNT(*) FROM '{out}'").fetchone()[0]