smart_fill_task_status

Author: SkywalkerSpace

seatable_api/main.py

@@ -1,4 +1,5 @@
 import io
+import time
 import json
 import re
 from datetime import datetime, timedelta
@@ -172,12 +173,6 @@ def _dtable_db_linked_records_url(self):
     def _dtable_db_insert_rows_url(self):
         return self.dtable_db_url + '/api/v1/insert-rows/' + self.dtable_uuid + '/'
 
-    def _smart_fill_from_text_url(self):
-        return self.dtable_db_url + '/api/v1/smart-fill-from-text/'
-
-    def _smart_fill_from_column_url(self):
-        return self.dtable_db_url + '/api/v1/smart-fill-from-column/'
-
     def _get_related_users_url(self):
         return '%(server_url)s/api/v2.1/dtables/%(dtable_uuid)s/related-users/' % {
             'server_url': self.server_url,
@@ -196,6 +191,16 @@ def _add_workflow_task_url(self, token):
             'token': token
         }
 
+    def _smart_fill_from_text_url(self):
+        return self.server_url + '/api/v2.1/smart-fill-from-text/'
+
+    def _smart_fill_from_column_url(self):
+        return self.server_url + '/api/v2.1/smart-fill-from-column/'
+
+    def _smart_fill_task_status_url(self, task_id):
+        return self.server_url + '/api/v2.1/smart-fill-task-status/?dtable_uuid=' + \
+            self.dtable_uuid + '&task_id=' + str(task_id)
+
     def _get_account_detail(self, account_name):
         url = self._third_party_accounts_url()
         params = {
@@ -966,7 +971,7 @@ def smart_fill_from_text(self, table_name, long_text, columns):
         :param table_name: str
         :param long_text: str
         :param columns: list
-        :return: list
+        :return: dict
         """
         url = self._smart_fill_from_text_url()
         json_data = {
@@ -985,7 +990,7 @@ def smart_fill_from_column(self, table_name, long_text_column_name, columns, sta
         :param columns: list
         :param start: int
         :param limit: int
-        :return: list
+        :return: dict
         """
         url = self._smart_fill_from_column_url()
         json_data = {
@@ -999,6 +1004,21 @@ def smart_fill_from_column(self, table_name, long_text_column_name, columns, sta
         response = requests.post(url, json=json_data, headers=self.headers, timeout=self.timeout)
         return parse_response(response)
 
+    def smart_fill_task_status(self, task_id):
+        """
+        :param task_id: str
+        :return: dict
+        """
+        url = self._smart_fill_task_status_url(task_id)
+        print('start check task status, please wait...')
+        for i in range(300):
+            time.sleep(1)
+            response = requests.get(url, headers=self.headers, timeout=self.timeout)
+            task_status_response = parse_response(response)
+            is_finished = task_status_response['is_finished']
+            if is_finished:
+                return task_status_response
+        return {'error': 'smart fill task status timeout'}
 
 class Account(object):
     def __init__(self, login_name, password, server_url):