a D g@sddlZddlZddlmZddlmZddlmZddlm Z ddl m Z m Z ddl mZmZmZeZefd d ZeeejZeeejZeeejZeeejZd Zed Zd dZddZ defddZ!ddZ"ddZ#ddZ$ddZ%ddZ&ddZ'dS)N)partial)apps) connections) QueryDict) RelatedFields SearchField) MATCH_NONEPhrase PlainTextcCst|tst|}|tur:t|dkr*|S||t||St|dkrPtdnNt|dkrd|dSt|d}t||d|}t|||d}|||SdS)aD Has the same result as Python's reduce function, but performs the calculations in a different order. This is important when the operator is constructing data structures such as search query classes. This method will make the resulting data structures flatter, so operations that need to traverse them don't end up crashing with recursion errors. For example: Python's builtin reduce() function will do the following calculation: reduce(add, [1, 2, 3, 4, 5, 6, 7, 8]) (1 + (2 + (3 + (4 + (5 + (6 + (7 + 8))))))) When using this with query classes, it would create a large data structure with a depth of 7 Whereas balanced_reduce will execute this like so: balanced_reduce(add, [1, 2, 3, 4, 5, 6, 7, 8]) ((1 + 2) + (3 + 4)) + ((5 + 6) + (7 + 8)) Which only has a depth of 2 rz0reduce() of empty sequence with no initial valuerN) isinstancelistNOT_SETlenbalanced_reduce TypeError)operatorseqZ initializerZ break_pointZ first_setZ second_setrS/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/site-packages/wagtail/search/utils.pyrs      rz \b(\w+):(\w+|"[^"]+"|\'[^\']+\')cCs*|dt}|}tdd|}|S)Nz + )MAX_QUERY_STRING_LENGTHlowerresubstrip) query_stringrrrnormalise_query_stringKs rcCsRtdd}t|D]$}|\}}|||diqtd|}||fS)NT)Zmutablez"')rfilters_regexpfinditergroupsupdaterr)rfiltersZ match_objectkeyvaluerrrseparate_filters_from_queryWs   r(c Cst|\}}d}g}d|vr(|d}n |d}|D]B}|}|rr|rZ|t|n|t||pjtjd| }q6|r|dkrt|}qt|}n|}||fS)aT This takes a query string typed in by a user and extracts the following: - Quoted terms (for phrase search) - Filters For example, the following query: `hello "this is a phrase" live:true` would be parsed into: filters: {'live': 'true'} tokens: And([PlainText('hello'), Phrase('this is a phrase')]) F"')ror) r(splitrappendr r ZDEFAULT_OPERATORORAND) rrZ zero_termsr%Z is_phrasetokenspartspartZ search_queryrrrparse_query_stringbs*     r3cs$fddtD}||S)zI Returns all descendants of a model, including the model itself. csh|]}t|r|qSr) issubclass).0Z other_modelmodelrr s z(get_descendant_models..)rZ get_modelsadd)r7Zdescendant_modelsrr6rget_descendant_modelss   r:cCsddlm}|j|jS)Nr ContentType)"django.contrib.contenttypes.modelsr<objectsZ get_for_modelpkr7r<rrrget_content_type_pks rAcCs,ddlm}dd|jj|jDS)zR Returns content types ids for the ancestors of this model, excluding it. rr;cSsg|] }|jqSrr?r5ctrrr sz3get_ancestors_content_types_pks..)r=r<r>get_for_modelsZ_metaZget_parent_listvaluesr@rrrget_ancestors_content_types_pkss  rHcCs*ddlm}dd|jjt|DS)zT Returns content types ids for the descendants of this model, including it. rr;cSsg|] }|jqSrrBrCrrrrEsz5get_descendants_content_types_pks..)r=r<r>rFr:rGr@rrr!get_descendants_content_types_pkss  rIccs:|D]0}t|tr|Vqt|trt|jEdHqdS)N)r rrget_search_fieldsfields)Z search_fieldsZ search_fieldrrrrJs   rJcCsddtDS)NcSsg|]}|jdkr|qS)Z postgresql)vendor)r5 connectionrrrrEs z.get_postgresql_connections..)rallrrrrget_postgresql_connectionssrO)(rr functoolsrZ django.appsrZ django.dbrZ django.httprZwagtail.search.indexrrqueryr r r objectrror_r.and_r/r9ZADDmulZMULrcompiler!rr(r3r:rArHrIrJrOrrrrs0     .       /