i9dZddlmZddlmZddlmZ ddlmZ ddlm Z dZ n #e $rdZ YnwxYwGd d eeZGd d eZejdd fddZejdd fd dZd!dZd!dZd"dZd#dZd S)$aText collation support for string comparisons. This module provides collation (text comparison) functionality with optional PyICU support for advanced Unicode collation. Falls back to simple binary and case-insensitive comparisons when PyICU is not available. ) annotations)Callable)Enum)Collator)LocaleTFc"eZdZdZdZ dZ dZdS) CollationuText comparison collation strategies. For most users, use case_sensitive parameter in add_property_filter() instead of working with Collation directly. Examples: # Simple API (recommended for most users): searcher.add_property_filter("SUMMARY", "meeting", case_sensitive=False) # Advanced API (for power users): searcher.add_property_filter("SUMMARY", "Müller", collation=Collation.LOCALE, locale="de_DE") simpleunicodelocaleN)__name__ __module__ __qualname____doc__SIMPLEUNICODELOCALEW/root/projects/butler/venv/lib/python3.11/site-packages/icalendar_searcher/collation.pyr r s=  F G'F<>DrrN collationcase_sensitiveboolr str | NonereturnCallable[[str, str], bool]cP|tjkr|rtStS|tjtjfvr[t std|d|tjkr!|stdt||Std|Std|)aGet a collation function for substring matching. Args: collation: The collation strategy to use case_sensitive: Whether comparison should be case-sensitive locale: Locale string (e.g., "de_DE", "en_US") for LOCALE collation Returns: A function that takes (needle, haystack) and returns True if needle is found in haystack according to the collation rules. Raises: CollationError: If PyICU is required but not available, or if invalid parameters are provided. Examples: >>> match_fn = get_collation_function(Collation.SIMPLE, case_sensitive=False) >>> match_fn("test", "This is a TEST") True Collation '[' requires PyICU to be installed. Install with: pip install 'icalendar-searcher[collation]',LOCALE collation requires a locale parameterNUnknown collation: ) r r_binary_contains_case_insensitive_containsrr HAS_PYICUr_get_icu_containsrrr s rget_collation_functionr)Es2I$$$  .# #- - y()*:; ; ;  LiLLL   ( ( ( U$%STTT$V^<< <%T>:: :>9>>???rCallable[[str], bytes]c@|tjkr|rdSdS|tjtjfvr[tst d|d|tjkr!|st dt ||St d|St d|)aGet a collation function for generating sort keys. Args: collation: The collation strategy to use case_sensitive: Whether comparison should be case-sensitive locale: Locale string (e.g., "de_DE", "en_US") for LOCALE collation Returns: A function that takes a string and returns a sort key (bytes) that can be used for sorting according to the collation rules. Raises: CollationError: If PyICU is required but not available, or if invalid parameters are provided. Examples: >>> sort_key_fn = get_sort_key_function(Collation.SIMPLE, case_sensitive=False) >>> sorted(["Zebra", "apple", "Banana"], key=sort_key_fn) ['apple', 'Banana', 'Zebra'] c,|dSNzutf-8)encodess rz'get_sort_key_function..sQXXg..rcP|dSr-)lowerr.r/s rr1z'get_sort_key_function..sQWWYY--g66rr r!r"Nr#)r rrrr&r_get_icu_sort_keyr(s rget_sort_key_functionr5ws2I$$$  7.. .66 6 y()*:; ; ;  LiLLL   ( ( ( U$%STTT$V^<< <%T>:: :>9>>???rneedlestrhaystackc ||vS)z(Binary (case-sensitive) substring match.rr6r8s rr$r$s X rcR||vS)z!Case-insensitive substring match.r3r:s rr%r%s <<>>X^^-- --rcdfd }|S)aGGet ICU-based substring matcher. Note: This is a simplified implementation. PyICU doesn't expose ICU's StringSearch API which would be needed for proper substring matching with collation. For now, we use Python's built-in matching. Future enhancement: Implement proper collation-aware substring matching. r6r7r8rrc`r||vS||vS)zCheck if needle is in haystack. This is a fallback implementation until proper ICU StringSearch support is added. It provides reasonable behavior for most use cases. r<)r6r8rs r icu_containsz'_get_icu_contains..icu_containss5  6X% %<<>>X^^%5%55 5rr6r7r8r7rrr)r rr?s ` rr'r's) 6 6 6 6 6 6 rc|rt|ntj}tj||r tjntjdfd }|S)zGet ICU-based sort key function. Creates a collator instance and returns a function that generates sort keys. The collator strength is configured based on case_sensitive parameter. r0r7rbytesc.|S)z Generate ICU collation sort key.) getSortKey)r0collators r icu_sort_keyz'_get_icu_sort_key..icu_sort_keys""1%%%r)r0r7rrB) ICULocalegetRoot ICUCollatorcreateInstance setStrengthTERTIARY SECONDARY)r r icu_localerFrEs @rr4r4s '-E6""")2C2E2EJ)*55H 4[12222[2333&&&&&& r)rr rrr rrr)rr rrr rrr*r@)r rrrrr)r rrrrr*)r __future__rcollections.abcrenumricurrIrrGr& ImportErrorr7r Exceptionrrr)r5r$r%r'r4rrrrUs#"""""$$$$$$++++++''''''IIIII%<%<%<%<%<T%<%<%