def write_crs()

in data/geospatial/crs-gen.py [0:0]

• 17 lines of code
• 3 McCabe index (conditional complexity)

def write_crs(type, geometry, name, col_name="geometry", metadata=None):
    schema = pa.schema({"wkt": pa.utf8(), col_name: type})

    with parquet.ParquetWriter(
        HERE / name,
        schema,
        # Not sure if there's a way to write metadata without
        # storing the Arrow schema
        store_schema=metadata is not None,
        compression="none",
    ) as writer:
        batch = pa.record_batch(
            {
                "wkt": [geometry.wkt],
                col_name: type.wrap_array(pa.array([geometry.wkb])),
            }
        )
        writer.write_batch(batch)

        if metadata is not None:
            writer.add_key_value_metadata(metadata)