notionary 0.2.12__py3-none-any.whl → 0.2.13__py3-none-any.whl

notionary/__init__.py

@@ -14,10 +14,6 @@ from .elements.registry.block_registry_builder import (
     BlockRegistryBuilder,
 )
 
-from .telemetry.telemetry import NotionaryTelemetry
-
-_telemetry = NotionaryTelemetry()
-
 __all__ = [
     "NotionClient",
     "NotionDatabase",

notionary/cli/main.py

@@ -6,11 +6,35 @@ Notionary CLI - Integration Key Setup
 import click
 import os
 import platform
+import asyncio
+import logging
 from pathlib import Path
 from dotenv import load_dotenv
 from rich.console import Console
 from rich.panel import Panel
 from rich.prompt import Prompt, Confirm
+from rich.table import Table
+from rich import box
+from rich.progress import Progress, SpinnerColumn, TextColumn, TimeElapsedColumn
+from notionary.notion_client import NotionClient
+from notionary.database.database_discovery import DatabaseDiscovery
+
+# Disable logging for CLI usage
+def disable_notionary_logging():
+    """Disable logging for notionary modules when used in CLI"""
+    # Option 1: Set to WARNING level (recommended for CLI)
+    logging.getLogger('notionary').setLevel(logging.WARNING)
+    logging.getLogger('DatabaseDiscovery').setLevel(logging.WARNING)
+    logging.getLogger('NotionClient').setLevel(logging.WARNING)
+
+def enable_verbose_logging():
+    """Enable verbose logging for debugging (use with --verbose flag)"""
+    logging.getLogger('notionary').setLevel(logging.DEBUG)
+    logging.getLogger('DatabaseDiscovery').setLevel(logging.DEBUG)
+    logging.getLogger('NotionClient').setLevel(logging.DEBUG)
+
+# Initialize logging configuration for CLI
+disable_notionary_logging()
 
 console = Console()
 
@@ -46,12 +70,126 @@ def get_notion_secret() -> str:
     load_dotenv()
     return os.getenv("NOTION_SECRET", "")
 
+async def fetch_notion_databases_with_progress():
+    """Fetch databases using DatabaseDiscovery with progress animation"""
+    try:
+        # Initialize NotionClient and DatabaseDiscovery
+        client = NotionClient()
+        discovery = DatabaseDiscovery(client)
+        
+        # Create progress display with custom spinner
+        with Progress(
+            SpinnerColumn(spinner_name="dots12", style="cyan"),
+            TextColumn("[bold blue]Discovering databases..."),
+            TimeElapsedColumn(),
+            console=console,
+            transient=True
+        ) as progress:
+            # Add progress task
+            task = progress.add_task("Fetching...", total=None)
+            
+            # Fetch databases
+            databases = await discovery._discover(page_size=50)
+            
+            # Update progress to show completion
+            progress.update(task, description=f"[bold green]Found {len(databases)} databases!")
+            
+            # Brief pause to show completion
+            await asyncio.sleep(0.5)
+        
+        return {"databases": databases, "success": True}
+        
+    except Exception as e:
+        return {"error": str(e), "success": False}
+
+def show_databases_overview(api_key: str):
+    """Show available databases with nice formatting"""
+    console.print("\n[bold blue]🔍 Connecting to Notion...[/bold blue]")
+    
+    # Run async function in sync context
+    try:
+        result = asyncio.run(fetch_notion_databases_with_progress())
+    except Exception as e:
+        console.print(Panel.fit(
+            f"[bold red]❌ Unexpected error[/bold red]\n\n"
+            f"[red]{str(e)}[/red]\n\n"
+            "[yellow]Please check:[/yellow]\n"
+            "• Your internet connection\n"
+            "• Your integration key validity\n"
+            "• Try running the command again",
+            title="Connection Error"
+        ))
+        return
+    
+    if not result["success"]:
+        console.print(Panel.fit(
+            f"[bold red]❌ Could not fetch databases[/bold red]\n\n"
+            f"[red]{result['error']}[/red]\n\n"
+            "[yellow]Common issues:[/yellow]\n"
+            "• Check your integration key\n"
+            "• Make sure your integration has access to databases\n"
+            "• Visit your integration settings to grant access",
+            title="Connection Error"
+        ))
+        return
+    
+    databases = result["databases"]
+    
+    if not databases:
+        console.print(Panel.fit(
+            "[bold yellow]⚠️  No databases found[/bold yellow]\n\n"
+            "Your integration key is valid, but no databases are accessible.\n\n"
+            "[bold blue]To grant access:[/bold blue]\n"
+            "1. Go to any Notion database\n"
+            "2. Click the '...' menu (top right)\n"
+            "3. Go to 'Add connections'\n"
+            "4. Find and select your integration\n\n"
+            "[cyan]https://www.notion.so/help/add-and-manage-connections-with-the-api[/cyan]",
+            title="No Databases Available"
+        ))
+        return
+    
+    # Create beautiful table
+    table = Table(
+        title=f"📊 Available Databases ({len(databases)} found)",
+        box=box.ROUNDED,
+        title_style="bold green",
+        header_style="bold cyan"
+    )
+    
+    table.add_column("#", style="dim", justify="right", width=3)
+    table.add_column("Database Name", style="bold white", min_width=25)
+    table.add_column("ID", style="dim cyan", min_width=36)
+    
+    for i, (title, db_id) in enumerate(databases, 1):
+        table.add_row(
+            str(i),
+            title or "Untitled Database",
+            db_id
+        )
+    
+    console.print("\n")
+    console.print(table)
+    
+    # Success message with next steps
+    console.print(Panel.fit(
+        "[bold green]🎉 Setup Complete![/bold green]\n\n"
+        f"Found [bold cyan]{len(databases)}[/bold cyan] accessible database(s).\n"
+        "You can now use notionary in your Python code!\n\n"
+        "[bold yellow]💡 Tip:[/bold yellow] Run [cyan]notionary db[/cyan] anytime to see this overview again.",
+        title="Ready to Go!"
+    ))
+
 @click.group()
 @click.version_option()  # Automatische Version aus setup.py
-def main():
+@click.option('--verbose', '-v', is_flag=True, help='Enable verbose logging')
+def main(verbose):
     """
     Notionary CLI - Notion API Integration
     """
+    if verbose:
+        enable_verbose_logging()
+        console.print("[dim]Verbose logging enabled[/dim]")
     pass
 
 @main.command()
@@ -70,11 +208,19 @@ def init():
             title="Already Configured"
         ))
         
