chore!: update mutator to take kwargs (#19083)

* update mutator to take kwargs

* update updating.md

* lint

* test that the database name is properly passed in to the mutator
This commit is contained in:
Elizabeth Thompson
2022-03-18 16:01:27 -07:00
committed by GitHub
parent f341025d80
commit d645579cdd
7 changed files with 57 additions and 12 deletions

View File

@@ -55,7 +55,12 @@ class PrestoDBSQLValidator(BaseSQLValidator):
# Hook to allow environment-specific mutation (usually comments) to the SQL
sql_query_mutator = config["SQL_QUERY_MUTATOR"]
if sql_query_mutator:
sql = sql_query_mutator(sql, user_name, security_manager, database)
sql = sql_query_mutator(
sql,
user_name=user_name,
security_manager=security_manager,
database=database,
)
# Transform the final statement to an explain call before sending it on
# to presto to validate