瀏覽代碼

[notebook] Fix spark impersonation

Erick Tryzelaar 10 年之前
父節點
當前提交
4433c1b155
共有 1 個文件被更改,包括 4 次插入2 次删除
  1. 4 2
      apps/spark/src/spark/job_server_api.py

+ 4 - 2
apps/spark/src/spark/job_server_api.py

@@ -83,8 +83,9 @@ class JobServerApi(object):
   def get_status(self):
     return self._root.get('sessions')
 
-  def create_session(self, **kwargs):
-    return self._root.post('sessions', data=json.dumps(kwargs), contenttype='application/json')
+  def create_session(self, **properties):
+    properties['proxyUser'] = self.user
+    return self._root.post('sessions', data=json.dumps(properties), contenttype='application/json')
 
   def get_session(self, uuid):
     return self._root.get('sessions/%s' % uuid)
@@ -107,6 +108,7 @@ class JobServerApi(object):
     return self._root.get('batches')
 
   def submit_batch(self, properties):
+    properties['proxyUser'] = self.user
     return self._root.post('batches', data=json.dumps(properties), contenttype=_JSON_CONTENT_TYPE)
 
   def get_batch(self, uuid):