Merge pull request #58 from diffgram/add-type-to-compound-creation

Add file type to creating compound file
diffgram · Feb 22, 2023 · d14a1ed · d14a1ed
2 parents 8791e4b + 48061a5
commit d14a1ed
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/sdk/diffgram/file/compound_file.py b/sdk/diffgram/file/compound_file.py
@@ -109,12 +109,13 @@ class CompoundFile:
     child_files: List[CompoundChildFile]
     instance_list: List[dict]
 
-    def __init__(self, project: Project, name: str, directory_id: int, instance_list: List[dict] = []):
+    def __init__(self, project: Project, name: str, directory_id: int, instance_list: List[dict] = [], file_type: str = 'compound'):
         self.project = project
         self.name = name
         self.directory_id = directory_id
         self.child_files = []
         self.instance_list = instance_list
+        self.type = file_type
 
     @staticmethod
     def from_dict(project: Project, dir_id: int, dict_data: dict):
@@ -166,7 +167,8 @@ def __create_compound_parent_file(self):
         data = {
             'name': self.name,
             'directory_id': self.directory_id,
-            'instance_list': self.instance_list
+            'instance_list': self.instance_list,
+            'type': self.type
         }
         response = self.project.session.post(url = self.project.host + url,
                                              json = data)