-        # Option to reconfigure
-        if Confirm.ask("\n[yellow]Would you like to update your key?[/yellow]"):
+        # Option to reconfigure or show databases
+        choice = Prompt.ask(
+            "\n[yellow]What would you like to do?[/yellow]",
+            choices=["show", "update", "exit"],
+            default="show"
+        )
+        
+        if choice == "show":
+            show_databases_overview(existing_key)
+        elif choice == "update":
             setup_new_key()
         else:
-            console.print("\n[blue]No changes made. Happy coding! 🚀[/blue]")
+            console.print("\n[blue]Happy coding! 🚀[/blue]")
     else:
         # No key found, start setup
         console.print(Panel.fit(
@@ -86,6 +232,23 @@ def init():
         ))
         setup_new_key()
 
+@main.command()
+def db() -> None:
+    """
+    Show available Notion databases
+    """
+    existing_key = get_notion_secret()
+    
+    if not existing_key:
+        console.print(Panel.fit(
+            "[bold red]❌ No Integration Key found![/bold red]\n\n"
+            "Please run [cyan]notionary init[/cyan] first to set up your key.",
+            title="Not Configured"
+        ))
+        return
+    
+    show_databases_overview(existing_key)
+
 def setup_new_key():
     """Handle the key setup process"""
     try:
@@ -126,7 +289,8 @@ def setup_new_key():
         
         # Save the key
         if save_integration_key(integration_key):
-            return  # Success!
+            # Show databases overview after successful setup
+            show_databases_overview(integration_key)
         
     except KeyboardInterrupt:
         console.print("\n[yellow]Setup cancelled.[/yellow]")
@@ -170,7 +334,6 @@ def save_integration_key(integration_key: str) -> bool:
         if written_key == integration_key:
             console.print("\n[bold green]✅ Integration Key saved and verified![/bold green]")
             console.print(f"[dim]Configuration: {env_file}[/dim]")
-            console.print("\n[blue]Ready to use notionary in your Python code! 🚀[/blue]")
             return True
         else:
             console.print("\n[bold red]❌ Error: Key verification failed![/bold red]")

notionary/cli/onboarding.py

@@ -0,0 +1,116 @@
+import asyncio
+from dataclasses import dataclass
+from notionary import NotionDatabase
+
+@dataclass  
+class OnboardingPageResult:
+    url: str
+    tile: str
+    emoji: str
+
+async def generate_doc_for_database(
+    datbase_name: str,
+) -> OnboardingPageResult:
+    database = await NotionDatabase.from_database_name(datbase_name)
+    page = await database.create_blank_page()
+    
+    page_title = "Welcome to Notionary!"
+    page_icon = "📚"
+    
+    markdown_content = """!> [🚀] This page was created fully automatically and serves as a showcase of what is possible with Notionary.
+
+    ---
+
+    ## 🗃️ Working with Databases
+
+    Discover and manage your Notion databases programmatically:
+
+    ```python
+    import asyncio
+    from notionary import NotionDatabase, DatabaseDiscovery
+
+    async def main():
+    # Discover available databases
+    discovery = DatabaseDiscovery()
+    await discovery()
+
+    # Connect to a database by name
+    db = await NotionDatabase.from_database_name("Projects")
+
+    # Create a new page in the database
+    page = await db.create_blank_page()
+
+    # Query pages from database
+    async for page in db.iter_pages():
+        title = await page.get_title()
+        print(f"Page: {title}")
+
+    if __name__ == "__main__":
+    asyncio.run(main())
+    ```
+
+    ## 📄 Creating and Managing Pages
+    Create and update Notion pages with rich content:
+    ```python
+    import asyncio
+    from notionary import NotionPage
+
+    async def main():
+        # Create a page from URL
+        page = NotionPage.from_url("https://www.notion.so/your-page-url")
+
+        # Or find by name
+        page = await NotionPage.from_page_name("My Project Page")
+
+        # Update page metadata
+        await page.set_title("Updated Title")
+        await page.set_emoji_icon("🚀")
+        await page.set_random_gradient_cover()
+
+        # Add markdown content
+        markdown = '''
+        # Project Overview
+
+        !> [💡] This page was created programmatically using Notionary.
+
+        ## Features
+        - **Rich** Markdown support
+        - Async functionality
+        - Custom syntax extensions
+        '''
+
+        await page.replace_content(markdown)
+
+    if __name__ == "__main__":
+        asyncio.run(main())
+    ```
+
+    ## 📊 Tables and Structured Data
+    Create tables for organizing information:
+    FeatureStatusPriorityAPI IntegrationCompleteHighDocumentationIn ProgressMediumDatabase QueriesCompleteHighFile UploadsCompleteMedium
+
+    🎥 Media Embedding
+    Embed videos directly in your pages:
+    @[Caption](https://www.youtube.com/watch?v=dQw4w9WgXcQ) - Never gonna give you up!
+
+    Happy building with Notionary! 🎉"""
+
+    
+    await page.set_title(page_title)
+    await page.set_emoji_icon(page_icon)
+    await page.set_random_gradient_cover()
+    await page.append_markdown(markdown_content)
+    
+    url = await page.get_url()
+    
+    return OnboardingPageResult(
+        url=url,
+        tile=page_title,
+        emoji=page_icon,
+    )
+    
+    
+if __name__ == "__main__":
+    print("🚀 Starting Notionary onboarding page generation...")
+    result = asyncio.run(generate_doc_for_database("Wissen & Notizen"))
+    print(f"✅ Onboarding page created: {result.tile} {result.emoji} - {result.url}")

notionary/database/notion_database.py

@@ -3,7 +3,6 @@ from typing import Any, AsyncGenerator, Dict, List, Optional
 
 from notionary.notion_client import NotionClient
 from notionary.page.notion_page import NotionPage
-from notionary.telemetry import NotionaryTelemetry
 from notionary.util.warn_direct_constructor_usage import warn_direct_constructor_usage
 from notionary.util import LoggingMixin
 from notionary.util.page_id_utils import format_uuid
@@ -26,7 +25,6 @@ class NotionDatabase(LoggingMixin):
             token: Optional Notion API token
         """
         self.database_id = database_id
-        self._telemetry = NotionaryTelemetry()
         self._client = NotionClient(token=token)
 
     @classmethod

notionary/database/notion_database_factory.py

@@ -10,7 +10,6 @@ from notionary.exceptions.database_exceptions import (
     DatabaseParsingError,
     NotionDatabaseException,
 )
-from notionary.telemetry import track_usage
 from notionary.util import LoggingMixin
 from notionary.util import format_uuid
 from notionary.util import singleton
@@ -24,7 +23,6 @@ class NotionDatabaseFactory(LoggingMixin):
     """
 
     @classmethod
