diff --git a/main.py b/main.py index 855666d..8b52950 100644 --- a/main.py +++ b/main.py @@ -10,21 +10,21 @@ def main(): load_dotenv() # only on local run print(os.environ) - queries_list = os.environ['INPUT_QUERIES'].split(';') + queries_list = filter(None, map(str.strip, os.environ['INPUT_QUERIES'].split(';'))) sync = os.environ.get("INPUT_SYNC", False) warehouse = os.environ['INPUT_SNOWFLAKE_WAREHOUSE'] snowflake_account = os.environ['INPUT_SNOWFLAKE_ACCOUNT'] snowflake_username = os.environ['INPUT_SNOWFLAKE_USERNAME'] snowflake_password = os.environ['INPUT_SNOWFLAKE_PASSWORD'] snowflake_role = os.environ.get('INPUT_SNOWFLAKE_ROLE', '') - + with SnowflakeConnector(snowflake_account, snowflake_username, snowflake_password) as con: try: if snowflake_role != '': con.set_user_role(snowflake_role) except: pass - + con.set_db_warehouse(warehouse) query_results = []