notionary 0.2.22__py3-none-any.whl → 0.2.24__py3-none-any.whl

notionary/blocks/registry/block_registry_builder.py

@@ -1,264 +0,0 @@
-from __future__ import annotations
-
-from collections import OrderedDict
-from typing import TYPE_CHECKING, Self, Type
-
-from notionary.blocks.audio import AudioElement
-from notionary.blocks.base_block_element import BaseBlockElement
-from notionary.blocks.bookmark import BookmarkElement
-from notionary.blocks.breadcrumbs import BreadcrumbElement
-from notionary.blocks.bulleted_list import BulletedListElement
-from notionary.blocks.callout import CalloutElement
-from notionary.blocks.child_database import ChildDatabaseElement
-from notionary.blocks.code import CodeElement
-from notionary.blocks.column import ColumnElement, ColumnListElement
-from notionary.blocks.divider import DividerElement
-from notionary.blocks.embed import EmbedElement
-from notionary.blocks.equation import EquationElement
-from notionary.blocks.heading import HeadingElement
-from notionary.blocks.image_block import ImageElement
-from notionary.blocks.numbered_list import NumberedListElement
-from notionary.blocks.paragraph import ParagraphElement
-from notionary.blocks.quote import QuoteElement
-from notionary.blocks.table import TableElement
-from notionary.blocks.table_of_contents import TableOfContentsElement
-from notionary.blocks.todo import TodoElement
-from notionary.blocks.toggle import ToggleElement
-from notionary.blocks.toggleable_heading import ToggleableHeadingElement
-from notionary.blocks.video import VideoElement
-
-if TYPE_CHECKING:
-    from notionary.blocks.registry.block_registry import BlockRegistry
-
-
-class BlockRegistryBuilder:
-    """
-    True builder for constructing BlockRegistry instances.
-
-    This builder allows for incremental construction of registry instances
-    with specific configurations of block elements.
-    """
-
-    def __init__(self):
-        """Initialize a new builder with an empty element list."""
-        self._elements = OrderedDict()
-
-    @classmethod
-    def create_registry(cls) -> BlockRegistry:
-        """
-        Start with all standard elements in recommended order.
-        """
-        builder = cls()
-        return (
-            builder.with_headings()
-            .with_callouts()
-            .with_code()
-            .with_dividers()
-            .with_tables()
-            .with_bulleted_list()
-            .with_numbered_list()
-            .with_toggles()
-            .with_quotes()
-            .with_todos()
-            .with_bookmarks()
-            .with_images()
-            .with_videos()
-            .with_embeds()
-            .with_audio()
-            .with_paragraphs()
-            .with_toggleable_heading_element()
-            .with_columns()
-            .with_equation()
-            .with_table_of_contents()
-            .with_breadcrumbs()
-            .with_child_database()
-        ).build()
-
-    def remove_element(self, element_class: Type[BaseBlockElement]) -> Self:
-        """
-        Remove an element class from the registry configuration.
-
-        Args:
-            element_class: The element class to remove
-
-        Returns:
-            Self for method chaining
-        """
-        self._elements.pop(element_class.__name__, None)
-        return self
-
-    # WITH methods (existing)
-    def with_paragraphs(self) -> Self:
-        return self._add_element(ParagraphElement)
-
-    def with_headings(self) -> Self:
-        return self._add_element(HeadingElement)
-
-    def with_callouts(self) -> Self:
-        return self._add_element(CalloutElement)
-
-    def with_code(self) -> Self:
-        return self._add_element(CodeElement)
-
-    def with_dividers(self) -> Self:
-        return self._add_element(DividerElement)
-
-    def with_tables(self) -> Self:
-        return self._add_element(TableElement)
-
-    def with_bulleted_list(self) -> Self:
-        return self._add_element(BulletedListElement)
-
-    def with_numbered_list(self) -> Self:
-        return self._add_element(NumberedListElement)
-
-    def with_toggles(self) -> Self:
-        return self._add_element(ToggleElement)
-
-    def with_quotes(self) -> Self:
-        return self._add_element(QuoteElement)
-
-    def with_todos(self) -> Self:
-        return self._add_element(TodoElement)
-
-    def with_bookmarks(self) -> Self:
-        return self._add_element(BookmarkElement)
-
-    def with_images(self) -> Self:
-        return self._add_element(ImageElement)
-
-    def with_videos(self) -> Self:
-        return self._add_element(VideoElement)
-
-    def with_embeds(self) -> Self:
-        return self._add_element(EmbedElement)
-
-    def with_audio(self) -> Self:
-        return self._add_element(AudioElement)
-
-    def with_toggleable_heading_element(self) -> Self:
-        return self._add_element(ToggleableHeadingElement)
-
-    def with_columns(self) -> Self:
-        self._add_element(ColumnListElement)
-        self._add_element(ColumnElement)
-        return self
-
-    def with_equation(self) -> Self:
-        return self._add_element(EquationElement)
-
-    def with_table_of_contents(self) -> Self:
-        return self._add_element(TableOfContentsElement)
-
-    def with_breadcrumbs(self) -> Self:
-        return self._add_element(BreadcrumbElement)
-
-    def with_child_database(self) -> Self:
-        return self._add_element(ChildDatabaseElement)
-
-    def without_headings(self) -> Self:
-        return self.remove_element(HeadingElement)
-
-    def without_callouts(self) -> Self:
-        return self.remove_element(CalloutElement)
-
-    def without_code(self) -> Self:
-        return self.remove_element(CodeElement)
-
-    def without_dividers(self) -> Self:
-        return self.remove_element(DividerElement)
-
-    def without_tables(self) -> Self:
-        return self.remove_element(TableElement)
-
-    def without_bulleted_list(self) -> Self:
-        return self.remove_element(BulletedListElement)
-
-    def without_numbered_list(self) -> Self:
-        return self.remove_element(NumberedListElement)
-
-    def without_toggles(self) -> Self:
-        return self.remove_element(ToggleElement)
-
-    def without_quotes(self) -> Self:
-        return self.remove_element(QuoteElement)
-
-    def without_todos(self) -> Self:
-        return self.remove_element(TodoElement)
-
-    def without_bookmarks(self) -> Self:
-        return self.remove_element(BookmarkElement)
-
-    def without_images(self) -> Self:
-        return self.remove_element(ImageElement)
-
-    def without_videos(self) -> Self:
-        return self.remove_element(VideoElement)
-
-    def without_embeds(self) -> Self:
-        return self.remove_element(EmbedElement)
-
-    def without_audio(self) -> Self:
-        return self.remove_element(AudioElement)
-
-    def without_toggleable_heading_element(self) -> Self:
-        return self.remove_element(ToggleableHeadingElement)
-
-    def without_columns(self) -> Self:
-        self.remove_element(ColumnListElement)
-        self.remove_element(ColumnElement)
-        return self
-
-    def without_equation(self) -> Self:
-        return self.remove_element(EquationElement)
-
-    def without_table_of_contents(self) -> Self:
-        return self.remove_element(TableOfContentsElement)
-
-    def without_breadcrumbs(self) -> Self:
-        return self.remove_element(BreadcrumbElement)
-
-    def without_child_database(self) -> Self:
-        return self.remove_element(ChildDatabaseElement)
-
-    def build(self) -> BlockRegistry:
-        """
-        Build and return the configured BlockRegistry instance.
-        """
-        from notionary.blocks.registry.block_registry import BlockRegistry
-
-        # Ensure ParagraphElement is always present and at the end
-        self._ensure_paragraph_at_end()
-
-        registry = BlockRegistry()
-
-        # Add elements in the recorded order
-        for element_class in self._elements.values():
-            registry.register(element_class)
-
-        return registry
-
-    def _ensure_paragraph_at_end(self) -> None:
-        """
-        Internal method to ensure ParagraphElement is the last element in the registry.
-        If ParagraphElement is not present, it will be added.
-        """
-        # Remove if present, then always add at the end
-        self._elements.pop(ParagraphElement.__name__, None)
-        self._elements[ParagraphElement.__name__] = ParagraphElement
-
-    def _add_element(self, element_class: Type[BaseBlockElement]) -> Self:
-        """
-        Add an element class to the registry configuration.
-        If the element already exists, it's moved to the end.
-
-        Args:
-            element_class: The element class to add
-
-        Returns:
-            Self for method chaining
-        """
-        self._elements.pop(element_class.__name__, None)
-        self._elements[element_class.__name__] = element_class
-
-        return self