-    @track_usage('page_factory_method_used', {'method': 'from_page_id'})
     def from_database_id(
         cls, database_id: str, token: Optional[str] = None
     ) -> NotionDatabase:
@@ -59,7 +57,6 @@ class NotionDatabaseFactory(LoggingMixin):
             raise DatabaseConnectionError(error_msg) from e
 
     @classmethod
-    @track_usage('page_factory_method_used', {'method': 'from_url'})
     async def from_database_name(
         cls, database_name: str, token: Optional[str] = None
     ) -> NotionDatabase:

notionary/elements/audio_element.py

@@ -1,12 +1,11 @@
 import re
 from typing import Dict, Any, Optional, List
-from notionary.elements.notion_block_element import NotionBlockElement, auto_track_conversions
+from notionary.elements.notion_block_element import NotionBlockElement
 from notionary.prompting.element_prompt_content import (
     ElementPromptBuilder,
     ElementPromptContent,
 )
 
-@auto_track_conversions
 class AudioElement(NotionBlockElement):
     """
     Handles conversion between Markdown audio embeds and Notion audio blocks.

notionary/elements/bookmark_element.py

@@ -1,13 +1,12 @@
 import re
 from typing import Dict, Any, Optional, List, Tuple
 
-from notionary.elements.notion_block_element import NotionBlockElement, auto_track_conversions
+from notionary.elements.notion_block_element import NotionBlockElement
 from notionary.prompting.element_prompt_content import (
     ElementPromptBuilder,
     ElementPromptContent,
 )
 
-@auto_track_conversions
 class BookmarkElement(NotionBlockElement):
     """
     Handles conversion between Markdown bookmarks and Notion bookmark blocks.

notionary/elements/bulleted_list_element.py

@@ -1,13 +1,12 @@
 import re
 from typing import Dict, Any, Optional
-from notionary.elements.notion_block_element import NotionBlockElement, auto_track_conversions
+from notionary.elements.notion_block_element import NotionBlockElement
 from notionary.prompting.element_prompt_content import (
     ElementPromptBuilder,
     ElementPromptContent,
 )
 from notionary.elements.text_inline_formatter import TextInlineFormatter
 
-@auto_track_conversions
 class BulletedListElement(NotionBlockElement):
     """Class for converting between Markdown bullet lists and Notion bulleted list items."""
 

notionary/elements/callout_element.py

@@ -6,9 +6,8 @@ from notionary.prompting.element_prompt_content import (
     ElementPromptContent,
 )
 from notionary.elements.text_inline_formatter import TextInlineFormatter
-from notionary.elements.notion_block_element import NotionBlockElement, auto_track_conversions
+from notionary.elements.notion_block_element import NotionBlockElement
 
-@auto_track_conversions
 class CalloutElement(NotionBlockElement):
     """
     Handles conversion between Markdown callouts and Notion callout blocks.

notionary/elements/code_block_element.py

@@ -1,12 +1,11 @@
 import re
 from typing import Dict, Any, Optional, List, Tuple
-from notionary.elements.notion_block_element import NotionBlockElement, auto_track_conversions
+from notionary.elements.notion_block_element import NotionBlockElement
 from notionary.prompting.element_prompt_content import (
     ElementPromptBuilder,
     ElementPromptContent,
 )
 
-@auto_track_conversions
 class CodeBlockElement(NotionBlockElement):
     """
     Handles conversion between Markdown code blocks and Notion code blocks.

notionary/elements/column_element.py

@@ -1,14 +1,13 @@
 import re
 from typing import Dict, Any, Optional, List, Tuple, Callable
 
-from notionary.elements.notion_block_element import NotionBlockElement, auto_track_conversions
+from notionary.elements.notion_block_element import NotionBlockElement
 from notionary.page.formatting.spacer_rules import SPACER_MARKER
 from notionary.prompting.element_prompt_content import (
     ElementPromptBuilder,
     ElementPromptContent,
 )
 
-@auto_track_conversions
 class ColumnElement(NotionBlockElement):
     """
     Handles conversion between custom Markdown column syntax and Notion column blocks.

notionary/elements/divider_element.py

@@ -1,13 +1,12 @@
 import re
 from typing import Dict, Any, Optional
 
-from notionary.elements.notion_block_element import NotionBlockElement, auto_track_conversions
+from notionary.elements.notion_block_element import NotionBlockElement
 from notionary.prompting.element_prompt_content import (
     ElementPromptBuilder,
     ElementPromptContent,
 )
 
-@auto_track_conversions
 class DividerElement(NotionBlockElement):
     """
     Handles conversion between Markdown horizontal dividers and Notion divider blocks.

notionary/elements/embed_element.py

@@ -1,12 +1,11 @@
 import re
 from typing import Dict, Any, Optional, List
-from notionary.elements.notion_block_element import NotionBlockElement, auto_track_conversions
+from notionary.elements.notion_block_element import NotionBlockElement
 from notionary.prompting.element_prompt_content import (
     ElementPromptBuilder,
     ElementPromptContent,
 )
 
-@auto_track_conversions
 class EmbedElement(NotionBlockElement):
     """
     Handles conversion between Markdown embeds and Notion embed blocks.

notionary/elements/heading_element.py

@@ -1,14 +1,13 @@
 import re
 from typing import Dict, Any, Optional
 
-from notionary.elements.notion_block_element import NotionBlockElement, auto_track_conversions
+from notionary.elements.notion_block_element import NotionBlockElement
 from notionary.prompting.element_prompt_content import (
     ElementPromptBuilder,
     ElementPromptContent,
 )
 from notionary.elements.text_inline_formatter import TextInlineFormatter
 
-@auto_track_conversions
 class HeadingElement(NotionBlockElement):
     """Handles conversion between Markdown headings and Notion heading blocks."""
 

notionary/elements/image_element.py

@@ -1,12 +1,11 @@
 import re
 from typing import Dict, Any, Optional, List
-from notionary.elements.notion_block_element import NotionBlockElement, auto_track_conversions
+from notionary.elements.notion_block_element import NotionBlockElement
 from notionary.prompting.element_prompt_content import (
     ElementPromptBuilder,
     ElementPromptContent,
 )
 
-@auto_track_conversions
 class ImageElement(NotionBlockElement):
     """
     Handles conversion between Markdown images and Notion image blocks.

notionary/elements/mention_element.py

@@ -2,13 +2,12 @@ import re
 from typing import Dict, Any, Optional, List
 from typing_extensions import override
 
