Skip to content

Add conversational class #60

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

Merged
vitalii-bulyzhyn merged 8 commits into from
Feb 27, 2023
Merged

Add conversational class #60

Changes from 1 commit
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
6e2765b
Add conversational class
vitalii-bulyzhyn Feb 22, 2023
c06a303
Merge branch 'main' of https://github.com/diffgram/python-sdk into sc…
vitalii-bulyzhyn Feb 22, 2023
5d6a70e
Add file_type on conversation creation
vitalii-bulyzhyn Feb 22, 2023
295b311
Add neceray attribute creation on initialiaing of conversational
vitalii-bulyzhyn Feb 22, 2023
a2e5995
Finish conversational import
vitalii-bulyzhyn Feb 23, 2023
9f6721f
Fix functions input/output
vitalii-bulyzhyn Feb 24, 2023
8fe92ff
Add is_read_only to the attribute
vitalii-bulyzhyn Feb 24, 2023
739bccc
Add is_read_only to the attribute
vitalii-bulyzhyn Feb 24, 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
Prev Previous commit
Next Next commit
Finish conversational import
  • Loading branch information
@vitalii-bulyzhyn
vitalii-bulyzhyn committed Feb 23, 2023
commit a2e59950cfa885c1cd7c9b02dc0a0a3e61d36d11
49 changes: 42 additions & 7 deletions sdk/diffgram/file/conversational.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,6 @@
from .compound_file import CompoundFile
from uuid import uuid4
import operator

class Conversational:
def __init__(self, project, name):
Expand Down Expand Up @@ -30,9 +32,9 @@ def add_conversationa_attributes_if_doesnt_exist(self):
message_date_attribute = attribute

if message_author_attribute is None:
new_message_author_attribute = self.project.attribute.new(default_schema)
message_author_attribute = self.project.attribute.new(default_schema)
self.project.attribute.update(
new_message_author_attribute,
message_author_attribute,
prompt="Author",
kind="text",
name="message_author",
Expand All @@ -41,9 +43,9 @@ def add_conversationa_attributes_if_doesnt_exist(self):
)

if message_time_attribute is None:
new_message_time_attribute = self.project.attribute.new(default_schema)
message_time_attribute = self.project.attribute.new(default_schema)
self.project.attribute.update(
new_message_time_attribute,
message_time_attribute,
prompt="Time",
kind="time",
name="message_time",
Expand All @@ -52,16 +54,20 @@ def add_conversationa_attributes_if_doesnt_exist(self):
)

if message_date_attribute is None:
new_message_date_attribute = self.project.attribute.new(default_schema)
message_date_attribute = self.project.attribute.new(default_schema)
self.project.attribute.update(
new_message_date_attribute,
message_date_attribute,
prompt="Date",
kind="date",
name="message_date",
is_global = True,
global_type = 'file'
)

self.author_attribute = message_author_attribute
self.time_attribute = message_time_attribute
self.date_attribute = message_date_attribute

def add_message(self, message_file, author=None, time=None, date=None):
message_meta = {
"author": author,
Expand All @@ -73,5 +79,34 @@ def add_message(self, message_file, author=None, time=None, date=None):

self.parent.add_child_from_local(path=message_file, ordinal=len(self.messgaes_meta))

def _new_global_instance(self):
return {
"creation_ref_id": str(uuid4()),
"type": "global",
"attribute_groups": {}
}


def upload(self):
self.parent.upload()
self.parent.upload()
child_files = self.parent.fetch_child_files()
child_files.sort(key=operator.attrgetter('id'))

for index in range(0, len(child_files)):
global_instance_for_child = self._new_global_instance()

if self.messgaes_meta[index]["author"] is not None:
global_instance_for_child["attribute_groups"][self.author_attribute["id"]] = self.messgaes_meta[index]["author"]
if self.messgaes_meta[index]["time"] is not None:
global_instance_for_child["attribute_groups"][self.time_attribute["id"]] = self.messgaes_meta[index]["time"]
if self.messgaes_meta[index]["date"] is not None:
global_instance_for_child["attribute_groups"][self.date_attribute["id"]] = self.messgaes_meta[index]["date"]

payload = {
"instance_list": [global_instance_for_child],
"and_complete": False,
"child_file_save_id": child_files[index].id
}

response = self.project.session.post(url=self.project.host + f"/api/project/{self.project.project_string_id}/file/{child_files[index].id}/annotation/update", json=payload)
self.project.handle_errors(response)