adbc vs sfc

adbc_snowflake.py

import os

import adbc_driver_manager as adm
import adbc_driver_manager.dbapi
import snowflake.connector as sfc


def adbc():
    with adm.dbapi.connect(
        driver="adbc_driver_snowflake",
        db_kwargs={
            "uri": os.environ["SNOWFLAKE_URL"][len("snowflake://") :],
        },
    ) as con:
        with con.cursor() as cur:
            cur.execute(
                'SELECT * /* adbc */ FROM "SNOWFLAKE_SAMPLE_DATA".TPCH_SF1.LINEITEM'
            )
            result = cur.fetch_arrow_table()
            print(len(result))


def scp():
    with sfc.connect(
        user=os.environ["SNOWSQL_USER"],
        password=os.environ["SNOWSQL_PWD"],
        account=os.environ["SNOWSQL_ACCOUNT"],
        database=os.environ["SNOWSQL_DATABASE"],
        schema=os.environ["SNOWSQL_SCHEMA"],
        warehouse=os.environ["SNOWSQL_WAREHOUSE"],
    ) as con:
        with con.cursor() as cur:
            cur.execute(
                'SELECT * /* snowflake-connector-python */ FROM "SNOWFLAKE_SAMPLE_DATA".TPCH_SF1.LINEITEM'
            )
            result = cur.fetch_arrow_all()
            print(len(result))


if __name__ == "__main__":
    """Uncomment one of the calls."""
    # adbc()
    # scp()