adbc_postgresql_driver_query_to_pyarrow_table.py

1import adbc_driver_postgresql.dbapi
2import pyarrow
3
4# The driver is initialized with a connection URI
5uri = "postgresql://localhost:5432/postgres?user=postgres&password=password"
6
7with adbc_driver_postgresql.dbapi.connect(uri) as conn:
8    with conn.cursor() as cur:
9        # Execute a simple query
10        cur.execute("SELECT 1 as col")
11        
12        # Fetch the results as a PyArrow Table
13        table = cur.fetch_arrow_table()
14        print(table)