notionary 0.4.0__py3-none-any.whl → 0.4.2__py3-none-any.whl

notionary/page/page_metadata_update_client.py

@@ -2,7 +2,9 @@ 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.entity_metadata_update_client import (
+    EntityMetadataUpdateClient,
+)
 from notionary.shared.entity.schemas import NotionEntityUpdateDto
 
 
@@ -12,8 +14,14 @@ class PageMetadataUpdateClient(NotionHttpClient, EntityMetadataUpdateClient):
         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)
+    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)
+        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

@@ -44,7 +44,9 @@ class PagePropertyHttpClient(NotionHttpClient):
         property_type: type[PagePropertyT],
         current_property: PagePropertyT | None = None,
     ) -> NotionPageDto:
-        updated_property = self._create_updated_property(property_type, current_property, value)
+        updated_property = self._create_updated_property(
+            property_type, current_property, value
+        )
 
         properties = {property_name: updated_property}
         update_dto = PgePropertiesUpdateDto(properties=properties)
@@ -54,40 +56,64 @@ class PagePropertyHttpClient(NotionHttpClient):
     async def patch_title(self, property_name: str, title: str) -> NotionPageDto:
         return await self._patch_property(property_name, title, PageTitleProperty)
 
-    async def patch_rich_text_property(self, property_name: str, text: str) -> NotionPageDto:
+    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:
+    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:
+    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:
+    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:
+    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:
+    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_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:
+    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:
+    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:
+    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)
+        return await self._patch_property(
+            property_name, relation_ids, PageRelationProperty
+        )
 
     # TODO: Fix this shit here
     def _create_updated_property(
@@ -126,8 +152,12 @@ class PagePropertyHttpClient(NotionHttpClient):
             select_option = SelectOption(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)
+            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)

notionary/page/properties/factory.py

@@ -7,12 +7,16 @@ from notionary.shared.models.parent import DataSourceParent, ParentType
 
 
 class PagePropertyHandlerFactory:
-    def create_from_page_response(self, page_response: NotionPageDto) -> PagePropertyHandler:
+    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),
+            page_property_http_client=self._create_http_client(
+                page_id=page_response.id
+            ),
             parent_data_source=self._extract_parent_data_source_id(page_response),
         )
 

notionary/page/properties/service.py

@@ -3,7 +3,9 @@ from __future__ import annotations
 import asyncio
 from typing import TYPE_CHECKING, Never
 
