Skip to content

Smart fill #83

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 2 commits into
base: master
Choose a base branch
from
Open

Smart fill #83

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
1efc008
smart_fill
SkywalkerSpace Apr 4, 2023
3f6eb6d
smart_fill_task_status
SkywalkerSpace Apr 7, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
67 changes: 66 additions & 1 deletion seatable_api/main.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
import io
import time
import json
import re
from datetime import datetime, timedelta
Expand Down Expand Up @@ -190,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 = {
Expand Down Expand Up @@ -944,7 +955,6 @@ def add_workflow_task_with_existed_row(self, workflow_token, row_id, initiator=N
response = requests.post(url, data={'row_id': row_id, 'initiator': initiator}, headers=headers)
return parse_response(response)['task']


def big_data_insert_rows(self, table_name, rows_data):
url = self._dtable_db_insert_rows_url()
json_data = {
Expand All @@ -954,6 +964,61 @@ def big_data_insert_rows(self, table_name, rows_data):
response = requests.post(url, json=json_data, headers=self.headers, timeout=self.timeout)
return parse_response(response)

# ----- ----- ----- AI ----- ----- ----- #

def smart_fill_from_text(self, table_name, long_text, columns):
"""
:param table_name: str
:param long_text: str
:param columns: list
:return: dict
"""
url = self._smart_fill_from_text_url()
json_data = {
'dtable_uuid': self.dtable_uuid,
'table_name': table_name,
'long_text': long_text,
'columns': columns,
}
response = requests.post(url, json=json_data, headers=self.headers, timeout=self.timeout)
return parse_response(response)

def smart_fill_from_column(self, table_name, long_text_column_name, columns, start=0, limit=10):
"""
:param table_name: str
:param long_text_column_name: str
:param columns: list
:param start: int
:param limit: int
:return: dict
"""
url = self._smart_fill_from_column_url()
json_data = {
'dtable_uuid': self.dtable_uuid,
'table_name': table_name,
'long_text_column_name': long_text_column_name,
'columns': columns,
'start': start,
'limit': limit,
}
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)
response = parse_response(response)
is_finished = response['is_finished']
if is_finished:
return response
return {'error': 'smart fill task status timeout'}


class Account(object):
Expand Down