Pass job object instead of job_id to ResultProxy

treasure-data · May 28, 2015 · fad01e6 · fad01e6
1 parent 1be0300
commit fad01e6
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/pandas_td/td.py b/pandas_td/td.py
@@ -69,8 +69,6 @@ def execute(self, query, **kwargs):
 
         # issue query
         job = self.connection.client.query(self.database, query, **params)
-
-        # wait
         while not job.finished():
             time.sleep(2)
             job._update_status()
@@ -85,13 +83,13 @@ def execute(self, query, **kwargs):
                 job.status(),
                 job._debug['cmdout']))
 
-        # result as DataFrame
-        return ResultProxy(self, job.job_id)
+        # result
+        return ResultProxy(self, job)
 
 class ResultProxy(object):
-    def __init__(self, engine, job_id):
+    def __init__(self, engine, job):
         self.engine = engine
-        self.job = engine.connection.client.job(job_id)
+        self.job = job
         self._iter = None
 
     @property