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

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.11.dist-info → notionary-0.2.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: notionary
-Version: 0.2.11
+Version: 0.2.13
 Summary: A toolkit to convert between Markdown and Notion blocks
 Home-page: https://github.com/mathisarends/notionary
 Author: Mathis Arends
@@ -14,6 +14,7 @@ Requires-Dist: httpx>=0.28.0
 Requires-Dist: python-dotenv>=1.1.0
 Requires-Dist: pydantic>=2.11.4
 Requires-Dist: posthog>=3.0.0
+Requires-Dist: click>=8.0.0
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

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

@@ -1,30 +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=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=Z2mvdrsnZoOY58yTW7Ay2GFmDXRqVpHEaLQiZQuqa8M,7708
-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
@@ -33,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
@@ -52,16 +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.11.dist-info/licenses/LICENSE,sha256=zOm3cRT1qD49eg7vgw95MI79rpUAZa1kRBFwL2FkAr8,1120
-notionary-0.2.11.dist-info/METADATA,sha256=bPHXKSHttyuYUOKzPlc2cVT4WHlqk1BGTc6EFXIIjEk,7553
-notionary-0.2.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-notionary-0.2.11.dist-info/top_level.txt,sha256=fhONa6BMHQXqthx5PanWGbPL0b8rdFqhrJKVLf_adSs,10
-notionary-0.2.11.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-0.2.13.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+notionary = notionary.cli.main:main

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