import json
import snowflake.connector


with open('config.json', 'r') as f:
    config = json.load(f)
sf_user = config['snowflake']["user"]
sf_password = config['snowflake']["password"]
sf_account = config['snowflake']["account"]
# sf_database = config['snowflake']["database"]
sf_warehouse = config['snowflake']['warehouse']


### get query data in dictionary format
def snowflake_fetch_data(sql_query, sf_database):
    ctx = snowflake.connector.connect(
        user=sf_user,
        password=sf_password,
        account=sf_account,
        database=sf_database,
        warehouse=sf_warehouse
    )
    cs = ctx.cursor(snowflake.connector.DictCursor)
    try:
        cs.execute(sql_query)
        result_dict = cs.fetchall()
    finally:
        cs.close()
    ctx.close()
    return result_dict


if __name__ == '__main__':
    pass