documente_shared 0.1.12__tar.gz → 0.1.13__tar.gz

{documente_shared-0.1.12 → documente_shared-0.1.13}/PKG-INFO

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

{documente_shared-0.1.12 → documente_shared-0.1.13}/documente_shared/domain/entities.py

@@ -11,8 +11,8 @@ class DocumentProcess(object):
     digest: str
     status: DocumentProcessStatus
     file_path: str
-    processed_csv_path: str
-    processed_xlsx_path: str
+    processed_csv_path: Optional[str] = None
+    processed_xlsx_path: Optional[str] = None
     processing_time: Optional[Decimal] = None
     enqueued_at: Optional[datetime] = None
     started_at: Optional[datetime] = None

{documente_shared-0.1.12 → documente_shared-0.1.13}/documente_shared/infrastructure/s3_bucket.py

@@ -21,7 +21,7 @@ class S3Bucket(object):
         except self._resource.meta.client.exceptions.NoSuchKey:
             return None
 
-    def get_bytes(self, file_key: str) -> Optional[dict]:
+    def get_bytes(self, file_key: str) -> Optional[bytes]:
         file_context = self.get(file_key)
         if not file_context:
             return None
@@ -31,21 +31,21 @@ class S3Bucket(object):
             else None
         )
 
-    def upload(self, file_key, file_content, content_type: Optional[str] = None):
+    def upload(self, file_key: str, file_content, content_type: Optional[str] = None):
         optional_params = {'ContentType': content_type}
         return self._resource.Object(self.bucket_name, file_key).put(
             Body=file_content,
             **remove_none_values(optional_params),
         )
 
-    def delete(self, file_key):
+    def delete(self, file_key: str):
         return self._resource.Object(self.bucket_name, file_key).delete()
 
-    def get_url(self, file_key):
+    def get_url(self, file_key: str):
         return 'https://{bucket_url}.s3.amazonaws.com/{file_key}'.format(
             bucket_url=self.bucket_name,
             file_key=file_key,
         )
 
-    def read(self, file_key):
-        return self.get(file_name)['Body'].read()
+    def read(self, file_key: str) -> bytes:
+        return self.get(file_key)['Body'].read()

{documente_shared-0.1.12 → documente_shared-0.1.13}/pyproject.toml

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