-from notionary.blocks.rich_text.rich_text_markdown_converter import convert_rich_text_to_markdown
+from notionary.blocks.rich_text.rich_text_markdown_converter import (
+    convert_rich_text_to_markdown,
+)
 from notionary.exceptions.properties import (
     AccessPagePropertyWithoutDataSourceError,
     PagePropertyNotFoundError,
@@ -68,23 +70,33 @@ class PagePropertyHandler:
         return await convert_rich_text_to_markdown(title_property.title)
 
     def get_values_of_people_property(self, property_name: str) -> list[str]:
-        people_prop = self._get_typed_property_or_raise(property_name, PagePeopleProperty)
+        people_prop = self._get_typed_property_or_raise(
+            property_name, PagePeopleProperty
+        )
         return [person.name for person in people_prop.people if person.name]
 
     def get_value_of_created_time_property(self, name: str) -> str | None:
-        created_time_property = self._get_typed_property_or_raise(name, PageCreatedTimeProperty)
+        created_time_property = self._get_typed_property_or_raise(
+            name, PageCreatedTimeProperty
+        )
         return created_time_property.created_time
 
     async def get_values_of_relation_property(self, name: str) -> list[str]:
         from notionary import NotionPage
 
-        relation_property = self._get_typed_property_or_raise(name, PageRelationProperty)
+        relation_property = self._get_typed_property_or_raise(
+            name, PageRelationProperty
+        )
         relation_page_ids = [rel.id for rel in relation_property.relation]
-        notion_pages = [await NotionPage.from_id(page_id) for page_id in relation_page_ids]
+        notion_pages = [
+            await NotionPage.from_id(page_id) for page_id in relation_page_ids
+        ]
         return [page.title for page in notion_pages if page]
 
     def get_values_of_multiselect_property(self, name: str) -> list[str]:
-        multiselect_property = self._get_typed_property_or_raise(name, PageMultiSelectProperty)
+        multiselect_property = self._get_typed_property_or_raise(
+            name, PageMultiSelectProperty
+        )
         return [option.name for option in multiselect_property.multi_select]
 
     def get_value_of_url_property(self, name: str) -> str | None:
@@ -96,7 +108,9 @@ class PagePropertyHandler:
         return number_property.number
 
     def get_value_of_checkbox_property(self, name: str) -> bool:
-        checkbox_property = self._get_typed_property_or_raise(name, PageCheckboxProperty)
+        checkbox_property = self._get_typed_property_or_raise(
+            name, PageCheckboxProperty
+        )
         return checkbox_property.checkbox
 
     def get_value_of_date_property(self, name: str) -> str | None:
@@ -104,7 +118,9 @@ class PagePropertyHandler:
         return date_property.date.start if date_property.date else None
 
     async def get_value_of_rich_text_property(self, name: str) -> str:
-        rich_text_property = self._get_typed_property_or_raise(name, PageRichTextProperty)
+        rich_text_property = self._get_typed_property_or_raise(
+            name, PageRichTextProperty
+        )
         return await convert_rich_text_to_markdown(rich_text_property.rich_text)
 
     def get_value_of_email_property(self, name: str) -> str | None:
@@ -112,26 +128,36 @@ class PagePropertyHandler:
         return email_property.email
 
     def get_value_of_phone_number_property(self, name: str) -> str | None:
-        phone_property = self._get_typed_property_or_raise(name, PagePhoneNumberProperty)
+        phone_property = self._get_typed_property_or_raise(
+            name, PagePhoneNumberProperty
+        )
         return phone_property.phone_number
 
     # =========================================================================
     # Options Getters
     # =========================================================================
 
-    async def get_select_options_by_property_name(self, property_name: str) -> list[str]:
+    async def get_select_options_by_property_name(
+        self, property_name: str
+    ) -> list[str]:
         data_source = await self._get_parent_data_source_or_raise()
         return data_source.get_select_options_by_property_name(property_name)
 
-    async def get_multi_select_options_by_property_name(self, property_name: str) -> list[str]:
+    async def get_multi_select_options_by_property_name(
+        self, property_name: str
+    ) -> list[str]:
         data_source = await self._get_parent_data_source_or_raise()
         return data_source.get_multi_select_options_by_property_name(property_name)
 
-    async def get_status_options_by_property_name(self, property_name: str) -> list[str]:
+    async def get_status_options_by_property_name(
+        self, property_name: str
+    ) -> list[str]:
         data_source = await self._get_parent_data_source_or_raise()
         return data_source.get_status_options_by_property_name(property_name)
 
-    async def get_relation_options_by_property_name(self, property_name: str) -> list[str]:
+    async def get_relation_options_by_property_name(
+        self, property_name: str
+    ) -> list[str]:
         data_source = await self._get_parent_data_source_or_raise()
         return await data_source.get_relation_options_by_property_name(property_name)
 
@@ -151,7 +177,9 @@ class PagePropertyHandler:
         title_property_name = self._extract_title_property_name()
 
         self._get_typed_property_or_raise(title_property_name, PageTitleProperty)
-        updated_page = await self._property_http_client.patch_title(title_property_name, title)
+        updated_page = await self._property_http_client.patch_title(
+            title_property_name, title
+        )
         self._properties = updated_page.properties
 
     def _extract_title_property_name(self) -> str | None:
@@ -159,64 +187,104 @@ class PagePropertyHandler:
             return None
 
         return next(
-            (key for key, prop in self._properties.items() if isinstance(prop, PageTitleProperty)),
+            (
+                key
+                for key, prop in self._properties.items()
+                if isinstance(prop, PageTitleProperty)
+            ),
             None,
         )
 
     async def set_rich_text_property(self, property_name: str, text: str) -> None:
         self._get_typed_property_or_raise(property_name, PageRichTextProperty)
-        updated_page = await self._property_http_client.patch_rich_text_property(property_name, text)
+        updated_page = await self._property_http_client.patch_rich_text_property(
+            property_name, text
+        )
         self._properties = updated_page.properties
 
     async def set_url_property(self, property_name: str, url: str) -> None:
         self._get_typed_property_or_raise(property_name, PageURLProperty)
-        updated_page = await self._property_http_client.patch_url_property(property_name, url)
+        updated_page = await self._property_http_client.patch_url_property(
+            property_name, url
+        )
         self._properties = updated_page.properties
 
     async def set_email_property(self, property_name: str, email: str) -> None:
         self._get_typed_property_or_raise(property_name, PageEmailProperty)
-        updated_page = await self._property_http_client.patch_email_property(property_name, email)
+        updated_page = await self._property_http_client.patch_email_property(
+            property_name, email
+        )
         self._properties = updated_page.properties
 
-    async def set_phone_number_property(self, property_name: str, phone_number: str) -> None:
+    async def set_phone_number_property(
+        self, property_name: str, phone_number: str
+    ) -> None:
         self._get_typed_property_or_raise(property_name, PagePhoneNumberProperty)
-        updated_page = await self._property_http_client.patch_phone_property(property_name, phone_number)
+        updated_page = await self._property_http_client.patch_phone_property(
+            property_name, phone_number
+        )
         self._properties = updated_page.properties
 
-    async def set_number_property(self, property_name: str, number: int | float) -> None:
+    async def set_number_property(
+        self, property_name: str, number: int | float
+    ) -> None:
         self._get_typed_property_or_raise(property_name, PageNumberProperty)
-        updated_page = await self._property_http_client.patch_number_property(property_name, number)
+        updated_page = await self._property_http_client.patch_number_property(
+            property_name, number
+        )
         self._properties = updated_page.properties
 
     async def set_checkbox_property(self, property_name: str, checked: bool) -> None:
         self._get_typed_property_or_raise(property_name, PageCheckboxProperty)
-        updated_page = await self._property_http_client.patch_checkbox_property(property_name, checked)
+        updated_page = await self._property_http_client.patch_checkbox_property(
+            property_name, checked
+        )
         self._properties = updated_page.properties
 
-    async def set_date_property(self, property_name: str, date_value: str | dict) -> None:
+    async def set_date_property(
+        self, property_name: str, date_value: str | dict
+    ) -> None:
         self._get_typed_property_or_raise(property_name, PageDateProperty)
-        updated_page = await self._property_http_client.patch_date_property(property_name, date_value)
+        updated_page = await self._property_http_client.patch_date_property(
+            property_name, date_value
+        )
         self._properties = updated_page.properties
 
-    async def set_select_property_by_option_name(self, property_name: str, option_name: str) -> None:
+    async def set_select_property_by_option_name(
+        self, property_name: str, option_name: str
+    ) -> None:
         self._get_typed_property_or_raise(property_name, PageSelectProperty)
-        updated_page = await self._property_http_client.patch_select_property(property_name, option_name)
+        updated_page = await self._property_http_client.patch_select_property(
+            property_name, option_name
+        )
         self._properties = updated_page.properties
 
-    async def set_multi_select_property_by_option_names(self, property_name: str, option_names: list[str]) -> None:
+    async def set_multi_select_property_by_option_names(
+        self, property_name: str, option_names: list[str]
+    ) -> None:
         self._get_typed_property_or_raise(property_name, PageMultiSelectProperty)
-        updated_page = await self._property_http_client.patch_multi_select_property(property_name, option_names)
+        updated_page = await self._property_http_client.patch_multi_select_property(
+            property_name, option_names
+        )
         self._properties = updated_page.properties
 
-    async def set_status_property_by_option_name(self, property_name: str, status: str) -> None:
+    async def set_status_property_by_option_name(
+        self, property_name: str, status: str
+    ) -> None:
         self._get_typed_property_or_raise(property_name, PageStatusProperty)
-        updated_page = await self._property_http_client.patch_status_property(property_name, status)
+        updated_page = await self._property_http_client.patch_status_property(
+            property_name, status
+        )
         self._properties = updated_page.properties
 
-    async def set_relation_property_by_page_titles(self, property_name: str, page_titles: list[str]) -> None:
+    async def set_relation_property_by_page_titles(
+        self, property_name: str, page_titles: list[str]
+    ) -> None:
         self._get_typed_property_or_raise(property_name, PageRelationProperty)
         relation_ids = await self._convert_page_titles_to_ids(page_titles)
-        updated_page = await self._property_http_client.patch_relation_property(property_name, relation_ids)
+        updated_page = await self._property_http_client.patch_relation_property(
+            property_name, relation_ids
+        )
         self._properties = updated_page.properties
 
     async def _ensure_data_source_loaded(self) -> None:
@@ -226,7 +294,9 @@ class PagePropertyHandler:
             return
 
         self._parent_data_source = (
-            await NotionDataSource.from_id(self._parent_data_source_id) if self._parent_data_source_id else None
+            await NotionDataSource.from_id(self._parent_data_source_id)
+            if self._parent_data_source_id
+            else None
         )
         self._data_source_loaded = True
 
@@ -237,7 +307,9 @@ class PagePropertyHandler:
             raise AccessPagePropertyWithoutDataSourceError(self._parent_type)
         return self._parent_data_source
 
-    def _get_typed_property_or_raise(self, name: str, property_type: type[PagePropertyT]) -> PagePropertyT:
+    def _get_typed_property_or_raise(
+        self, name: str, property_type: type[PagePropertyT]
+    ) -> PagePropertyT:
         prop = self._properties.get(name)
 
         if prop is None:
@@ -267,6 +339,8 @@ class PagePropertyHandler:
         if not page_titles:
             return []
 
-        pages = await asyncio.gather(*[NotionPage.from_title(title=title) for title in page_titles])
+        pages = await asyncio.gather(
+            *[NotionPage.from_title(title=title) for title in page_titles]
+        )
 
         return [page.id for page in pages if page]

notionary/page/service.py

@@ -2,7 +2,9 @@ from collections.abc import Callable
 from typing import Self
 
 from notionary.blocks.client import NotionBlockHttpClient
-from notionary.blocks.rich_text.rich_text_markdown_converter import convert_rich_text_to_markdown
+from notionary.blocks.rich_text.rich_text_markdown_converter import (
+    convert_rich_text_to_markdown,
+)
 from notionary.comments.models import Comment
 from notionary.comments.service import CommentService
 from notionary.page.content.factory import PageContentServiceFactory
@@ -71,7 +73,9 @@ class NotionPage(Entity):
         page_property_handler_factory: PagePropertyHandlerFactory,
     ) -> Self:
         title_task = cls._extract_title_from_dto(dto)
