snowflake_python_connector_create_table_insert_query.py

1import snowflake.connector
2
3# Gets the version
4ctx = snowflake.connector.connect(
5    user='<user_name>',
6    password='<password>',
7    account='<account_identifier>'
8    )
9cs = ctx.cursor()
10try:
11    cs.execute("SELECT current_version()")
12    one_row = cs.fetchone()
13    print(one_row[0])
14
15    # Create a database, schema, and table
16    cs.execute("CREATE OR REPLACE DATABASE demo_db")
17    cs.execute("CREATE OR REPLACE SCHEMA demo_schema")
18    cs.execute("CREATE OR REPLACE TABLE test_table(col1 string, col2 integer)")
19
20    # Insert data
21    cs.execute("INSERT INTO test_table(col1, col2) VALUES(%s, %s)", ('test string 1', 123))
22    cs.execute("INSERT INTO test_table(col1, col2) VALUES(%s, %s)", ('test string 2', 456))
23
24    # Query the data
25    sql = "SELECT col1, col2 FROM test_table"
26    cs.execute(sql)
27
28    # Fetch and print the results
29    for (col1, col2) in cs:
30        print('{0}, {1}'.format(col1, col2))
31
32finally:
33    cs.close()
34    ctx.close()