notionary/markdown/markdown_document_model.py

@@ -1,228 +0,0 @@
-from __future__ import annotations
-
-from typing import Literal, Optional, Union
-
-from pydantic import BaseModel, Field
-
-from notionary.blocks.bookmark.bookmark_markdown_node import BookmarkMarkdownBlockParams
-from notionary.blocks.bulleted_list.bulleted_list_markdown_node import (
-    BulletedListMarkdownBlockParams,
-)
-from notionary.blocks.callout.callout_markdown_node import CalloutMarkdownBlockParams
-from notionary.blocks.divider.divider_markdown_node import DividerMarkdownBlockParams
-from notionary.blocks.embed.embed_markdown_node import EmbedMarkdownBlockParams
-from notionary.blocks.equation.equation_element_markdown_node import (
-    EquationMarkdownBlockParams,
-)
-from notionary.blocks.file.file_element_markdown_node import FileMarkdownNodeParams
-
-# Import all the existing params models
-from notionary.blocks.heading.heading_markdown_node import HeadingMarkdownBlockParams
-from notionary.blocks.image_block.image_markdown_node import ImageMarkdownBlockParams
-from notionary.blocks.numbered_list.numbered_list_markdown_node import (
-    NumberedListMarkdownBlockParams,
-)
-from notionary.blocks.paragraph.paragraph_markdown_node import (
-    ParagraphMarkdownBlockParams,
-)
-from notionary.blocks.quote.quote_markdown_node import QuoteMarkdownBlockParams
-from notionary.blocks.table.table_markdown_node import TableMarkdownBlockParams
-from notionary.blocks.table_of_contents.table_of_contents_markdown_node import (
-    TableOfContentsMarkdownBlockParams,
-)
-from notionary.blocks.todo.todo_markdown_node import TodoMarkdownBlockParams
-from notionary.blocks.video.video_markdown_node import VideoMarkdownBlockParams
-
-
-class HeadingBlock(BaseModel):
-    type: Literal["heading"] = "heading"
-    params: HeadingMarkdownBlockParams
-
-
-class ParagraphBlock(BaseModel):
-    type: Literal["paragraph"] = "paragraph"
-    params: ParagraphMarkdownBlockParams
-
-
-class QuoteBlock(BaseModel):
-    type: Literal["quote"] = "quote"
-    params: QuoteMarkdownBlockParams
-
-
-class BulletedListBlock(BaseModel):
-    type: Literal["bulleted_list"] = "bulleted_list"
-    params: BulletedListMarkdownBlockParams
-
-
-class NumberedListBlock(BaseModel):
-    type: Literal["numbered_list"] = "numbered_list"
-    params: NumberedListMarkdownBlockParams
-
-
-class TodoBlock(BaseModel):
-    type: Literal["todo"] = "todo"
-    params: TodoMarkdownBlockParams
-
-
-class CalloutBlock(BaseModel):
-    type: Literal["callout"] = "callout"
-    params: CalloutMarkdownBlockParams
-
-
-class CodeBlock(BaseModel):
-    type: Literal["code"] = "code"
-    params: CodeBlock
-
-
-class ImageBlock(BaseModel):
-    type: Literal["image"] = "image"
-    params: ImageMarkdownBlockParams
-
-
-class VideoBlock(BaseModel):
-    type: Literal["video"] = "video"
-    params: VideoMarkdownBlockParams
-
-
-class AudioBlock(BaseModel):
-    type: Literal["audio"] = "audio"
-    params: FileMarkdownNodeParams
-
-
-class FileBlock(BaseModel):
-    type: Literal["file"] = "file"
-    params: FileMarkdownNodeParams
-
-
-class PdfBlock(BaseModel):
-    type: Literal["pdf"] = "pdf"
-    params: FileMarkdownNodeParams
-
-
-class BookmarkBlock(BaseModel):
-    type: Literal["bookmark"] = "bookmark"
-    params: BookmarkMarkdownBlockParams
-
-
-class EmbedBlock(BaseModel):
-    type: Literal["embed"] = "embed"
-    params: EmbedMarkdownBlockParams
-
-
-class TableBlock(BaseModel):
-    type: Literal["table"] = "table"
-    params: TableMarkdownBlockParams
-
-
-class DividerBlock(BaseModel):
-    type: Literal["divider"] = "divider"
-    params: DividerMarkdownBlockParams
-
-
-class EquationBlock(BaseModel):
-    type: Literal["equation"] = "equation"
-    params: EquationMarkdownBlockParams
-
-
-class TableOfContentsBlock(BaseModel):
-    type: Literal["table_of_contents"] = "table_of_contents"
-    params: TableOfContentsMarkdownBlockParams
-
-
-# Special blocks for nested content
-class ToggleBlockParams(BaseModel):
-    title: str
-    children: list[MarkdownBlock] = Field(default_factory=list)
-
-
-class ToggleBlock(BaseModel):
-    type: Literal["toggle"] = "toggle"
-    params: ToggleBlockParams
-
-
-class ToggleableHeadingBlockParams(BaseModel):
-    text: str
-    level: int = Field(ge=1, le=3)
-    children: list[MarkdownBlock] = Field(default_factory=list)
-
-
-class ToggleableHeadingBlock(BaseModel):
-    type: Literal["toggleable_heading"] = "toggleable_heading"
-    params: ToggleableHeadingBlockParams
-
-
-class ColumnBlockParams(BaseModel):
-    columns: list[list[MarkdownBlock]] = Field(default_factory=list)
-    width_ratios: Optional[list[float]] = None
-
-
-class ColumnBlock(BaseModel):
-    type: Literal["columns"] = "columns"
-    params: ColumnBlockParams
-
-
-# Union of all possible blocks
-MarkdownBlock = Union[
-    HeadingBlock,
-    ParagraphBlock,
-    QuoteBlock,
-    BulletedListBlock,
-    NumberedListBlock,
-    TodoBlock,
-    CalloutBlock,
-    CodeBlock,
-    ImageBlock,
-    VideoBlock,
-    AudioBlock,
-    FileBlock,
-    PdfBlock,
-    BookmarkBlock,
-    EmbedBlock,
-    TableBlock,
-    DividerBlock,
-    EquationBlock,
-    TableOfContentsBlock,
-    ToggleBlock,
-    ToggleableHeadingBlock,
-    ColumnBlock,
-]
-
-
-# Update forward references
-ToggleBlockParams.model_rebuild()
-ToggleableHeadingBlockParams.model_rebuild()
-ColumnBlockParams.model_rebuild()
-
-
-class MarkdownDocumentModel(BaseModel):
-    """
-    Complete document model for generating Markdown via MarkdownBuilder.
-    Perfect for LLM structured output!
-
-    Example:
-        {
-            "blocks": [
-                {
-                    "type": "heading",
-                    "params": {"text": "My Document", "level": 1}
-                },
-                {
-                    "type": "paragraph",
-                    "params": {"text": "Introduction text"}
-                },
-                {
-                    "type": "pdf",
-                    "params": {"url": "https://example.com/doc.pdf", "caption": "Important PDF"}
-                }
-            ]
-        }
-    """
-
-    blocks: list[MarkdownBlock] = Field(default_factory=list)
-
-    def to_markdown(self) -> str:
-        """Convert the model directly to markdown string."""
-        from notionary.markdown.markdown_builder import MarkdownBuilder
-
-        builder = MarkdownBuilder.from_model(self)
-        return builder.build()

notionary/markdown/markdown_node.py

@@ -1,30 +0,0 @@
-from __future__ import annotations
-
-from abc import ABC, abstractmethod
-
-
-class MarkdownNode(ABC):
-    """
-    Abstract base class for all Markdown block elements.
-    Enforces implementation of to_markdown().
-    """
-
-    @abstractmethod
-    def to_markdown(self) -> str:
-        """
-        Returns the Markdown representation of the block.
-        Must be implemented by subclasses.
-        """
-        pass
-
-    @classmethod
-    @abstractmethod
-    def from_params(cls, params) -> MarkdownNode:
-        """
-        Creates an instance from a params object.
-        Must be implemented by subclasses.
-        """
-        pass
-
-    def __str__(self):
-        return self.to_markdown()

notionary/page/writer/markdown_to_notion_formatting_post_processor.py

@@ -1,73 +0,0 @@
-"""
-Post-processor for handling block formatting in Markdown to Notion conversion.
-
-Handles block formatting tasks like adding empty paragraphs before media blocks
-and other formatting-related post-processing.
-"""
-
-from typing import cast
-
-from notionary.blocks.models import BlockCreateRequest
-from notionary.blocks.types import BlockType
-from notionary.blocks.paragraph.paragraph_models import (
-    CreateParagraphBlock,
-    ParagraphBlock,
-)
-
-
-class MarkdownToNotionFormattingPostProcessor:
-    """Handles block formatting post-processing for Notion blocks."""
-
-    BLOCKS_NEEDING_EMPTY_PARAGRAPH: set[BlockType] = {
-        BlockType.DIVIDER,
-        BlockType.FILE,
-        BlockType.IMAGE,
-        BlockType.PDF,
-        BlockType.VIDEO,
-    }
-
-    def process(self, blocks: list[BlockCreateRequest]) -> list[BlockCreateRequest]:
-        """Process blocks with all formatting steps."""
-        if not blocks:
-            return blocks
-
-        return self._add_empty_paragraphs_for_media_blocks(blocks)
-
-    def _add_empty_paragraphs_for_media_blocks(
-        self, blocks: list[BlockCreateRequest]
-    ) -> list[BlockCreateRequest]:
-        """Add empty paragraphs before configured block types."""
-        if not blocks:
-            return blocks
-
-        result: list[BlockCreateRequest] = []
-
-        for i, block in enumerate(blocks):
-            block_type = block.type
-
-            if (
-                block_type in self.BLOCKS_NEEDING_EMPTY_PARAGRAPH
-                and i > 0
-                and not self._is_empty_paragraph(result[-1] if result else None)
-            ):
-
-                # Create empty paragraph block inline
-                empty_paragraph = CreateParagraphBlock(
-                    paragraph=ParagraphBlock(rich_text=[])
-                )
-                result.append(empty_paragraph)
-
-            result.append(block)
-
-        return result
-
-    def _is_empty_paragraph(self, block: BlockCreateRequest | None) -> bool:
-        if not block or block.type != BlockType.PARAGRAPH:
-            return False
-        if not isinstance(block, CreateParagraphBlock):
-            return False
-
-        para_block = cast(CreateParagraphBlock, block)
-        paragraph: ParagraphBlock | None = para_block.paragraph
-        if not paragraph:
-            return False