[DNA-0000]: Add Trino original user header for impersonated users (#29)

* [DNA-00000]: Fix dependencies version * [DNA-00000]: Add Trino original user header
careem · Aug 6, 2024 · ba920a0 · ba920a0
1 parent f2ca720
commit ba920a0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/redash/query_runner/trino.py b/redash/query_runner/trino.py
@@ -112,7 +112,10 @@ def run_query(self, query, user):
             host=self.configuration.get("host", ""),
             port=self.configuration.get("port", 8080),
             catalog=self.configuration.get("catalog", "hive"),
-            http_headers={trino.constants.HEADER_CLIENT_INFO: str(user)},
+            http_headers={
+                trino.constants.HEADER_CLIENT_INFO: str(user),
+                "X-Trino-Original-User": str(user)
+            },
             source="Redash",
             schema=self.configuration.get("schema", "default"),
             user=self.configuration.get("username"),