Skip to content

Instantly share code, notes, and snippets.

@zkan
Forked from ML-engineer/duckdb_bq_storage_api.py
Created July 8, 2022 11:06
Show Gist options
  • Save zkan/afce05487834101ddf3b653277339c8e to your computer and use it in GitHub Desktop.
Save zkan/afce05487834101ddf3b653277339c8e to your computer and use it in GitHub Desktop.
Read BQ table to DuckDB directly from storage read api
import duckdb
from google.cloud import bigquery
bqclient = bigquery.Client()
table = bigquery.TableReference.from_string(
"bigquery-public-data.utility_us.country_code_iso"
)
rows = bqclient.list_rows(table)
country_code_iso = rows.to_arrow(create_bqstorage_client=True)
cursor = duckdb.connect()
print(cursor.execute('SELECT * FROM country_code_iso').fetchall())
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment