o .j9@sdZddlmZddlmZddlmZzddlmZ ddlm Z dZ Wn e y/dZ YnwGd d d eeZGd d d eZejdd fd%ddZejdd fd&ddZd'ddZd'dd Zd(d!d"Zd)d#d$Zd 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@s 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__SIMPLEUNICODELOCALErrQ/home/thesage/.local/lib/python3.10/site-packages/icalendar_searcher/collation.pyrsrc@seZdZdZdS)CollationErrorz4Raised when collation operation cannot be performed.N)r r r rrrrrr?srN collationcase_sensitiveboolr str | NonereturnCallable[[str, str], bool]cCsr|tjkr |r tStS|tjtjfvr2tstd|d|tjkr-|s(tdt||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: ) rr_binary_contains_case_insensitive_containsrr HAS_PYICUr_get_icu_containsrrr rrrget_collation_functionEs     r$Callable[[str], bytes]cCsz|tjkr|r ddSddS|tjtjfvr6ts!td|d|tjkr1|s,tdt||Std|Std|) 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'] cSs |dSNzutf-8)encodesrrrs z'get_sort_key_function..cSs|dSr&)lowerr'r(rrrr*srrrNr)rrrrr!r_get_icu_sort_keyr#rrrget_sort_key_functionws     r-needlestrhaystackcCs||vS)z(Binary (case-sensitive) substring match.rr.r0rrrrsrcCs||vS)z!Case-insensitive substring match.r+r1rrrr sr csd fdd }|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. r.r/r0rrcsr||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. r2r1rrr icu_containssz'_get_icu_contains..icu_containsNr.r/r0r/rrr)r rr4rr3rr"s  r"csN|rt|nt}t||rtjntjdfdd }|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. r)r/rbytescs |S)z Generate ICU collation sort key.) getSortKeyr(collatorrr icu_sort_keys z'_get_icu_sort_key..icu_sort_keyN)r)r/rr6) ICULocalegetRoot ICUCollatorcreateInstance setStrengthTERTIARY SECONDARY)r r icu_localer:rr8rr,s  r,)rrrrr rrr)rrrrr rrr%r5)r rrrrr)r rrrrr%)r __future__rcollections.abcrenumricurr=rr;r! ImportErrorr/r Exceptionrrr$r-rr r"r,rrrrs2      ( 3 7