AHj> dZddlmZddlmZddlmZddlmZddl m Z m Z ddl Z ddlmZdd lmZdd lmZGd d Ze j*j,Gd dZedk(reZej5ddyy)zm Fuzzy matcher. This class is used by the [command palette](/guide/command_palette) to match search terms. ) annotations) lru_cache) itemgetter)finditer)IterableSequenceN)LRUCache)Content)StyleczeZdZdZ d dd d dZddZeedddZdd Z dd Z y ) FuzzySearchzaPerforms a fuzzy search. Unlike a regex solution, this will finds all possible matches. i) cache_sizec2||_t||_y)zInitialize fuzzy search. Args: case_sensitive: Is the match case sensitive? cache_size: Number of queries to cache. N)case_sensitiver cache)selfrrs E/root/tools/cai/cai_env/lib/python3.12/site-packages/textual/fuzzy.py__init__zFuzzySearch.__init__s-MU N  c||f}||jvr|j|Sdgf}t|j||td|}||j|<|S)zMatch against a query. Args: query: The fuzzy query. candidate: A candidate to check,. Returns: A pair of (score, tuple of offsets). `(0, ())` for no result. r)keydefault)rmax_matchr)rquery candidate cache_keyrresults rmatchzFuzzySearch.match+sbI&  "::i( (03RyT[[ 2 1 wW & 9 ri)maxsizecnttd|Dchc]}|jc}Scc}w)Nz\w+) frozensetrstart)clsrr s rget_first_letterszFuzzySearch.get_first_letters>s*Xfi5PQE%++-QRRQs2c|j|}t|}|t|j|z}d}|^}}|D]} | |dzk7r|dz }| }||dz z |z } |d| | zzz}|S)zoScore a search. Args: search: Search object. Returns: Score. )r&len intersection) rr positions first_letters offset_countscoregroups last_offsetoffsetsoffsetnormalized_groupss rr.zFuzzySearch.scoreCs..y9 9~ #c-*D*DY*O&PP ) g !Fq(!  K ! *VaZ8LH '*;;<< rc # Kgd}|js |j}|j}|j}||vrJ|j|}t t ||t |z}|||||k(rdndz|fyt|D]\}}t ||z } g} j| |} |j|| x} dk7r4| j| | dz} | | k\rn|j|| x} dk7r4| sdy| ddz}gt |d fd  gdD]}||||fyw) Nrg@g?r()rc|D]?}|r ||dkDsg||}t|k(rj|4||dzAy)zRecursively match offsets. Args: offsets: A list of offsets. positions_index: Index of query letter. r5r(N)r)append)r1positions_indexr2 new_offsets get_offsetsletter_positionspossible_offsets query_lengths rr;z'FuzzySearch._match..get_offsetssb+?; F&72;"6"4G"4V"4K;'<7(// <#K11DE  Fr)r1z list[int]r9intreturnNone) rlowerr.rfindlistranger) enumerater8find)rrrpositionr.query_locationr1r2letter last_indexr+indexlocationr;r<r=r>s @@@@rrzFuzzySearch._match^s-/""!)IKKME  I &__U3N5#e*1LMNGi)I4FSCP  '. (NFFY&0J#%I  # #I .E(~~fe<<8C  * 1 J& )~~fe<<8C  |a'H (-/5z  F F B' 5G 7+W4 4 5s D EAEN)F)rboolrr?r@rA)rstrrrOr@ztuple[float, Sequence[int]])rrOr@zfrozenset[int])rrOr+z Sequence[int]r@float)rrOrrOr@z%Iterable[tuple[float, Sequence[int]]]) __name__ __module__ __qualname____doc__rr classmethodrr&r.rr6rrr r s{ &+  AI  "  ;>    &tSS69595%(95 .95rr cteZdZdZddd d dZed dZed dZeddZdd Z dd Z y)MatcherzA fuzzy matcher.NF) match_stylerch||_| tdn||_||_t |_y)aInitialise the fuzzy matching object. Args: query: A query as typed in by the user. match_style: The style to use to highlight matched portions of a string. case_sensitive: Should matching be case sensitive? NT)reverse)_queryr _match_style_case_sensitiver fuzzy_search)rrrXrs rrzMatcher.__init__s1 3>3FE$/K-'Mrc|jS)zThe query string to look for.)r[rs rrz Matcher.querys{{rc|jS)zBThe style that will be used to highlight hits in the matched text.)r\r`s rrXzMatcher.match_styles   rc|jS)zIs this matcher case sensitive?)r]r`s rrzMatcher.case_sensitives###rcT|jj|j|dS)zMatch the candidate against the query. Args: candidate: Candidate string to match against the query. Returns: Strength of the match from 0 to 1. r)r^r r)rrs rr z Matcher.matchs&  &&tzz9=a@@rctj|}|jj|j|\}}|s|S|D]6}||j r|j |j||dz}8|S)zHighlight the candidate with the fuzzy match. Args: candidate: The candidate string to match against the query. Returns: A [`Text`][rich.text.Text] object with highlighted matches. r()r from_markupr^r risspacestylizer\)rrcontentr.r1r2s r highlightzMatcher.highlights%%i0**00YGwN QFV$,,.!//$*;*;VVaZP Qr)rrOrXz Style | NonerrNr@rA)r@rO)r@r )r@rN)rrOr@rP)rrOr@r ) rQrRrSrTrpropertyrrXrr rir6rrrWrWs %)$ **" *  *  *&!!$$ ArrW__main__zfoo.barz foo/egg.bar)rT __future__r functoolsroperatorrrertypingrr rich.reprrich textual.cacher textual.contentr textual.visualr r reprautorWrQr^r r6rrrxs#%"# A5A5H@@@F z=Ly-0r