-        page_property_handler = page_property_handler_factory.create_from_page_response(dto)
+        page_property_handler = page_property_handler_factory.create_from_page_response(
+            dto
+        )
 
         title = await title_task
 
@@ -92,7 +96,9 @@ class NotionPage(Entity):
         comment_service = CommentService()
 
         page_content_service_factory = PageContentServiceFactory()
-        page_content_service = page_content_service_factory.create(page_id=dto.id, block_client=block_client)
+        page_content_service = page_content_service_factory.create(
+            page_id=dto.id, block_client=block_client
+        )
 
         metadata_update_client = PageMetadataUpdateClient(page_id=dto.id)
 
@@ -109,7 +115,11 @@ class NotionPage(Entity):
     @staticmethod
     async def _extract_title_from_dto(response: NotionPageDto) -> str:
         title_property = next(
-            (prop for prop in response.properties.values() if isinstance(prop, PageTitleProperty)),
+            (
+                prop
+                for prop in response.properties.values()
+                if isinstance(prop, PageTitleProperty)
+            ),
             None,
         )
         rich_text_title = title_property.title if title_property else []
@@ -135,10 +145,14 @@ class NotionPage(Entity):
         return await self._comment_service.list_all_comments_for_page(page_id=self._id)
 
     async def post_top_level_comment(self, comment: str) -> None:
