Need to run context executed every time we run something against DB

[pavankkurella]

I am trying to set context before each DB call.
for sql lab if we add it before execute_sql_statement in sql_lab that will be applicable only for custom sql lab queries.
can we add it somewhere to run before all database calls?

[dosubot[bot]]

Hello @pavankkurella! I'm Dosu, a friendly bot here to help you solve bugs, answer your questions, and assist with becoming a contributor. I'm ready to help you while we wait for a human maintainer.

To run context before all database calls in Apache Superset, you can utilize the mutate_sql_based_on_config hook. This hook allows you to modify the SQL query or execute code before the query is executed against the database. It is designed to allow environment-specific mutations and can be applied to all database calls, not just those from SQL Lab.

Here's a snippet showing how this hook is used:

# Hook to allow environment-specific mutation (usually comments) to the SQL
sql = database.mutate_sql_based_on_config(sql)

By implementing your context-setting logic within this hook, you can ensure it runs before every database call ^[1].

To continue talking to Dosu, mention @dosu.

---

^{Help Dosu learn! Give it feedback: Great Response | Irrelevant Answer | Incorrect Sources | Too Verbose | Hallucination | Bug Report | Other}