SQLAlchemy和雪花查询ID

Question

在使用Snowflake connector-python时，可以获得snowflake查询Id，即来自cursor对象的sfqid属性。

当使用雪花的SQLAlchemy工具包时，有可能获得该属性吗？文档页面没有提到这一点- https://docs.snowflake.com/en/user-guide/sqlalchemy.html。

谢谢你，埃里克

Answer 1

我发现的一种方法是使用函数LAST_QUERY_ID，如下所示：

results = connection.execute("SELECT * FROM CITIBIKE_TRIPS LIMIT 1").fetchone()
query_id = connection.execute("SELECT LAST_QUERY_ID()").fetchone()
print(query_id)

我会得到类似这样的结果：

$ python test_sqlalchemy.py 
('019edb4b-0502-8a31-0000-16490cd95072',)

可能不是最理想的方式。

Answer 2

您可以使用sqlalchemy的execute方法并获得对SnowflakeCursor的引用，如下所示：

import os

from sqlalchemy import create_engine
from dotenv import load_dotenv

load_dotenv()

snowflake_username = os.getenv('SNOWFLAKE_USERNAME')
snowflake_password = os.getenv('SNOWFLAKE_PASSWORD')
snowflake_account = os.getenv('SNOWFLAKE_ACCOUNT')
snowflake_warehouse = os.getenv('SNOWFLAKE_WAREHOUSE')
snowflake_database = 'simon_db'
snowflake_schema = 'public'


if __name__ == '__main__':
    engine = create_engine(
        'snowflake://{user}:{password}@{account}/{db}/{schema}?warehouse={warehouse}'.format(
            user=snowflake_username,
            password=snowflake_password,
            account=snowflake_account,
            db=snowflake_database,
            schema=snowflake_schema,
            warehouse=snowflake_warehouse,
        )
    )

    connection = engine.connect()

    results = connection.execute("SELECT * FROM TEST_TABLE")
    queryId = results.cursor.sfqid

    print(f"queryId = {queryId}")
    print(f"results: {results.fetchone()}")

    connection.close()
    engine.dispose()

这将打印出来：

queryId = 019edbef-0000-114f-0000-0f9500612j23
results: ('n/a', '2021-01-01')