documente_shared 0.1.159__tar.gz → 0.1.161b0__tar.gz

{documente_shared-0.1.159 → documente_shared-0.1.161b0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: documente_shared
-Version: 0.1.159
+Version: 0.1.161b0
 Summary: Shared utilities for Documente AI projects
 Author-email: Tech <tech@llamitai.com>
 License: MIT

{documente_shared-0.1.159 → documente_shared-0.1.161b0}/documente_shared/domain/entities/processing_case.py

@@ -7,6 +7,7 @@ from documente_shared.domain.constants import la_paz_tz
 from documente_shared.domain.entities.processing_case_item import ProcessingCaseItem
 from documente_shared.domain.enums.common import ProcessingStatus
 from documente_shared.domain.enums.processing_case import ProcessingCaseType
+from documente_shared.domain.helpers.values import optional_datetime_str
 
 
 @dataclass
@@ -21,6 +22,8 @@ class ProcessingCase(object):
     failed_at: Optional[datetime] = None
     feedback: Optional[list | dict] = None
     completed_at: Optional[datetime] = None
+    created_at: Optional[datetime] = None
+    updated_at: Optional[datetime] = None
     metadata: Optional[dict] = None
     items: Optional[List[ProcessingCaseItem]] = None
 
@@ -30,10 +33,10 @@ class ProcessingCase(object):
     def __eq__(self, other: 'ProcessingCase') -> bool:
         if not other:
             return False
-
         return (
             self.uuid == other.uuid
             and self.name == other.name
+            and self.tenant_slug == other.tenant_slug
             and self.status == other.status
             and self.case_type == other.case_type
             and self.enqueued_at == other.enqueued_at
@@ -41,16 +44,19 @@ class ProcessingCase(object):
             and self.failed_at == other.failed_at
             and self.feedback == other.feedback
             and self.completed_at == other.completed_at
+            and self.created_at == other.created_at
+            and self.updated_at == other.updated_at
             and self.metadata == other.metadata
+            and self.items == other.items
         )
 
     @property
-    def strategy_id(self) ->str:
+    def strategy_id(self) -> str:
         return str(self.case_type)
 
     @property
     def is_procesable(self) -> bool:
-        return self.items and len(self.items) > 0
+        return bool(self.items)
 
     @property
     def is_queue_procesable(self) -> bool:
@@ -86,18 +92,19 @@ class ProcessingCase(object):
                 str(self.case_type)
                 if self.case_type else None
             ),
-            'enqueued_at': self.enqueued_at.isoformat() if self.enqueued_at else None,
-            'started_at': self.started_at.isoformat() if self.started_at else None,
-            'failed_at': self.failed_at.isoformat() if self.failed_at else None,
+            'enqueued_at': optional_datetime_str(self.enqueued_at),
+            'started_at': optional_datetime_str(self.started_at),
+            'failed_at': optional_datetime_str(self.failed_at),
             'feedback': self.feedback,
-            'completed_at': self.completed_at.isoformat() if self.completed_at else None,
             'metadata': self.metadata,
+            'completed_at': optional_datetime_str(self.completed_at),
+            'updated_at': optional_datetime_str(self.updated_at),
             'items': [item.to_dict for item in self.items],
         }
 
     @property
     def to_queue_dict(self) -> dict:
-        data = self.to_dict
+        data = self.to_dict.copy()
         data["items"] = [
             item.to_queue_dict for item in self.items
         ]
@@ -105,7 +112,7 @@ class ProcessingCase(object):
 
     @property
     def to_persist_dict(self) -> dict:
-        persist_data = self.to_dict
+        persist_data = self.to_dict.copy()
         persist_data["items"] = [
             item.to_dict for item in self.items
         ]
@@ -183,8 +190,6 @@ class ProcessingCase(object):
         else:
             self.status = ProcessingStatus.PENDING
 
