valitydev/redash
14
0
Fork 0
mirror of https://github.com/valitydev/redash.git synced 2024-11-07 01:25:16 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add get_current_user() into Python Data Source (#3088)

Browse Source
This commit is contained in:
YOSHIDA Katsuhiko 2018-11-25 21:11:27 +09:00 committed by Arik Fraimovich
parent 5648de9ba8
commit a6b782e0ce
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
redash/query_runner/python.py
View File

@ -211,10 +211,15 @@ class Python(BaseQueryRunner):
return json_loads(query.latest_query_data.data)
def get_current_user(self):
return self._current_user.to_dict()
def test_connection(self):
pass
def run_query(self, query, user):
self._current_user = user
try:
error = None
@ -239,6 +244,7 @@ class Python(BaseQueryRunner):
restricted_globals = dict(__builtins__=builtins)
restricted_globals["get_query_result"] = self.get_query_result
restricted_globals["get_source_schema"] = self.get_source_schema
restricted_globals["get_current_user"] = self.get_current_user
restricted_globals["execute_query"] = self.execute_query
restricted_globals["add_result_column"] = self.add_result_column
restricted_globals["add_result_row"] = self.add_result_row