documente_shared 0.1.160__tar.gz → 0.1.161b0__tar.gz

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

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

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/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.160 → documente_shared-0.1.161b0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "documente_shared"
-version = "0.1.160"
+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.160 → documente_shared-0.1.161b0}/uv.lock

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