[gKSSKJr SSKrSSKJrJrJrJr SSKJ r SSK J r \R"\ 5r"SS5rg)) defaultdictN)CallableDictListSet)BaseLunrException)Tokenc\rSrSr%Sr0r\\\4\ S'Sr Sr Sr \ SSj5r\ S 5rS rS rS rS rSrS\S\\4SjrSSjrSSjrSrSrSrg)Pipeline zlunr.Pipelines maintain a list of functions to be applied to all tokens in documents entering the search index and queries ran agains the index. registered_functionsc:/Ul[[5UlgN)_stackrset_skipselfs //usr/lib/python3/dist-packages/lunr/pipeline.py__init__Pipeline.__init__s&( /:3/? c,[UR5$r)lenrrs r__len__Pipeline.__len__s4;;rcdSRSRSUR555$)Nz,c38# UHoRv M g7fr)label).0fns r $Pipeline.__repr__..s6V+Bxx+s)formatjoinrrs r__repr__Pipeline.__repr__s&&--chh6V$++6V.VWWrNcU=(d URnX R;a[RSU5 X!lXRUR'g)z&Register a function with the pipeline.z+Overwriting existing registered function %sN)__name__r logwarningr )clsr"r s rregister_functionPipeline.register_functionsD$ ,, , KKEu M-/  *rcU"5nUH$nURUnURU5 M& U$![a [SR U55ef=f)z'Loads a previously serialised pipeline.z$Cannot load unregistered function {})r addKeyErrorrr%)r- serialisedpipelinefn_namer"s rload Pipeline.load)sk5!G !--g6  R " ':AA'J s 4%AcnUH/nURU5 URRU5 M1 g)zAdds new functions to the end of the pipeline. Functions must accept three arguments: - Token: A lunr.Token object which will be updated - i: The index of the token in the set - tokens: A list of tokens representing the set N)warn_if_function_not_registeredrappend)rargsr"s rr1 Pipeline.add9s.B  0 0 4 KK  r "rc URUR;$![a2 [R SR [ USU555 gf=f)NzbFunction "{}" is not registered with pipeline. This may cause problems when serialising the index.r )r r AttributeErrorr+r,r%getattrrr"s rr9(Pipeline.warn_if_function_not_registeredEsT 88t888 8  KKFFLfB,G  s9AAcURU5 URRU5nURRUS-U5 g![an[ S5UeSnAff=f)zLAdds a single function after a function that already exists in the pipeline.Cannot find existing_fnNr9rindexinsert ValueErrorrr existing_fnnew_fnrFes rafterPipeline.afterPsd ,,V4 FKK%%k2E KK  uqy& 1 F#$=>A E Fs:A A) A$$A)cURU5 URRU5nURRX25 g![an[ S5UeSnAff=f)zOAdds a single function before a function that already exists in the pipeline. rDNrErIs rbeforePipeline.beforeZs^ ,,V4 FKK%%k2E KK  u - F#$=>A E Fs6A A% A  A%c\URRU5 g![a gf=f)z%Removes a function from the pipeline.N)rremoverHr@s rrSPipeline.removefs+  KK  r "   s  ++r" field_namesc@URURU5 g)z Make the pipeline skip the function based on field name we're processing. This relies on passing the field name to Pipeline.run(). N)rupdate)rr"rUs rskip Pipeline.skipms 2k*rc4URHnU(aX RU;aM/n[U5HVupVU"XeU5nU(dM[U[[ 45(aUR U5 MEURU5 MX UnM U$)a Runs the current list of functions that make up the pipeline against the passed tokens. :param tokens: The tokens to process. :param field_name: The name of the field these tokens belongs to, can be ommited. Used to skip some functions based on field names. )rr enumerate isinstancelisttupleextendr:)rtokens field_namer"resultsitokenresults rrun Pipeline.runus++BjJJrN:G%f-Ef-ftUm44NN6*NN6*.F!$ rcz[X5nURU/5Vs/sHn[U5PM sn$s snf)a-Convenience method for passing a string through a pipeline and getting strings out. This method takes care of wrapping the passed string in a token and mapping the resulting tokens back to strings. .. note:: This ignores the skipped functions since we can't access field names from this context. )r rfstr)rstringmetadatardtkns r run_stringPipeline.run_strings7f'$(HHeW$56$5SC$5666s8c/Ulgr)rrs rresetPipeline.resets  rcXURVs/sHoRPM sn$s snfr)rr r@s r serializePipeline.serializes #';;/;R;///s')rrr)r* __module__ __qualname____firstlineno____doc__r rrir__annotations__rrr' classmethodr.r6r1r9rMrPrSrrXrfrmrprs__static_attributes__rrr r s 13$sH}-2@ X 00   # F F+x+d3i+: 70rr ) collectionsrloggingtypingrrrrlunr.exceptionsr lunr.tokenr getLoggerr*r+r r|rrrs3#,,- !V0V0r