-
Notifications
You must be signed in to change notification settings - Fork 16.3k
Description
Apache Airflow Provider(s)
snowflake
Versions of Apache Airflow Providers
apache-airflow-providers-snowflake==2.3.0
Apache Airflow version
2.2.2
Operating System
Amazon Linux AMI
Deployment
MWAA
Deployment details
No response
What happened
If the sql parameter is an empty list, the execution_info list variable is attempted to be returned when it hasn't been initialized.
The execution_info variable is defined only within parsing through each sql query, so if the sql queries list is empty, it never gets defined.
[...]
snowflake_hook.run(sql=queries, autocommit=True)
File "/usr/local/airflow/.local/lib/python3.7/site-packages/airflow/providers/snowflake/hooks/snowflake.py", line 304, in run
return execution_info
UnboundLocalError: local variable 'execution_info' referenced before assignment
What you think should happen instead
The function could either return an empty list or None.
Perhaps the execution_info variable definition could just be moved further up in the function definition so that returning it at the end doesn't raise issues.
Or, there should be a check in the run implementation to see if the sql argument is empty or not, and appropriately handle what to return from there.
How to reproduce
Pass an empty list to the sql argument when calling SnowflakeHook.run().
Anything else
My script that utilizes the SnowflakeHook.run() function is automated in a way where there isn't always a case that there are sql queries to run.
Of course, on my end I would update my code to first check if the sql queries list is populated before calling the hook to run.
However, it would save for unintended exceptions if the hook's run() function also appropriately handles what gets returned in the event that the sql argument is empty.
Are you willing to submit PR?
- Yes I am willing to submit a PR!
Code of Conduct
- I agree to follow this project's Code of Conduct