-        await self._comment_service.create_comment_on_page(page_id=self._id, text=comment)
+        await self._comment_service.create_comment_on_page(
+            page_id=self._id, text=comment
+        )
 
     async def post_reply_to_discussion(self, discussion_id: str, comment: str) -> None:
-        await self._comment_service.reply_to_discussion_by_id(discussion_id=discussion_id, text=comment)
+        await self._comment_service.reply_to_discussion_by_id(
+            discussion_id=discussion_id, text=comment
+        )
 
     async def set_title(self, title: str) -> None:
         await self.properties.set_title_property(title)

notionary/shared/entity/client.py

@@ -1,7 +1,9 @@
 from typing import TypeVar, override
 
 from notionary.http.client import NotionHttpClient
-from notionary.shared.entity.entity_metadata_update_client import EntityMetadataUpdateClient
+from notionary.shared.entity.entity_metadata_update_client import (
+    EntityMetadataUpdateClient,
+)
 from notionary.shared.entity.schemas import EntityResponseDto, NotionEntityUpdateDto
 
 ResponseType = TypeVar("ResponseType", bound=EntityResponseDto)
@@ -22,7 +24,9 @@ class GenericEntityMetadataUpdateClient(NotionHttpClient, EntityMetadataUpdateCl
 
     @override
     async def patch_metadata(self, updated_data: NotionEntityUpdateDto) -> ResponseType:
-        updated_data_dict = updated_data.model_dump(exclude_unset=True, exclude_none=True)
+        updated_data_dict = updated_data.model_dump(
+            exclude_unset=True, exclude_none=True
+        )
         url = f"{self._path_segment}/{self._entity_id}"
 
         response_dict = await self.patch(url, data=updated_data_dict)

notionary/shared/entity/dto_parsers.py

@@ -1,4 +1,6 @@
-from notionary.blocks.rich_text.rich_text_markdown_converter import RichTextToMarkdownConverter
+from notionary.blocks.rich_text.rich_text_markdown_converter import (
+    RichTextToMarkdownConverter,
+)
 from notionary.shared.entity.schemas import Describable, Titled
 
 

notionary/shared/entity/entity_metadata_update_client.py

@@ -7,7 +7,9 @@ from notionary.shared.models.icon import EmojiIcon, Icon
 
 class EntityMetadataUpdateClient(ABC):
     @abstractmethod
-    async def patch_metadata(self, updated_data: NotionEntityUpdateDto) -> EntityResponseDto: ...
+    async def patch_metadata(
+        self, updated_data: NotionEntityUpdateDto
+    ) -> EntityResponseDto: ...
 
     async def patch_emoji_icon(self, emoji: str) -> EntityResponseDto:
         icon = EmojiIcon(emoji=emoji)
@@ -18,7 +20,9 @@ class EntityMetadataUpdateClient(ABC):
         icon = ExternalFile.from_url(icon_url)
         return await self._patch_icon(icon)
 
-    async def patch_icon_from_file_upload(self, file_upload_id: str) -> EntityResponseDto:
+    async def patch_icon_from_file_upload(
+        self, file_upload_id: str
+    ) -> EntityResponseDto:
         icon = FileUploadFile.from_id(id=file_upload_id)
         return await self._patch_icon(icon)
 
@@ -34,7 +38,9 @@ class EntityMetadataUpdateClient(ABC):
         cover = ExternalFile.from_url(cover_url)
         return await self._patch_cover(cover)
 
-    async def patch_cover_from_file_upload(self, file_upload_id: str) -> EntityResponseDto:
+    async def patch_cover_from_file_upload(
+        self, file_upload_id: str
+    ) -> EntityResponseDto:
         cover = FileUploadFile.from_id(id=file_upload_id)
         return await self._patch_cover(cover)
 

notionary/shared/entity/schemas.py

@@ -32,7 +32,7 @@ class EntityResponseDto(BaseModel):
 
 
 class NotionEntityUpdateDto(BaseModel):
-    icon: File | None = None
+    icon: Icon | None = None
     cover: File | None = None
     in_trash: bool | None = None