def make()

in tpch/make_data.py [0:0]

• 14 lines of code
• 2 McCabe index (conditional complexity)

def make(scale_factor: int, output_path: str):
    statements = [
        "install tpch",
        "load tpch",
        f"call dbgen(sf = {scale_factor})",
    ]
    execute(statements)

    statements = []
    for row in conn.execute("show tables").fetchall():
        table = row[0]
        statements.append(
            f"copy {table} to '{output_path}/{table}.parquet' (format parquet, compression zstd)"
        )
    execute(statements)