streamlit-octostar-utils 0.4.2.dev22__tar.gz → 0.4.2.dev24__tar.gz

{streamlit_octostar_utils-0.4.2.dev22 → streamlit_octostar_utils-0.4.2.dev24}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: streamlit-octostar-utils
-Version: 0.4.2.dev22
+Version: 0.4.2.dev24
 Summary: 
 License: MIT
 License-File: LICENSE

{streamlit_octostar_utils-0.4.2.dev22 → streamlit_octostar_utils-0.4.2.dev24}/pyproject.toml

@@ -5,7 +5,7 @@ include = '\.pyi?$'
 
 [tool.poetry]
 name = "streamlit-octostar-utils"
-version = "0.4.2-dev.22"
+version = "0.4.2-dev.24"
 description = ""
 license = "MIT"
 authors = ["Octostar"]

{streamlit_octostar_utils-0.4.2.dev22 → streamlit_octostar_utils-0.4.2.dev24}/streamlit_octostar_utils/api_crafter/contents.py

@@ -628,6 +628,13 @@ class WorkspaceAttachmentContents(Contents):
             super().close()
     
     def truncate(self, size: Optional[int] = None) -> int:
+        if not self._buffer and (size == 0 or (size is None and self._position == 0)):
+            self._buffer = BytesIO()
+            self._fully_loaded = True
+            self._size = 0
+            self._position = 0
+            self._modified = True
+            return 0
         self._modified = True
         return self._ensure_buffer().truncate(size)
     
@@ -902,6 +909,13 @@ class TemporaryAttachmentContents(Contents):
             super().close()
     
     def truncate(self, size: Optional[int] = None) -> int:
+        if not self._buffer and (size == 0 or (size is None and self._position == 0)):
+            self._buffer = BytesIO()
+            self._fully_loaded = True
+            self._size = 0
+            self._position = 0
+            self._modified = True
+            return 0
         self._modified = True
         return self._ensure_buffer().truncate(size)
     

{streamlit_octostar_utils-0.4.2.dev22 → streamlit_octostar_utils-0.4.2.dev24}/streamlit_octostar_utils/api_crafter/nifi.py

@@ -25,7 +25,7 @@ from ..core.dict import recursive_update_dict, travel_dict, jsondict_hash
 from ..core.timestamp import now, string_to_datetime
 from .fastapi import DefaultErrorRoute, Route
 from ..ontology.inheritance import is_child_concept as is_child_concept_fn, get_label, get_label_keys
-from .contents import Contents, WorkspaceAttachmentContents, TemporaryAttachmentContents
+from .contents import Contents, WorkspaceAttachmentContents, TemporaryAttachmentContents, MemoryContents
 
 
 class RelationshipName(BaseModel):
@@ -49,6 +49,8 @@ OS_RESERVED_FIELDS = [
     "entity_label", "os_created_at", "os_created_by",
     "os_last_updated_at", "os_last_updated_by",
 ]
+MAX_IN_MEMORY_SIZE_BYTES = 5_242_880
+
 
 class NifiProxyEntityModel(BaseModel):
     entity_id: str
@@ -482,6 +484,7 @@ class NifiContextManager(object):
                 filetype=filetype,
                 fields=record,
             )
+            target.truncate(0)
             target.write(data)
             target.flush()
             file._contents = target
@@ -956,6 +959,12 @@ class NifiEntity(object):
         )
         if isinstance(file, Contents):
             child_entity._contents = file
+        elif len(file) <= MAX_IN_MEMORY_SIZE_BYTES:
+            child_entity._contents = MemoryContents(
+                entity_type=os_entity_type,
+                filetype=filetype,
+                initial_data=file,
+            )
         else:
             temp_filename = f"tmp_{child_entity.record['os_entity_uid']}"
             temp_contents = TemporaryAttachmentContents(
@@ -964,6 +973,7 @@ class NifiEntity(object):
                 filename=temp_filename,
                 client=self.context.client,
             )
+            temp_contents.truncate(0)
             temp_contents.write(file)
             temp_contents.flush()
             child_entity._contents = temp_contents

streamlit_octostar_utils-0.4.2.dev22/streamlit_octostar_utils/octostar/constants.py

@@ -1 +0,0 @@
-OS_XMP_NAMESPACE = "http://octostar.io/ns/1.0/"