documente_shared 0.1.24__tar.gz → 0.1.26__tar.gz

{documente_shared-0.1.24 → documente_shared-0.1.26}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: documente_shared
-Version: 0.1.24
+Version: 0.1.26
 Summary: Shared utilities for Documente AI projects
 License: MIT
 Author: Tech

{documente_shared-0.1.24 → documente_shared-0.1.26}/documente_shared/domain/entities.py

@@ -19,7 +19,7 @@ def remove_slash_from_path(path: str) -> str:
 class DocumentProcess(object):
     digest: str
     status: DocumentProcessStatus
-    file_path: str
+    file_path: Optional[str] = None
     file_bytes: Optional[bytes] = None
     category: Optional[DocumentProcessCategory] = None
     sub_category: Optional[DocumentProcessSubCategory] = None

{documente_shared-0.1.24 → documente_shared-0.1.26}/documente_shared/domain/enums.py

@@ -8,6 +8,7 @@ class DocumentProcessStatus(BaseEnum):
     COMPLETED = 'COMPLETED'
     FAILED = 'FAILED'
     DELETED = 'DELETED'
+    CANCELLED = 'CANCELLED'
 
 
 class DocumentProcessCategory(BaseEnum):

{documente_shared-0.1.24 → documente_shared-0.1.26}/documente_shared/infrastructure/s3_bucket.py

@@ -3,6 +3,8 @@ import boto3
 from dataclasses import dataclass
 from typing import Optional
 
+from documente_shared.domain.entities import remove_slash_from_path
+
 
 def remove_none_values(data: dict) -> dict:  # noqa: WPS110
     return {key: value for key, value in data.items() if value is not None}  # noqa: WPS110
@@ -22,7 +24,8 @@ class S3Bucket(object):
             return None
 
     def get_bytes(self, file_key: str) -> Optional[bytes]:
-        file_context = self.get(file_key)
+        cleaned_file_key = remove_slash_from_path(file_key)
+        file_context = self.get(cleaned_file_key)
         if not file_context:
             return None
         return (
@@ -32,19 +35,22 @@ class S3Bucket(object):
         )
 
     def upload(self, file_key: str, file_content, content_type: Optional[str] = None):
+        cleaned_file_key = remove_slash_from_path(file_key)
         optional_params = {'ContentType': content_type}
-        return self._resource.Object(self.bucket_name, file_key).put(
+        return self._resource.Object(self.bucket_name, cleaned_file_key).put(
             Body=file_content,
             **remove_none_values(optional_params),
         )
 
     def delete(self, file_key: str):
-        return self._resource.Object(self.bucket_name, file_key).delete()
+        cleaned_file_key = remove_slash_from_path(file_key)
+        return self._resource.Object(self.bucket_name, cleaned_file_key).delete()
 
     def get_url(self, file_key: str):
+        cleaned_file_key = remove_slash_from_path(file_key)
         return 'https://{bucket_url}.s3.amazonaws.com/{file_key}'.format(
             bucket_url=self.bucket_name,
-            file_key=file_key,
+            file_key=cleaned_file_key,
         )
 
     def read(self, file_key: str) -> bytes:

{documente_shared-0.1.24 → documente_shared-0.1.26}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "documente_shared"
-version = "0.1.24"
+version = "0.1.26"
 description = "Shared utilities for Documente AI projects"
 authors = ["Tech <tech@llamitai.com>"]
 license = "MIT"