-from notionary.elements.notion_block_element import NotionBlockElement, auto_track_conversions
+from notionary.elements.notion_block_element import NotionBlockElement
 from notionary.prompting.element_prompt_content import (
     ElementPromptBuilder,
     ElementPromptContent,
 )
 
-@auto_track_conversions
 class MentionElement(NotionBlockElement):
     """
     Handles conversion between Markdown mentions and Notion mention elements.

notionary/elements/notion_block_element.py

@@ -3,7 +3,6 @@ from typing import Dict, Any, Optional
 from abc import ABC
 
 from notionary.prompting.element_prompt_content import ElementPromptContent
-from notionary.telemetry import track_usage
 
 
 class NotionBlockElement(ABC):
@@ -33,38 +32,4 @@ class NotionBlockElement(ABC):
 
     @classmethod
     def get_llm_prompt_content(cls) -> ElementPromptContent:
-        """Returns a dictionary with information for LLM prompts about this element."""
-
-
-def auto_track_conversions(cls):
-    """
-    Decorator der sich auch auf Subklassen vererbt.
-    """
-    conversion_methods = ['markdown_to_notion', 'notion_to_markdown']
-    
-    original_init_subclass = getattr(cls, '__init_subclass__', None)
-    
-    @classmethod
-    def __init_subclass__(cls_inner, **kwargs):
-        # Original __init_subclass__ aufrufen
-        if original_init_subclass:
-            original_init_subclass(**kwargs)
-        
-        # Tracking für Subklasse hinzufügen
-        for method_name in conversion_methods:
-            if hasattr(cls_inner, method_name):
-                original_method = getattr(cls_inner, method_name)
-                
-                if isinstance(original_method, classmethod):
-                    func = original_method.__func__
-                    
-                    @track_usage(f"{cls_inner.__name__.lower()}_{method_name}")
-                    @classmethod
-                    @wraps(func)
-                    def tracked_method(cls_ref, *args, **kwargs):
-                        return func(cls_ref, *args, **kwargs)
-                    
-                    setattr(cls_inner, method_name, tracked_method)
-    
-    cls.__init_subclass__ = __init_subclass__
-    return cls
+        """Returns a dictionary with information for LLM prompts about this element."""

notionary/elements/numbered_list_element.py

@@ -1,13 +1,12 @@
 import re
 from typing import Dict, Any, Optional
-from notionary.elements.notion_block_element import NotionBlockElement, auto_track_conversions
+from notionary.elements.notion_block_element import NotionBlockElement
 from notionary.prompting.element_prompt_content import (
     ElementPromptBuilder,
     ElementPromptContent,
 )
 from notionary.elements.text_inline_formatter import TextInlineFormatter
 
-@auto_track_conversions
 class NumberedListElement(NotionBlockElement):
     """Class for converting between Markdown numbered lists and Notion numbered list items."""
 

notionary/elements/paragraph_element.py

@@ -1,13 +1,12 @@
 from typing import Dict, Any, Optional
 
-from notionary.elements.notion_block_element import NotionBlockElement, auto_track_conversions
+from notionary.elements.notion_block_element import NotionBlockElement
 from notionary.prompting.element_prompt_content import (
     ElementPromptBuilder,
     ElementPromptContent,
 )
 from notionary.elements.text_inline_formatter import TextInlineFormatter
 
-@auto_track_conversions
 class ParagraphElement(NotionBlockElement):
     """Handles conversion between Markdown paragraphs and Notion paragraph blocks."""
 

notionary/elements/qoute_element.py

@@ -1,12 +1,11 @@
 import re
 from typing import Dict, Any, Optional, List, Tuple
-from notionary.elements.notion_block_element import NotionBlockElement, auto_track_conversions
+from notionary.elements.notion_block_element import NotionBlockElement
 from notionary.prompting.element_prompt_content import (
     ElementPromptBuilder,
     ElementPromptContent,
 )
 
-@auto_track_conversions
 class QuoteElement(NotionBlockElement):
     """Class for converting between Markdown blockquotes and Notion quote blocks."""
 

notionary/elements/table_element.py

@@ -1,13 +1,12 @@
 import re
 from typing import Dict, Any, Optional, List, Tuple
-from notionary.elements.notion_block_element import NotionBlockElement, auto_track_conversions
+from notionary.elements.notion_block_element import NotionBlockElement
 from notionary.elements.text_inline_formatter import TextInlineFormatter
 from notionary.prompting.element_prompt_content import (
     ElementPromptBuilder,
     ElementPromptContent,
 )
 
-@auto_track_conversions
 class TableElement(NotionBlockElement):
     """
     Handles conversion between Markdown tables and Notion table blocks.

notionary/elements/todo_element.py

@@ -1,13 +1,12 @@
 import re
 from typing import Dict, Any, Optional
-from notionary.elements.notion_block_element import NotionBlockElement, auto_track_conversions
+from notionary.elements.notion_block_element import NotionBlockElement
 from notionary.prompting.element_prompt_content import (
     ElementPromptBuilder,
     ElementPromptContent,
 )
 from notionary.elements.text_inline_formatter import TextInlineFormatter
 
-@auto_track_conversions
 class TodoElement(NotionBlockElement):
     """
     Handles conversion between Markdown todo items and Notion to_do blocks.

notionary/elements/toggle_element.py

@@ -1,13 +1,12 @@
 import re
 from typing import Dict, Any, Optional, List, Tuple, Callable
 
-from notionary.elements.notion_block_element import NotionBlockElement, auto_track_conversions
+from notionary.elements.notion_block_element import NotionBlockElement
 from notionary.prompting.element_prompt_content import (
     ElementPromptBuilder,
     ElementPromptContent,
 )
 
-@auto_track_conversions
 class ToggleElement(NotionBlockElement):
     """
     Improved ToggleElement class using pipe syntax instead of indentation.

notionary/elements/toggleable_heading_element.py

@@ -1,14 +1,13 @@
 import re
 from typing import Dict, Any, Optional, List, Tuple, Callable
 
-from notionary.elements.notion_block_element import NotionBlockElement, auto_track_conversions
+from notionary.elements.notion_block_element import NotionBlockElement
 from notionary.prompting.element_prompt_content import (
     ElementPromptBuilder,
     ElementPromptContent,
 )
 from notionary.elements.text_inline_formatter import TextInlineFormatter
 
-@auto_track_conversions
 class ToggleableHeadingElement(NotionBlockElement):
     """Handles conversion between Markdown collapsible headings and Notion toggleable heading blocks with pipe syntax."""
 

notionary/elements/video_element.py

@@ -1,12 +1,11 @@
 import re
 from typing import Dict, Any, Optional, List
