JavierLopezT commented on a change in pull request #11350:
URL: https://github.com/apache/airflow/pull/11350#discussion_r506189681
##########
File path: airflow/providers/snowflake/hooks/snowflake.py
##########
@@ -139,3 +139,13 @@ def _get_aws_credentials(self) -> Tuple[Optional[Any],
Optional[Any]]:
def set_autocommit(self, conn, autocommit: Any) -> None:
conn.autocommit(autocommit)
+
+ def run(self, sql, autocommit=False, parameters=None):
+ """Snowflake-connector doesn't allow natively the execution of
multiple SQL statements in the same call. So for
+ allowing to pass files or strings with several queries this method is
coded, that relies on run from DBApiHook"""
+ if isinstance(sql, str):
+ queries = sql.split(';')
Review comment:
A query that searches for ';' has been a concern to me. I don't know how
often is that, though, but I think I could add some parameter or something to
make a different behavior in that case (based in re, for instance)
I'll take a look at execute_string
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
