notionary 0.2.26__py3-none-any.whl → 0.2.28__py3-none-any.whl

notionary/page/page_context.py

@@ -1,12 +1,11 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Optional
-from dataclasses import dataclass
 from contextvars import ContextVar
+from dataclasses import dataclass
+from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from notionary.database.client import NotionDatabaseClient
-    from notionary.file_upload import NotionFileUploadClient
+    from notionary.file_upload import FileUploadHttpClient
 
 
 @dataclass(frozen=True)
@@ -14,35 +13,27 @@ class PageContextProvider:
     """Context object that provides dependencies for block conversion operations."""
 
     page_id: str
-    database_client: NotionDatabaseClient
-    file_upload_client: NotionFileUploadClient
+    file_upload_client: FileUploadHttpClient
 
 
 # Context variable
-_page_context: ContextVar[Optional[PageContextProvider]] = ContextVar(
-    "page_context", default=None
-)
+_page_context: ContextVar[PageContextProvider | None] = ContextVar("page_context", default=None)
 
 
 def get_page_context() -> PageContextProvider:
     """Get current page context or raise if not available."""
     context = _page_context.get()
     if context is None:
-        raise RuntimeError(
-            "No page context available. Use 'async with page_context(...)'"
-        )
+        raise RuntimeError("No page context available. Use 'async with page_context(...)'")
     return context
 
 
 class page_context:
-    """Async-only context manager for page operations."""
-
-    def __init__(self, provider: PageContextProvider):
+    def __init__(self, provider: PageContextProvider) -> None:
         self.provider = provider
         self._token = None
 
     def _set_context(self) -> PageContextProvider:
-        """Helper to set context and return provider."""
         self._token = _page_context.set(self.provider)
         return self.provider
 

notionary/page/page_http_client.py

@@ -0,0 +1,15 @@
+from notionary.http.client import NotionHttpClient
+from notionary.page.schemas import NotionPageDto
+
+
+class NotionPageHttpClient(NotionHttpClient):
+    def __init__(
+        self,
+        page_id: str,
+    ):
+        super().__init__()
+        self._page_id = page_id
+
+    async def get_page(self) -> NotionPageDto:
+        response = await self.get(f"pages/{self._page_id}")
+        return NotionPageDto.model_validate(response)

notionary/page/page_metadata_update_client.py

@@ -0,0 +1,19 @@
+from typing import override
+
+from notionary.http.client import NotionHttpClient
+from notionary.page.schemas import NotionPageDto
+from notionary.shared.entity.entity_metadata_update_client import EntityMetadataUpdateClient
+from notionary.shared.entity.schemas import NotionEntityUpdateDto
+
+
+class PageMetadataUpdateClient(NotionHttpClient, EntityMetadataUpdateClient):
+    def __init__(self, page_id: str, timeout: int = 30) -> None:
+        super().__init__(timeout)
+        self._page_id = page_id
+
+    @override
+    async def patch_metadata(self, updated_data: NotionEntityUpdateDto) -> NotionPageDto:
+        updated_data_dict = updated_data.model_dump(exclude_unset=True, exclude_none=True)
+
+        response_dict = await self.patch(f"pages/{self._page_id}", data=updated_data_dict)
+        return NotionPageDto.model_validate(response_dict)

notionary/page/properties/client.py