-from notionary.elements.notion_block_element import NotionBlockElement, auto_track_conversions
+from notionary.elements.notion_block_element import NotionBlockElement
 from notionary.prompting.element_prompt_content import (
     ElementPromptBuilder,
     ElementPromptContent,
 )
 
-@auto_track_conversions
 class VideoElement(NotionBlockElement):
     """
     Handles conversion between Markdown video embeds and Notion video blocks.

notionary/page/notion_page_factory.py

@@ -2,7 +2,6 @@ from typing import List, Optional, Dict, Any, Tuple
 from difflib import SequenceMatcher
 
 from notionary import NotionPage, NotionClient
-from notionary.telemetry import track_usage
 from notionary.util import LoggingMixin
 from notionary.util import format_uuid, extract_and_validate_page_id
 from notionary.util import singleton
@@ -20,7 +19,6 @@ class NotionPageFactory(LoggingMixin):
     EARLY_STOP_THRESHOLD = 0.95
 
     @classmethod
-    @track_usage('page_factory_method_used', {'method': 'from_page_id'})
     def from_page_id(cls, page_id: str, token: Optional[str] = None) -> NotionPage:
         """Create a NotionPage from a page ID."""
         
@@ -36,7 +34,6 @@ class NotionPageFactory(LoggingMixin):
             raise
 
     @classmethod
-    @track_usage('page_factory_method_used', {'method': 'from_url'})
     def from_url(cls, url: str, token: Optional[str] = None) -> NotionPage:
         """Create a NotionPage from a Notion URL."""
 
@@ -56,7 +53,6 @@ class NotionPageFactory(LoggingMixin):
             raise
 
     @classmethod
-    @track_usage('page_factory_method_used', {'method': 'from_page_name'})
     async def from_page_name(
         cls, page_name: str, token: Optional[str] = None
     ) -> NotionPage:

{notionary-0.2.12.dist-info → notionary-0.2.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: notionary
-Version: 0.2.12
+Version: 0.2.13
 Summary: A toolkit to convert between Markdown and Notion blocks
 Home-page: https://github.com/mathisarends/notionary
 Author: Mathis Arends

{notionary-0.2.12.dist-info → notionary-0.2.13.dist-info}/RECORD

@@ -1,32 +1,32 @@
-notionary/__init__.py,sha256=I4kPJ_LEy_IIQuBVGZ-U_kiIUWeRUP9DZ_F3w9BvS-4,809
+notionary/__init__.py,sha256=U7I4nffaEt1Gfg6N7TFupC_GQVZttLhW5u0qjVX7gP4,717
 notionary/notion_client.py,sha256=gkREAr8LkUUKK9cOvq72r8jNjlXDleBP2fYm7LjjbjM,7311
-notionary/cli/main.py,sha256=GmUfdrJZjHYo689ZhrMuA4Lr_sBE2YeqrkFloXgrfvY,6663
-notionary/cli/onboarding.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+notionary/cli/main.py,sha256=0tmX9y_xJksKPV8evfUIxp2MTKsHAvNcq76CW4PZCSs,12836
+notionary/cli/onboarding.py,sha256=I7G6eaEw_WWmbfBKBa4nHgItGyZg6iwtEw9pZylsqa4,3376
 notionary/database/database_discovery.py,sha256=l9IjthwdA_Y_k_JXcAW-KnvZDwNYylIbsrQ5cpgtb5w,4484
-notionary/database/notion_database.py,sha256=yTGMK1EOuJDNiP6boc_dvWDLTTEhOOYa_ZdJWiOP19c,7695
-notionary/database/notion_database_factory.py,sha256=pOu4MDu7jaSkNQHk-TSEezsJ1pPGJXiSTOouygh2JuY,6783
+notionary/database/notion_database.py,sha256=vbMu8FRao0nXRkSK68Q6637ypiikyMpP2zOmzGWsgpU,7595
+notionary/database/notion_database_factory.py,sha256=6AK9c63R8qyQX46VY-eNL1DufHe6pK_Erka0QEBVO90,6594
 notionary/database/models/page_result.py,sha256=Vmm5_oYpYAkIIJVoTd1ZZGloeC3cmFLMYP255mAmtaw,233
-notionary/elements/audio_element.py,sha256=wXSo0DQs_KIVdLB-Ax5Yk4puW-_gFwAbJHcgcpwZjoA,5392
-notionary/elements/bookmark_element.py,sha256=bDJEMTLC1E3qUIjphFJC9l60ZAeJqZOWEvS__Y5Iq00,8192
-notionary/elements/bulleted_list_element.py,sha256=Ax6tIGtc92O57QKVWRKi2Sgkbmd6KqkSmssTcj17Tjg,2711
-notionary/elements/callout_element.py,sha256=w45gJogoqf4Z9aVqSq7toaU06IWWM_et5dCWywj_oeo,4539
-notionary/elements/code_block_element.py,sha256=PrsV58fzblmZz0vRkuAG2TT0kpiMglq0UlKaR-B-dJw,7576
-notionary/elements/column_element.py,sha256=yfWSLEClcxrWLteQClabaIvjay2h8krB6u5rfbnBRBI,12781
-notionary/elements/divider_element.py,sha256=bdGTgMgW8Uk6Fu7oy2UBrDaQ_2XiLzPNwF3JmrBS00I,2379
-notionary/elements/embed_element.py,sha256=QYyiKTbAM6COPqe61a3jSg6m2a4orUpA6l37lrn4-G0,4626
-notionary/elements/heading_element.py,sha256=XDO_hgiG-J2nVF3hwcpvsXte6uGDAAwyHx661lD-1H8,3215
-notionary/elements/image_element.py,sha256=agV8ZNwoJNOIvaJNARs2osdbe1T5MHPNbrzu7L19uRI,4805
-notionary/elements/mention_element.py,sha256=-8GbeaiGNglw5MZulomSa-hUh32HaZowAROeCgzMqdA,8274
-notionary/elements/notion_block_element.py,sha256=cou_87mAkj7U3LLU0kXJrbxVl8VusnlVBYYb4cchOPo,2603
-notionary/elements/numbered_list_element.py,sha256=iiDzjOwc5Fnv2RQsaTLeOtRVQdshG4rRQJa4LMJjCvY,2710
-notionary/elements/paragraph_element.py,sha256=EWUac9oq5zXso0svE5vTKNWlpq6sRdJxxVWYMEmYLz4,3302
-notionary/elements/qoute_element.py,sha256=cIa7S0G9Kb4qR5EXWOLJGPn0IzWm1zLujnzzO6t4-D0,6169
-notionary/elements/table_element.py,sha256=acIEoDz8tblj9pPmE5Gr0Ov-S9qJOTkEDNcUPBfsN4I,11296
+notionary/elements/audio_element.py,sha256=m1zMUYkpQNPCx9S0KpU85ps7pnm9Pzu4lglKYzOozF4,5343
+notionary/elements/bookmark_element.py,sha256=M_vGJfBVVhUDa7sDfHB622m1Q9wEn4Lw0pCmuDkxpvQ,8143
+notionary/elements/bulleted_list_element.py,sha256=dyWNu28l_fG6fob5zQzMadnWb5g_B2n7ixCHwaniPdE,2662
+notionary/elements/callout_element.py,sha256=OVaRLdxNFXTJnMzmfg3SZClOAYfJ4oORSr_Skm5EOPA,4490
+notionary/elements/code_block_element.py,sha256=hBMn3VpFJLgq12llvdSFkjLX05poMCIZWsBaUF9ZOQA,7527
+notionary/elements/column_element.py,sha256=VpPe2yVvozMFIladOMxMh0Q1nsND_XysYxpxWoIH8eU,12732
+notionary/elements/divider_element.py,sha256=sqfs1YRVqsEFlKdBOX9p8K2Ise2lDb19HwcwHM_R7nU,2330
+notionary/elements/embed_element.py,sha256=PkoaycRel6bAVkP1XHv57vuLVwFmzgeZYKqM5AT8Lg0,4577
+notionary/elements/heading_element.py,sha256=zgqC6alKoaucpeWDpDJd-TIsbMoJzNI5_BGP-D8Xbsc,3166
+notionary/elements/image_element.py,sha256=rd9P9BT9YUwofO24JadH7bMtQCJLxVHdjHF1_ykGS-g,4756
+notionary/elements/mention_element.py,sha256=-AVZ8rn6y-S9Paw-vUDZpa8nhn0dR5L_0qNjd9dQI_s,8225
+notionary/elements/notion_block_element.py,sha256=utWHp_JZSRQgNIWiZfJfzwl7y2ALuDqGTY-Ve3OoQeg,1285
+notionary/elements/numbered_list_element.py,sha256=kgJfQQ5FIKZVRjygd6sLJKkE9-IDlx-MFAEaxWVsg84,2661
+notionary/elements/paragraph_element.py,sha256=oOIRkknpEEAw5Pr81D6c-L6cGRvaZGtggLT3iVYGC4Q,3253
+notionary/elements/qoute_element.py,sha256=SCvNWht-38EwMIsDg19VPajKNc1IhZBP3oVxLQw_yVA,6120
+notionary/elements/table_element.py,sha256=6yTY0fNT_Ae2ncwGQE0CSsfCxGNvnDblTTzd3kdz-Sw,11247
 notionary/elements/text_inline_formatter.py,sha256=q1WePwTxhSkjhTFylcyAbhxaWLo_sjYS3Q_mIPgsKb4,8625
-notionary/elements/todo_element.py,sha256=r7H1KcEW7Iq6pQrhsPRBwHn_3ok26tR2fJf53U_kr5I,4171
-notionary/elements/toggle_element.py,sha256=caPT8wAo7ccwT2pzUzySrImnUlVB2cigFBGe32vwjJk,11129
-notionary/elements/toggleable_heading_element.py,sha256=mUzR1HrkOJEgzjIgVgtInvLoi4YzsPJ0pR3FO_OlD5U,10007
-notionary/elements/video_element.py,sha256=Y2JOvb4fovnY2nPAtTPSbdtk-HaRdWqg6RXr-_a2WOw,5774
+notionary/elements/todo_element.py,sha256=BKw3KvjdDsZIlSXoqV52z9R-a3KBzkH-9DKo4dN7m9w,4122
+notionary/elements/toggle_element.py,sha256=sg3LfblBRa0pIrcSBHPRknq6FYWB5-mpDGvA7PBOZ9U,11080
+notionary/elements/toggleable_heading_element.py,sha256=sKvjD_x61QYOM4l8gfq6VodpLWqxf8FR2JGWXNeee08,9958
+notionary/elements/video_element.py,sha256=IlB88CmBueYPSFh6p7kxE5zVjcZBQJJ1K953G7kg99M,5725
 notionary/elements/registry/block_registry.py,sha256=g0id_Q6guzTyNY6HfnB9AjOBvCR4CvtpnUeFAY8kgY0,5027
 notionary/elements/registry/block_registry_builder.py,sha256=5zRKnw2102rAeHpANs6Csu4DVufOazf1peEovChWcgs,9572
 notionary/exceptions/database_exceptions.py,sha256=I-Tx6bYRLpi5pjGPtbT-Mqxvz3BFgYTiuZxknJeLxtI,2638
@@ -35,7 +35,7 @@ notionary/models/notion_block_response.py,sha256=gzL4C6K9QPcaMS6NbAZaRceSEnMbNwY
 notionary/models/notion_database_response.py,sha256=FMAasQP20S12J_KMdMlNpcHHwxFKX2YtbE4Q9xn-ruQ,1213
 notionary/models/notion_page_response.py,sha256=r4fwMwwDocj92JdbSmyrzIqBKsnEaz4aDUiPabrg9BM,1762
 notionary/page/notion_page.py,sha256=CnEr5S425t7r8n4mZERwShlXsXnR2G7bbYjO8yb2oaU,18032
-notionary/page/notion_page_factory.py,sha256=p75yhr6FdYl9aG8Ttpf5pyuQiMJB0CAYMkfcVLI8HP0,12028
+notionary/page/notion_page_factory.py,sha256=_dsvxn3xmjZFQw3fKIhnTwiIQfk7p8pxVOFj1GMvBZc,11763
 notionary/page/notion_to_markdown_converter.py,sha256=vUQss0J7LUFLULGvW27PjaTFuWi8OsRQAUBowSYorkM,6408
 notionary/page/content/notion_page_content_chunker.py,sha256=kWJnV9GLU5YLgSVPKOjwMBbG_CMAmVRkuDtwJYb_UAA,3316
 notionary/page/content/page_content_retriever.py,sha256=MoRNwVyBacQEPFu-XseahKEFait0q8tjuhFUXHOBrMo,2208
@@ -54,17 +54,14 @@ notionary/page/relations/notion_page_title_resolver.py,sha256=LN89y-Tc0Rk81TiTeA
 notionary/page/relations/page_database_relation.py,sha256=nkelofYzfuIFjmM7vR6IGJpWUG9XPmSDnU1WR8WtQrs,2231
 notionary/prompting/element_prompt_content.py,sha256=tHref-SKA81Ua_IQD2Km7y7BvFtHl74haSIjHNYE3FE,4403
 notionary/prompting/markdown_syntax_prompt_generator.py,sha256=_1qIYlqSfI6q6Fut10t6gGwTQuS8c3QBcC_5DBme9Mo,5084
-notionary/telemetry/__init__.py,sha256=qdFGrhOhUb_HPMLS4kLSr1BJNfb5-LzHwLmx-NYCDsw,156
-notionary/telemetry/telemetry.py,sha256=SEwEsegNhX7xblKnlk9qCDqKOTtf8sGtnNWbLiOVk7g,8182
-notionary/telemetry/track_usage_decorator.py,sha256=rmxwySb1PfZI1mXXU0G_W9Fpn6ByslYrDjsU4JYnMJ4,2421
 notionary/util/__init__.py,sha256=ra1jHFFiQNWYDzmVb81OVhtshzkZ0GcLVI8YDODYj3w,235
 notionary/util/logging_mixin.py,sha256=d5sRSmUtgQeuckdNBkO025IXPGe4oOb-7ueVAIP8amU,1846
 notionary/util/page_id_utils.py,sha256=EYNMxgf-7ghzL5K8lKZBZfW7g5CsdY0Xuj4IYmU8RPk,1381
 notionary/util/singleton.py,sha256=CKAvykndwPRZsA3n3MAY_XdCR59MBjjKP0vtm2BcvF0,428
 notionary/util/warn_direct_constructor_usage.py,sha256=vyJR73F95XVSRWIbyij-82IGOpAne9SBPM25eDpZfSU,1715
-notionary-0.2.12.dist-info/licenses/LICENSE,sha256=zOm3cRT1qD49eg7vgw95MI79rpUAZa1kRBFwL2FkAr8,1120
-notionary-0.2.12.dist-info/METADATA,sha256=vZNayJWGNmJAJBfitVjQXYEkJC2YjJRE0fmEK6s3Aeg,7582
-notionary-0.2.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-notionary-0.2.12.dist-info/entry_points.txt,sha256=V7X21u3QNm7h7p6Cx0Sx2SO3mtmA7gVwXM8lNYnv9fk,54
-notionary-0.2.12.dist-info/top_level.txt,sha256=fhONa6BMHQXqthx5PanWGbPL0b8rdFqhrJKVLf_adSs,10
-notionary-0.2.12.dist-info/RECORD,,
+notionary-0.2.13.dist-info/licenses/LICENSE,sha256=zOm3cRT1qD49eg7vgw95MI79rpUAZa1kRBFwL2FkAr8,1120
+notionary-0.2.13.dist-info/METADATA,sha256=jXcnz9Gqxx8vFbnLdrLhr_TiByxrWd_OMIxPPqyqHGY,7582
+notionary-0.2.13.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+notionary-0.2.13.dist-info/entry_points.txt,sha256=V7X21u3QNm7h7p6Cx0Sx2SO3mtmA7gVwXM8lNYnv9fk,54
+notionary-0.2.13.dist-info/top_level.txt,sha256=fhONa6BMHQXqthx5PanWGbPL0b8rdFqhrJKVLf_adSs,10
+notionary-0.2.13.dist-info/RECORD,,

notionary/telemetry/__init__.py

@@ -1,7 +0,0 @@
-from .telemetry import NotionaryTelemetry
-from .track_usage_decorator import track_usage
-
-__all__ = [
-    "NotionaryTelemetry",
-    "track_usage",
-]

notionary/telemetry/telemetry.py

@@ -1,226 +0,0 @@
-import os
-import uuid
-import atexit
-import signal
-import threading
-from pathlib import Path
-from typing import Dict, Any, Optional
-from posthog import Posthog
-from dotenv import load_dotenv
-
-from notionary.util import LoggingMixin
-from notionary.util import singleton
-
-load_dotenv()
-
-@singleton
-class NotionaryTelemetry(LoggingMixin):
-    """
-    Anonymous telemetry for Notionary - enabled by default.
-    Disable via: ANONYMIZED_TELEMETRY=false
-    """
-
-    USER_ID_PATH = str(Path.home() / ".cache" / "notionary" / "telemetry_user_id")
-    PROJECT_API_KEY = (
-        "phc_gItKOx21Tc0l07C1taD0QPpqFnbWgWjVfRjF6z24kke"  # write-only so no worries
-    )
-    HOST = "https://eu.i.posthog.com"
-    
-    _logged_init_message = False
-
-    def __init__(self):
-        # Default: enabled, disable via ANONYMIZED_TELEMETRY=false
-        telemetry_setting = os.getenv("ANONYMIZED_TELEMETRY", "true").lower()
-        self.enabled = telemetry_setting != "false"
-
-        self._user_id = None
-        self._client = None
-        self._shutdown_lock = threading.Lock()
-        self._is_shutdown = False
-        self._shutdown_registered = False
-
-        if self.enabled:
-            self._initialize_client()
-            self._register_shutdown_handlers()
-
-    def _register_shutdown_handlers(self):
-        """Register shutdown handlers for clean exit"""
-        with self._shutdown_lock:
-            if self._shutdown_registered:
-                return
-            
-            try:
-                # Register atexit handler for normal program termination
-                atexit.register(self._atexit_handler)
-                
-                # Register signal handlers for SIGINT (Ctrl+C) and SIGTERM
-                signal.signal(signal.SIGINT, self._signal_handler)
-                signal.signal(signal.SIGTERM, self._signal_handler)
-                
-                self._shutdown_registered = True
-                self.logger.debug("Telemetry shutdown handlers registered")
-                
-            except Exception as e:
-                self.logger.debug(f"Failed to register shutdown handlers: {e}")
-
-    def _signal_handler(self, signum, frame):
-        """Handle SIGINT (Ctrl+C) and SIGTERM signals"""
-        signal_name = "SIGINT" if signum == signal.SIGINT else f"SIG{signum}"
-        self.logger.debug(f"Received {signal_name}, shutting down telemetry...")
-        
-        self.shutdown(timeout=5.0)  # Quick shutdown for signals
-        
-        # Let the original signal handler take over (or exit)
-        if signum == signal.SIGINT:
-            # Restore default handler and re-raise
-            signal.signal(signal.SIGINT, signal.SIG_DFL)
-            os.kill(os.getpid(), signal.SIGINT)
-
-    def _atexit_handler(self):
-        """Handle normal program exit"""
-        self.logger.debug("Normal program exit, shutting down telemetry...")
-        self.shutdown(timeout=10.0)
-
-    @property
-    def user_id(self) -> str:
-        """Anonymous, persistent user ID"""
-        if self._user_id:
-            return self._user_id
-
-        try:
-            if not os.path.exists(self.USER_ID_PATH):
-                os.makedirs(os.path.dirname(self.USER_ID_PATH), exist_ok=True)
-                with open(self.USER_ID_PATH, "w") as f:
-                    new_user_id = str(uuid.uuid4())
-                    f.write(new_user_id)
-                self._user_id = new_user_id
-            else:
-                with open(self.USER_ID_PATH, "r") as f:
-                    self._user_id = f.read().strip()
-
-            return self._user_id
-        except Exception as e:
-            self.logger.debug(f"Error getting user ID: {e}")
-            return "anonymous_user"
-
-    def capture(self, event_name: str, properties: Optional[Dict[str, Any]] = None):
-        """
-        Safe event tracking that never affects library functionality
-
-        Args:
-            event_name: Event name (e.g. 'page_factory_used')
-            properties: Event properties as dictionary
-        """
-        if not self.enabled or not self._client or self._is_shutdown:
-            return
-
-        try:
-            # Add base properties
-            event_properties = {
-                "library": "notionary",
-                "library_version": self._get_notionary_version(),
-                **(properties or {}),
-            }
-
-            self._client.capture(
-                distinct_id=self.user_id, event=event_name, properties=event_properties
-            )
-
-        except Exception:
-            pass
-
-    def flush(self, timeout: float = 5.0):
-        """
-        Flush events with timeout
-        
-        Args:
-            timeout: Maximum time to wait for flush to complete
-        """
-        if not self.enabled or not self._client or self._is_shutdown:
-            return
-
-        try:
-            # PostHog flush doesn't support timeout directly, so we do it in a thread
-            flush_thread = threading.Thread(target=self._client.flush)
-            flush_thread.daemon = True
-            flush_thread.start()
-            flush_thread.join(timeout=timeout)
-            
-            if flush_thread.is_alive():
-                self.logger.warning(f"Telemetry flush timed out after {timeout}s")
-            else:
-                self.logger.debug("Telemetry events flushed successfully")
-                
-        except Exception as e:
-            self.logger.debug(f"Error during telemetry flush: {e}")
-
-    def shutdown(self, timeout: float = 10.0):
-        """
-        Clean shutdown of telemetry with timeout
-        
-        Args:
-            timeout: Maximum time to wait for shutdown
-        """
-        with self._shutdown_lock:
-            if self._is_shutdown:
-                return
-            
-            self._is_shutdown = True
-            
-        try:
-            if self._client:
-                # First try to flush remaining events
-                self.logger.debug("Flushing telemetry events before shutdown...")
-                self.flush(timeout=timeout * 0.7)  # Use 70% of timeout for flush
-                
-                # Then shutdown the client
-                shutdown_thread = threading.Thread(target=self._client.shutdown)
-                shutdown_thread.daemon = True
-                shutdown_thread.start()
-                shutdown_thread.join(timeout=timeout * 0.3)  # Use 30% for shutdown
-                
-                if shutdown_thread.is_alive():
-                    self.logger.warning(f"Telemetry client shutdown timed out after {timeout}s")
-                else:
-                    self.logger.debug("Telemetry client shut down successfully")
-                    
-        except Exception as e:
-            self.logger.debug(f"Error during telemetry shutdown: {e}")
-        finally:
-            self._client = None
-
-    def _initialize_client(self):
-        """Initializes PostHog client and shows startup message"""
-        try:
-            self._client = Posthog(
-                project_api_key=self.PROJECT_API_KEY,
-                host=self.HOST,
-                disable_geoip=True,
-            )
-            if not self._logged_init_message:
-                self.logger.info(
-                    "Anonymous telemetry enabled to improve Notionary. "
-                    "To disable: export ANONYMIZED_TELEMETRY=false"
-                )
-                self._logged_init_message = True
-
-            self._track_initialization()
-
-        except Exception as e:
-            self.logger.debug(f"Telemetry initialization failed: {e}")
-            self.enabled = False
-            self._client = None
-
-    def _track_initialization(self):
-        """Tracks library initialization"""
-        self.capture(
-            "notionary_initialized",
-            {
-                "version": self._get_notionary_version(),
-            },
-        )
-
-    def _get_notionary_version(self) -> str:
-        """Determines the Notionary version"""
-        import notionary
-        return getattr(notionary, "__version__", "0.2.10")

notionary/telemetry/track_usage_decorator.py

@@ -1,76 +0,0 @@
-from functools import wraps
-from typing import Any, Callable, Dict, Optional
-from notionary.telemetry import NotionaryTelemetry
-
-
-def track_usage(event_name: Optional[str] = None, properties: Optional[Dict[str, Any]] = None):
-    """
-    Simple decorator to track function usage.
-    
-    Args:
-        event_name: Custom event name (defaults to function name)
-        properties: Additional properties to track
-        
-    Usage:
-        @track_usage()
-        def my_function():
-            pass
-            
-        @track_usage('custom_event_name')
-        def my_function():
-            pass
-            
-        @track_usage('custom_event', {'feature': 'advanced'})
-        def my_function():
-            pass
-    """
-    def decorator(func: Callable) -> Callable:
-        @wraps(func)
-        def wrapper(*args, **kwargs):
-            telemetry = NotionaryTelemetry()
-            
-            # Generate event name and properties
-            event = event_name or _generate_event_name(func, args)
-            event_properties = _build_properties(func, args, properties)
-            
-            # Track and execute
-            telemetry.capture(event, event_properties)
-            return func(*args, **kwargs)
-        
-        return wrapper
-    return decorator
-
-
-def _get_class_name(func: Callable, args: tuple) -> Optional[str]:
-    """Extract class name from function or arguments."""
-    if args and hasattr(args[0], '__class__'):
-        return args[0].__class__.__name__
-    
-    if hasattr(func, '__qualname__') and '.' in func.__qualname__:
-        return func.__qualname__.split('.')[0]
-    
-    return None
-
-
-def _generate_event_name(func: Callable, args: tuple) -> str:
-    """Generate event name from function and class info."""
-    class_name = _get_class_name(func, args)
-    
-    if class_name:
-        return f"{class_name.lower()}_{func.__name__}_used"
-    
-    return f"{func.__name__}_used"
-
-
-def _build_properties(func: Callable, args: tuple, properties: Optional[Dict[str, Any]]) -> Dict[str, Any]:
-    """Build event properties with function and class info."""
-    event_properties = {
-        'function_name': func.__name__,
-        **(properties or {})
-    }
-    
-    class_name = _get_class_name(func, args)
-    if class_name:
-        event_properties['class_name'] = class_name
-    
-    return event_properties