factorize a bit in helper method

Author: FChataigner

dataikuapi/dss/future.py

@@ -10,6 +10,15 @@ def __init__(self, client, job_id, state=None):
        self.state = state
        self.state_is_peek = True
 
+    @classmethod
+    def get_result_wait_if_needed(cls, client, ret):
+        if 'jobId' in ret:
+            future = DSSFuture(client, ret["jobId"], ret)
+            future.wait_for_result()
+            return future.get_result()
+        else:
+            return ret['result']
+
     def abort(self):
         """
         Abort the future

dataikuapi/dss/project.py

@@ -891,12 +891,7 @@ def list_sql_tables(self, connection_name, schema_name=None):
 
         def to_schema_table_pair(x):
             return {"schema":x.get("schema", None), "table":x["table"]}
-        if 'jobId' in ret:
-            future = self.client.get_future(ret["jobId"])
-            future.wait_for_result()
-            return [to_schema_table_pair(x) for x in future.get_result()]
-        else:
-            return [to_schema_table_pair(x) for x in ret['result']]
+        return [to_schema_table_pair(x) for x in DSSFuture.get_result_wait_if_needed(self.client, ret)['tables']]
 
     def list_hive_tables(self, hive_database):
         """
@@ -910,12 +905,7 @@ def list_hive_tables(self, hive_database):
 
         def to_schema_table_pair(x):
             return {"schema":x.get("databaseName", None), "table":x["table"]}
-        if 'jobId' in ret:
-            future = self.client.get_future(ret["jobId"])
-            future.wait_for_result()
-            return [to_schema_table_pair(x) for x in future.get_result()['tables']]
-        else:
-            return [to_schema_table_pair(x) for x in ret['result']['tables']]
+        return [to_schema_table_pair(x) for x in DSSFuture.get_result_wait_if_needed(self.client, ret)['tables']]
 
 class TablesImportDefinition(object):
     """