@@ -0,0 +1,144 @@
+from typing import Any
+
+from pydantic import BaseModel
+
+from notionary.blocks.rich_text.models import RichText
+from notionary.http.client import NotionHttpClient
+from notionary.page.properties.models import (
+    DateValue,
+    PageCheckboxProperty,
+    PageDateProperty,
+    PageEmailProperty,
+    PageMultiSelectProperty,
+    PageNumberProperty,
+    PagePhoneNumberProperty,
+    PagePropertyT,
+    PageRelationProperty,
+    PageRichTextProperty,
+    PageSelectProperty,
+    PageStatusProperty,
+    PageTitleProperty,
+    PageURLProperty,
+    RelationItem,
+    SelectOption,
+    StatusOption,
+)
+from notionary.page.schemas import NotionPageDto, PgePropertiesUpdateDto
+
+
+class PagePropertyHttpClient(NotionHttpClient):
+    def __init__(self, page_id: str) -> None:
+        super().__init__()
+        self._page_id = page_id
+
+    async def patch_page(self, data: BaseModel) -> NotionPageDto:
+        data_dict = data.model_dump(exclude_unset=True, exclude_none=True)
+
+        result_dict = await self.patch(f"pages/{self._page_id}", data=data_dict)
+        return NotionPageDto.model_validate(result_dict)
+
+    async def _patch_property(
+        self,
+        property_name: str,
+        value: Any,
+        property_type: type[PagePropertyT],
+        current_property: PagePropertyT | None = None,
+    ) -> NotionPageDto:
+        updated_property = self._create_updated_property(property_type, current_property, value)
+
+        properties = {property_name: updated_property}
+        update_dto = PgePropertiesUpdateDto(properties=properties)
+
+        return await self.patch_page(update_dto)
+
+    async def patch_title(self, title: str) -> NotionPageDto:
+        return await self._patch_property("title", title, PageTitleProperty)
+
+    async def patch_rich_text_property(self, property_name: str, text: str) -> NotionPageDto:
+        return await self._patch_property(property_name, text, PageRichTextProperty)
+
+    async def patch_url_property(self, property_name: str, url: str) -> NotionPageDto:
+        return await self._patch_property(property_name, url, PageURLProperty)
+
+    async def patch_email_property(self, property_name: str, email: str) -> NotionPageDto:
+        return await self._patch_property(property_name, email, PageEmailProperty)
+
+    async def patch_phone_property(self, property_name: str, phone: str) -> NotionPageDto:
+        return await self._patch_property(property_name, phone, PagePhoneNumberProperty)
+
+    async def patch_number_property(self, property_name: str, number: int | float) -> NotionPageDto:
+        return await self._patch_property(property_name, number, PageNumberProperty)
+
+    async def patch_checkbox_property(self, property_name: str, checked: bool) -> NotionPageDto:
+        return await self._patch_property(property_name, checked, PageCheckboxProperty)
+
+    async def patch_select_property(self, property_name: str, value: str) -> NotionPageDto:
+        return await self._patch_property(property_name, value, PageSelectProperty)
+
+    async def patch_multi_select_property(self, property_name: str, values: list[str]) -> NotionPageDto:
+        return await self._patch_property(property_name, values, PageMultiSelectProperty)
+
+    async def patch_date_property(self, property_name: str, date_value: str | dict) -> NotionPageDto:
+        return await self._patch_property(property_name, date_value, PageDateProperty)
+
+    async def patch_status_property(self, property_name: str, status: str) -> NotionPageDto:
+        return await self._patch_property(property_name, status, PageStatusProperty)
+
+    async def patch_relation_property(self, property_name: str, relation_ids: str | list[str]) -> NotionPageDto:
+        if isinstance(relation_ids, str):
+            relation_ids = [relation_ids]
+        return await self._patch_property(property_name, relation_ids, PageRelationProperty)
+
+    # TODO: Fix this shit here
+    def _create_updated_property(
+        self,
+        property_type: type[PagePropertyT],
+        current_property: PagePropertyT | None,
+        value: Any,
+    ) -> PagePropertyT:
+        """
+        Creates an updated property instance based on the property type and new value.
+        """
+        # Get the property ID from the current property if it exists
+        property_id = current_property.id if current_property else ""
+
+        if property_type == PageTitleProperty:
+            return PageTitleProperty(
+                id=property_id,
+                title=[RichText(type="text", text={"content": str(value)})],
+            )
+        elif property_type == PageRichTextProperty:
+            return PageRichTextProperty(
+                id=property_id,
+                rich_text=[RichText(type="text", text={"content": str(value)})],
+            )
+        elif property_type == PageURLProperty:
+            return PageURLProperty(id=property_id, url=str(value))
+        elif property_type == PageEmailProperty:
+            return PageEmailProperty(id=property_id, email=str(value))
+        elif property_type == PagePhoneNumberProperty:
+            return PagePhoneNumberProperty(id=property_id, phone_number=str(value))
+        elif property_type == PageNumberProperty:
+            return PageNumberProperty(id=property_id, number=float(value))
+        elif property_type == PageCheckboxProperty:
+            return PageCheckboxProperty(id=property_id, checkbox=bool(value))
+        elif property_type == PageSelectProperty:
+            select_option = SelectOption(id="", name=str(value))
+            return PageSelectProperty(id=property_id, select=select_option)
+        elif property_type == PageMultiSelectProperty:
+            multi_select_options = [SelectOption(id="", name=str(item)) for item in value]
+            return PageMultiSelectProperty(id=property_id, multi_select=multi_select_options)
+        elif property_type == PageDateProperty:
+            if isinstance(value, dict) and "start" in value:
+                date_value = DateValue(**value)
+            else:
+                date_value = DateValue(start=str(value))
+            return PageDateProperty(id=property_id, date=date_value)
+        elif property_type == PageStatusProperty:
+            status_option = StatusOption(id="", name=str(value))
+            return PageStatusProperty(id=property_id, status=status_option)
+        elif property_type == PageRelationProperty:
+            relation_items = [RelationItem(id=str(item)) for item in value]
+            return PageRelationProperty(id=property_id, relation=relation_items)
+        else:
+            raise ValueError(f"Unsupported property type: {property_type}")

