prefect_snowflake_connector_query_flow_quickstart.py

1from prefect import flow
2from prefect_snowflake import SnowflakeConnector
3from prefect_snowflake.database import snowflake_query
4
5@flow
6def example_snowflake_query_flow():
7    # The connector stores connection configuration like account, user, and credentials
8    snowflake_connector = SnowflakeConnector(
9        account="your_account_identifier",
10        user="your_username",
11        password="your_password",
12        database="your_database",
13        schema="your_schema",
14        warehouse="your_warehouse",
15        role="your_role"
16    )
17
18    # Execute a query and return the result
19    # In a real scenario, you might use a pre-configured block: 
20    # snowflake_connector = SnowflakeConnector.load("my-snowflake-connector")
21    
22    result = snowflake_query(
23        query="SELECT current_version()",
24        snowflake_connector=snowflake_connector
25    )
26    return result
27
28if __name__ == "__main__":
29    example_snowflake_query_flow()