Add conversational class

diffgram · vitalii-bulyzhyn · Feb 27, 2023 · Feb 22, 2023 · Feb 22, 2023 · Feb 22, 2023
commit 6e2765be7f792319de5cadb3d5cfa618fb3a094b
diff --git a/.gitignore b/.gitignore
@@ -31,5 +31,9 @@ sdk/diffgram/utils/__pycache__/
 
 venv/*
 
+diffgram.egg-info
+
+playground.py
+
 sdk/build/*
 sdk/dist/
diff --git a/sdk/diffgram/core/core.py b/sdk/diffgram/core/core.py
@@ -16,6 +16,8 @@
 from diffgram.brain.train import Train
 from diffgram.export.export import Export
 from diffgram.task.task import Task
+from diffgram.schema.schema import Schema
+from diffgram.schema.attribute import Attribute
 from requests.auth import HTTPBasicAuth
 
 
@@ -86,6 +88,8 @@ def __init__(
 
         self.label_schema_list = self.get_label_schema_list()
 
+        self.schema = Schema(self)
+        self.attribute = Attribute(self)
 
 
     def get_member_list(self):

diff --git a/sdk/diffgram/file/conversational.py b/sdk/diffgram/file/conversational.py
@@ -0,0 +1,25 @@
+from .compound_file import CompoundFile
+
+class Conversational:
+    def __init__(self, project, name):
+        self.parent = CompoundFile(
+            project=project, 
+            name=name, 
+            directory_id=project.default_directory.id
+        )
+
+        self.messgaes_meta = []
+
+    def add_message(self, message_file, author=None, time=None, date=None):
+        message_meta = {
+            "author": author,
+            "time": time,
+            "date": date
+        }
+
+        self.messgaes_meta.append(message_meta)
+
+        self.parent.add_child_from_local(path=message_file, ordinal=len(self.messgaes_meta))
+
+    def upload(self):
+        self.parent.upload()
diff --git a/sdk/diffgram/schema/__init__.py b/sdk/diffgram/schema/__init__.py
diff --git a/sdk/diffgram/schema/attribute.py b/sdk/diffgram/schema/attribute.py
@@ -0,0 +1,61 @@
+class Attribute:
+  def __init__(self, project):
+    self.project = project
+
+  def list(self, schema):
+    url = f'/api/v1/project/{self.project.project_string_id}/attribute/template/list'
+    payload = {
+      "schema_id": schema['id'],
+      "mode": "from_project"
+    }
+
+    response = self.project.session.post(url = self.project.host + url, json=payload)
+
+    self.project.handle_errors(response)
+    data = response.json()
+
+    return data
+
+  def new(self, schema):
+    url = f'/api/v1/project/{self.project.project_string_id}/attribute/group/new'
+    payload = {
+      "schema_id": schema['id']
+    }
+
+    response = self.project.session.post(url = self.project.host + url, json=payload)
+
+    self.project.handle_errors(response)
+    data = response.json()
+
+    return data
+
+  def update(self, 
+             attribute,
+             prompt,
+             kind,
+             is_global = False,
+             label_file_list = None,
+             global_type = 'file'
+            ):
+    url = f'/api/v1/project/{self.project.project_string_id}/attribute/group/update'
+    payload = {
+      "group_id": attribute['attribute_template_group']['id'],
+      "mode": "UPDATE",
+      "prompt": prompt,
+      "kind": kind,
+      "is_global": is_global,
+      "label_file_list": label_file_list,
+      "global_type": global_type
+    }
+
+    response = self.project.session.post(url = self.project.host + url, json=payload)
+
+    self.project.handle_errors(response)
+    data = response.json()
+
+    return data
+
+  def add_options(self, attribute, options):
+    pass
+
+
diff --git a/sdk/diffgram/schema/schema.py b/sdk/diffgram/schema/schema.py
@@ -0,0 +1,48 @@
+class Schema:
+   def __init__(self, project):
+      self.project = project
+
+   def list(self):
+      url = f'/api/v1/project/{self.project.project_string_id}/labels-schema'
+      response = self.project.session.get(url = self.project.host + url)
+      self.project.handle_errors(response)
+      data = response.json()
+
+      return data
+
+   def default_schema(self):
+      url = f'/api/v1/project/{self.project.project_string_id}/labels-schema'
+      response = self.project.session.get(url = self.project.host + url)
+      self.project.handle_errors(response)
+      data = response.json()
+
+      default_schema = None
+
+      for schema in data:
+         if schema['is_default'] == True:
+            default_schema = schema
+
+      return default_schema
+
+
+   def new(self, name):
+      url = f'/api/v1/project/{self.project.project_string_id}/labels-schema/new'
+      payload = {
+         "name": name
+      }
+
+      response = self.project.session.post(url = self.project.host + url, json=payload)
+
+      self.project.handle_errors(response)
+      data = response.json()
+
+      return data
+
+   def update(self, name):
+       #Todo
+       pass
+
+   def archive(self, name):
+       #Todo
+       pass
+