notionary/page/properties/factory.py

@@ -0,0 +1,26 @@
+from typing import cast
+
+from notionary.page.properties.client import PagePropertyHttpClient
+from notionary.page.properties.service import PagePropertyHandler
+from notionary.page.schemas import NotionPageDto
+from notionary.shared.models.parent import DataSourceParent, ParentType
+
+
+class PagePropertyHandlerFactory:
+    def create_from_page_response(self, page_response: NotionPageDto) -> PagePropertyHandler:
+        return PagePropertyHandler(
+            properties=page_response.properties,
+            parent_type=page_response.parent.type,
+            page_url=page_response.url,
+            page_property_http_client=self._create_http_client(page_id=page_response.id),
+            parent_data_source=self._extract_parent_data_source_id(page_response),
+        )
+
+    def _create_http_client(self, page_id: str) -> PagePropertyHttpClient:
+        return PagePropertyHttpClient(page_id=page_id)
+
+    def _extract_parent_data_source_id(self, response: NotionPageDto) -> str | None:
+        if response.parent.type != ParentType.DATA_SOURCE_ID:
+            return None
+        data_source_parent = cast(DataSourceParent, response.parent)
+        return data_source_parent.data_source_id if data_source_parent else None

notionary/page/properties/models.py

@@ -0,0 +1,307 @@
+from enum import StrEnum
+from typing import Annotated, Any, Literal, TypeVar
+
+from pydantic import BaseModel, Field
+
+from notionary.blocks.rich_text.models import RichText
+from notionary.shared.properties.type import PropertyType
+from notionary.user.schemas import PersonUserResponseDto, UserResponseDto
+
+# ============================================================================
+# Base Models
+# ============================================================================
+
+
+class PageProperty(BaseModel):
+    id: str
+    type: str
+
+
+class StatusOption(BaseModel):
+    id: str
+    name: str
+
+
+class SelectOption(BaseModel):
+    id: str
+    name: str
+
+
+class RelationItem(BaseModel):
+    id: str
+
+
+class DateValue(BaseModel):
+    start: str
+    end: str | None = None
+    time_zone: str | None = None
+
+
+# ============================================================================
+# File Models
+# ============================================================================
+
+
+class FileType(StrEnum):
+    EXTERNAL = "external"
+    FILE = "file"
+
+
+class ExternalFile(BaseModel):
+    """External file hosted outside of Notion."""
+
+    url: str
+
+
+class NotionFile(BaseModel):
+    """File uploaded to Notion with expiration."""
+
+    url: str
+    expiry_time: str
+
+
+class FileObject(BaseModel):
+    """File object can be external or uploaded to Notion."""
+
+    name: str
+    type: FileType
+    external: ExternalFile | None = None
+    file: NotionFile | None = None
+
+
+# ============================================================================
+# Formula Models
+# ============================================================================
+
+
+class FormulaValueType(StrEnum):
+    BOOLEAN = "boolean"
+    DATE = "date"
+    NUMBER = "number"
+    STRING = "string"
+
+
+class FormulaValue(BaseModel):
+    type: FormulaValueType
+    boolean: bool | None = None
+    date: DateValue | None = None
+    number: float | None = None
+    string: str | None = None
+
+
+# ============================================================================
+# Rollup Models
+# ============================================================================
+
+
+class RollupValueType(StrEnum):
+    NUMBER = "number"
+    DATE = "date"
+    ARRAY = "array"
+    INCOMPLETE = "incomplete"
+    UNSUPPORTED = "unsupported"
+
+
+class RollupValue(BaseModel):
+    type: RollupValueType
+    function: str  # e.g., "sum", "count", "average", "max", "min", etc.
+    number: float | None = None
+    date: DateValue | None = None
+    array: list[Any] | None = None
+
+
+# ============================================================================
+# Unique ID Models
+# ============================================================================
+
+
+class UniqueIdValue(BaseModel):
+    number: int  # Auto-incrementing count
+    prefix: str | None = None  # Optional prefix like "RL"
+
+
+# ============================================================================
+# Verification Models
+# ============================================================================
+
+
+class VerificationState(StrEnum):
+    VERIFIED = "verified"
+    UNVERIFIED = "unverified"
+
+
+class VerificationValue(BaseModel):
+    state: VerificationState
+    verified_by: UserResponseDto | None = None
+    date: DateValue | None = None  # start = verification date, end = expiration
+
+
+# ============================================================================
+# Page Property Classes
+# ============================================================================
+
+
+class PageStatusProperty(PageProperty):
+    type: Literal[PropertyType.STATUS] = PropertyType.STATUS
+    status: StatusOption | None = None
+    options: list[StatusOption] = Field(default_factory=list)
+
+    @property
+    def option_names(self) -> list[str]:
+        return [option.name for option in self.options]
+
+
+class PageRelationProperty(PageProperty):
+    type: Literal[PropertyType.RELATION] = PropertyType.RELATION
+    relation: list[RelationItem] = Field(default_factory=list)
+    has_more: bool = False
+
+
+class PageURLProperty(PageProperty):
+    type: Literal[PropertyType.URL] = PropertyType.URL
+    url: str | None = None
+
+
+class PageRichTextProperty(PageProperty):
+    type: Literal[PropertyType.RICH_TEXT] = PropertyType.RICH_TEXT
+    rich_text: list[RichText] = Field(default_factory=list)
+
+
+class PageMultiSelectProperty(PageProperty):
+    type: Literal[PropertyType.MULTI_SELECT] = PropertyType.MULTI_SELECT
+    multi_select: list[SelectOption] = Field(default_factory=list)
+    options: list[SelectOption] = Field(default_factory=list)
+
+    @property
+    def option_names(self) -> list[str]:
+        return [option.name for option in self.options]
+
+
+class PageSelectProperty(PageProperty):
+    type: Literal[PropertyType.SELECT] = PropertyType.SELECT
+    select: SelectOption | None = None
+    options: list[SelectOption] = Field(default_factory=list)
+
+    @property
+    def option_names(self) -> list[str]:
+        return [option.name for option in self.options]
+
+
+class PagePeopleProperty(PageProperty):
+    type: Literal[PropertyType.PEOPLE] = PropertyType.PEOPLE
+    people: list[PersonUserResponseDto] = Field(default_factory=list)
+
+
+class PageDateProperty(PageProperty):
+    type: Literal[PropertyType.DATE] = PropertyType.DATE
+    date: DateValue | None = None
+
+
+class PageTitleProperty(PageProperty):
+    type: Literal[PropertyType.TITLE] = PropertyType.TITLE
+    title: list[RichText] = Field(default_factory=list)
+
+
+class PageNumberProperty(PageProperty):
+    type: Literal[PropertyType.NUMBER] = PropertyType.NUMBER
+    number: float | None = None
+
+
+class PageCheckboxProperty(PageProperty):
+    type: Literal[PropertyType.CHECKBOX] = PropertyType.CHECKBOX
+    checkbox: bool = False
+
+
+class PageEmailProperty(PageProperty):
+    type: Literal[PropertyType.EMAIL] = PropertyType.EMAIL
+    email: str | None = None
+
+
+class PagePhoneNumberProperty(PageProperty):
+    type: Literal[PropertyType.PHONE_NUMBER] = PropertyType.PHONE_NUMBER
+    phone_number: str | None = None
+
+
+class PageCreatedTimeProperty(PageProperty):
+    type: Literal[PropertyType.CREATED_TIME] = PropertyType.CREATED_TIME
+    created_time: str  # ISO 8601 datetime - read-only
+
+
+class PageLastEditedTimeProperty(PageProperty):
+    type: Literal[PropertyType.LAST_EDITED_TIME] = PropertyType.LAST_EDITED_TIME
+    last_edited_time: str  # ISO 8601 datetime - read-only
+
+
+class PageCreatedByProperty(PageProperty):
+    type: Literal[PropertyType.CREATED_BY] = PropertyType.CREATED_BY
+    created_by: UserResponseDto  # User object - read-only
+
+
+class PageLastEditedByProperty(PageProperty):
+    type: Literal[PropertyType.LAST_EDITED_BY] = PropertyType.LAST_EDITED_BY
+    last_edited_by: UserResponseDto  # User object - read-only
+
+
+class PageFilesProperty(PageProperty):
+    type: Literal[PropertyType.FILES] = PropertyType.FILES
+    files: list[FileObject] = Field(default_factory=list)
+
+
+class PageFormulaProperty(PageProperty):
+    type: Literal[PropertyType.FORMULA] = PropertyType.FORMULA
+    formula: FormulaValue
+
+
+class PageRollupProperty(PageProperty):
+    type: Literal[PropertyType.ROLLUP] = PropertyType.ROLLUP
+    rollup: RollupValue
+
+
+class PageUniqueIdProperty(PageProperty):
+    type: Literal[PropertyType.UNIQUE_ID] = PropertyType.UNIQUE_ID
+    unique_id: UniqueIdValue
+
+
+class PageVerificationProperty(PageProperty):
+    type: Literal[PropertyType.VERIFICATION] = PropertyType.VERIFICATION
+    verification: VerificationValue
+
+
+class PageButtonProperty(PageProperty):
+    type: Literal[PropertyType.BUTTON] = PropertyType.BUTTON
+    button: dict[str, Any] = Field(default_factory=dict)
+
+
+# ============================================================================
+# Discriminated Union
+# ============================================================================
+
+
+DiscriminatedPageProperty = Annotated[
+    PageStatusProperty
+    | PageRelationProperty
+    | PageURLProperty
+    | PageRichTextProperty
+    | PageMultiSelectProperty
+    | PageSelectProperty
+    | PagePeopleProperty
+    | PageDateProperty
+    | PageTitleProperty
+    | PageNumberProperty
+    | PageCheckboxProperty
+    | PageEmailProperty
+    | PagePhoneNumberProperty
+    | PageCreatedTimeProperty
+    | PageLastEditedTimeProperty
+    | PageCreatedByProperty
+    | PageLastEditedByProperty
+    | PageFilesProperty
+    | PageFormulaProperty
+    | PageRollupProperty
+    | PageUniqueIdProperty
+    | PageVerificationProperty
+    | PageButtonProperty,
+    Field(discriminator="type"),
+]
+
+PagePropertyT = TypeVar("PagePropertyT", bound=PageProperty)