-
-
     @classmethod
     def from_dict(cls, data: dict) -> 'ProcessingCase':
         return cls(
@@ -202,6 +207,8 @@ class ProcessingCase(object):
             feedback=data.get('feedback'),
             metadata=data.get('metadata', {}),
             completed_at=get_datetime_from_data(input_datetime=data.get('completed_at')),
+            created_at=get_datetime_from_data(input_datetime=data.get('created_at')),
+            updated_at=get_datetime_from_data(input_datetime=data.get('updated_at')),
             items=[
                 ProcessingCaseItem.from_dict(item_dict)
                 for item_dict in data.get('items', [])
@@ -216,4 +223,3 @@ class ProcessingCase(object):
             for item_dict in data.get('items', [])
         ]
         return instance
-

documente_shared-0.1.161b0/documente_shared/domain/entities/workspace_document.py

@@ -0,0 +1,51 @@
+from uuid import UUID
+from datetime import datetime
+from decimal import Decimal
+from pydantic import BaseModel, Field
+
+from documente_shared.domain.entities.in_memory_document import InMemoryDocument
+from documente_shared.domain.enums.common import ProcessingStatus
+from documente_shared.domain.enums.workspace import WorkspaceSource
+from documente_shared.domain.helpers.values import optional_str
+
+
+class WorkspaceDocument(BaseModel):
+    uuid: UUID
+    name: str
+    digest: str | None = Field(default=None)
+    status: ProcessingStatus
+    tenant_id: UUID
+    workspace_id: UUID
+    document_type_id: UUID | None = Field(default=None)
+    source: WorkspaceSource | None = Field(default=None)
+
+    document: InMemoryDocument | None = Field(default=None)
+    document_url: str | None = Field(default=None)
+    processing_time: Decimal | None = Field(default=None)
+    metadata: dict | None = Field(default=None)
+    extraction: dict | None = Field(default=None)
+    uploaded_at: datetime | None = Field(default=None)
+    created_at: datetime | None = Field(default=None)
+    updated_at: datetime | None = Field(default=None)
+    started_at: datetime | None = Field(default=None)
+    completed_at: datetime | None = Field(default=None)
+    failed_at: datetime | None = Field(default=None)
+
+    @property
+    def to_persist_dict(self) -> dict:
+        return {
+            "tenant_id": self.tenant_id,
+            "workspace_id": self.workspace_id,
+            "doctype_id": self.document_type_id,
+            "name": self.name,
+            "digest": self.digest,
+            "status": str(self.status),
+            "source": optional_str(self.source),
+            "uploaded_at": self.uploaded_at,
+            "started_at": self.started_at,
+            "completed_at": self.completed_at,
+            "failed_at": self.failed_at,
+            "processing_time": self.processing_time,
+            "metadata": self.metadata or {},
+            "extraction": self.extraction or {},
+        }

documente_shared-0.1.161b0/documente_shared/domain/entities/workspace_document_page.py

@@ -0,0 +1,27 @@
+from datetime import datetime
+from uuid import UUID
+
+from documente_shared.domain.entities.in_memory_document import InMemoryDocument
+from pydantic import BaseModel, Field
+
+
+class WorkspaceDocumentPage(BaseModel):
+    uuid: UUID
+    tenant_id: UUID
+    workspace_id: UUID
+    workspace_document_id: UUID
+    page_number: int
+    page_file: InMemoryDocument | None = Field(default=None)
+    page_file_url: str | None = Field(default=None)
+    created_at: datetime | None = Field(default=None)
+    synced_at: datetime | None = Field(default=None)
+
+    @property
+    def to_persist_dict(self) -> dict:
+        return {
+            "tenant_id": self.tenant_id,
+            "workspace_id": self.workspace_id,
+            "workspace_document_id": self.workspace_document_id,
+            "page_number": self.page_number,
+            "synced_at": self.synced_at,
+        }

documente_shared-0.1.161b0/documente_shared/domain/enums/workspace.py

@@ -0,0 +1,19 @@
+from documente_shared.domain.base_enum import BaseEnum
+
+
+class WorkspaceSource(BaseEnum):
+    WEB = "WEB"
+    MOBILE = "MOBILE"
+    TOOLS = "TOOLS"
+
+    @property
+    def is_web(self):
+        return self == self.WEB
+
+    @property
+    def is_mobile(self):
+        return self == self.MOBILE
+
+    @property
+    def is_tools(self):
+        return self == self.TOOLS

documente_shared-0.1.161b0/documente_shared/domain/helpers/values.py

@@ -0,0 +1,13 @@
+from datetime import datetime
+
+
+def optional_str(input_value: object | None = None) -> str | None:
+    if input_value is None:
+        return None
+    return str(input_value)
+
+
+def optional_datetime_str(input_datetime: datetime | None = None) -> str | None:
+    if input_datetime is None:
+        return None
+    return input_datetime.isoformat()

documente_shared-0.1.161b0/documente_shared/domain/repositories/workspace_document.py

@@ -0,0 +1,42 @@
+from abc import ABC, abstractmethod
+from typing import Any, Optional
+from uuid import UUID
+
+from documente_shared.domain.entities.workspace_document import WorkspaceDocument
+from documente_shared.domain.enums.common import ProcessingStatus
+
+
+class WorkspaceDocumentRepository(ABC):
+    @abstractmethod
+    def find(self, uuid: UUID) -> Optional[WorkspaceDocument]:
+        raise NotImplementedError
+
+    @abstractmethod
+    def find_by_digest(self, workspace_id: UUID, digest: str) -> Optional[WorkspaceDocument]:
+        raise NotImplementedError
+
+    @abstractmethod
+    def filter_by_workspace(
+        self,
+        workspace_id: UUID,
+        status: Optional[ProcessingStatus] = None,
+        limit: Optional[int] = None,
+        offset: int = 0,
+    ) -> list[WorkspaceDocument]:
+        raise NotImplementedError
+
+    @abstractmethod
+    def filter(self, workspace_id: UUID) -> list[WorkspaceDocument]:
+        raise NotImplementedError
+
+    @abstractmethod
+    def filter_paginated(self, workspace_id: UUID, page_params: Any) -> Any:
+        raise NotImplementedError
+
+    @abstractmethod
+    def persist(self, instance: WorkspaceDocument) -> WorkspaceDocument:
+        raise NotImplementedError
+
+    @abstractmethod
+    def delete(self, uuid: UUID) -> None:
+        raise NotImplementedError

documente_shared-0.1.161b0/documente_shared/domain/repositories/workspace_document_page.py

@@ -0,0 +1,41 @@
+from abc import ABC, abstractmethod
+from typing import Any, Optional
+from uuid import UUID
+
+from documente_shared.domain.entities.workspace_document_page import WorkspaceDocumentPage
+
+
+class WorkspaceDocumentPageRepository(ABC):
+    @abstractmethod
+    def find(self, uuid: UUID) -> Optional[WorkspaceDocumentPage]:
+        raise NotImplementedError
+
+    @abstractmethod
+    def find_by_document(self, workspace_document_id: UUID) -> list[WorkspaceDocumentPage]:
+        raise NotImplementedError
+
+    @abstractmethod
+    def find_by_document_and_page_number(
+        self, workspace_document_id: UUID, page_number: int
+    ) -> Optional[WorkspaceDocumentPage]:
+        raise NotImplementedError
+
+    @abstractmethod
+    def filter(self, workspace_document_id: UUID) -> list[WorkspaceDocumentPage]:
+        raise NotImplementedError
+
+    @abstractmethod
+    def filter_paginated(self, workspace_document_id: UUID, page_params: Any) -> Any:
+        raise NotImplementedError
+
+    @abstractmethod
+    def persist(self, instance: WorkspaceDocumentPage) -> WorkspaceDocumentPage:
+        raise NotImplementedError
+
+    @abstractmethod
+    def delete(self, uuid: UUID) -> None:
+        raise NotImplementedError
+
+    @abstractmethod
+    def delete_by_document(self, workspace_document_id: UUID) -> None:
+        raise NotImplementedError

{documente_shared-0.1.159 → documente_shared-0.1.161b0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "documente_shared"
-version = "0.1.159"
+version = "0.1.161b"
 description = "Shared utilities for Documente AI projects"
 authors = [{ name = "Tech", email = "tech@llamitai.com" }]
 license = { text = "MIT" }

{documente_shared-0.1.159 → documente_shared-0.1.161b0}/uv.lock

@@ -129,7 +129,7 @@ wheels = [
 
 [[package]]
 name = "documente-shared"
-version = "0.1.159"
+version = "0.1.161b0"
 source = { editable = "." }
 dependencies = [
     { name = "boto3" },