U {hIZ@sddlZddlZddlmZddlmZmZmZmZddl m Z ddl m Z ddl mZmZmZddlmZmZddlmZmZmZdd lmZmZdd lmZdd lmZdd lm Z Gd ddeZ!GdddZ"Gddde e"e!Z#GdddeZ$GdddeZ%dS)N) defaultdict)AnyDictIterableTuple)apps) ContentType) CharFieldPrefetchQ)ExistsOuterRef)CastLengthSubstr) BaseIterable ModelIterable)MP_NodeQuerySet)Site)SearchableQuerySetMixincseZdZdZfddZde_d$ddZd%dd Zd&d d Zd d Z ddZ ddZ d'ddZ d(ddZ d)ddZddZddZddZd*ddZd+d d!Zd,d"d#ZZS)- TreeQuerySetz] Extends Treebeard's MP_NodeQuerySet with additional useful tree-related operations. cstdS)zNRedefine the delete method unbound, so we can set the queryset_only parameter.N)superdeleteself __class__U/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/wagtail/query.pyrszTreeQuerySet.deleteTFcCs2t|jdt|jd@}|s.|t|jdM}|S)NZpath__startswith)Z depth__gtepk)r pathdepthr!rother inclusiveqrrrdescendant_of_qszTreeQuerySet.descendant_of_qcCs||||S)z This filters the QuerySet to only contain pages that descend from the specified page. If inclusive is set to True, it will also contain the page itself (instead of just its descendants). )filterr(rr%r&rrr descendant_of%szTreeQuerySet.descendant_ofcCs||||S)z This filters the QuerySet to not contain any pages that descend from the specified page. If inclusive is set to True, it will also exclude the specified page. )excluder(r*rrrnot_descendant_of-szTreeQuerySet.not_descendant_ofcCs||t|jdd@S)Nr#)r(r r#rr%rrr child_of_q5szTreeQuerySet.child_of_qcCs|||S)zq This filters the QuerySet to only contain pages that are direct children of the specified page. )r)r1r0rrrchild_of8szTreeQuerySet.child_ofcCs|||S)zt This filters the QuerySet to not contain any pages that are direct children of the specified page. )r,r1r0rrr not_child_of>szTreeQuerySet.not_child_ofcsRfddtdtjdjddD}t|d}|sN|tjdM}|S)Ncsg|]}jd|qS)rr").0posr%rr Esz.TreeQuerySet.ancestor_of_q..rr.)Zpath__inr )rangelenr"steplenr r!)rr%r&pathsr'rr7r ancestor_of_qDs  zTreeQuerySet.ancestor_of_qcCs||||S)z This filters the QuerySet to only contain pages that are ancestors of the specified page. If inclusive is set to True, it will also include the specified page. )r)r=r*rrr ancestor_ofPszTreeQuerySet.ancestor_ofcCs||||S)z This filters the QuerySet to not contain any pages that are ancestors of the specified page. If inclusive is set to True, it will also exclude the specified page. )r,r=r*rrrnot_ancestor_ofXszTreeQuerySet.not_ancestor_ofcCst|j|jdS)Nr4)r model_get_parent_path_from_pathr"r0rrr parent_of_q`szTreeQuerySet.parent_of_qcCs|||S)z] This filters the QuerySet to only contain the parent of the specified page. )r)rBr0rrr parent_ofcszTreeQuerySet.parent_ofcCs|||S)zX This filters the QuerySet to exclude the parent of the specified page. )r,rBr0rrr not_parent_ofiszTreeQuerySet.not_parent_ofcCs:t|j|jdt|jd@}|s6|t|jdM}|S)Nrr/r )r r@rAr"r#r!r$rrr sibling_of_qos zTreeQuerySet.sibling_of_qcCs||||S)a! This filters the QuerySet to only contain pages that are siblings of the specified page. By default, inclusive is set to True so it will include the specified page in the results. If inclusive is set to False, the page will be excluded from the results. )r)rEr*rrr sibling_ofyszTreeQuerySet.sibling_ofcCs||||S)a$ This filters the QuerySet to not contain any pages that are siblings of the specified page. By default, inclusive is set to True so it will exclude the specified page from the results. If inclusive is set to False, the page will be included in the results. )r,rEr*rrrnot_sibling_ofszTreeQuerySet.not_sibling_of)F)F)F)F)F)F)T)T)T)__name__ __module__ __qualname____doc__rZ queryset_onlyr(r+r-r1r2r3r=r>r?rBrCrDrErFrG __classcell__rrrrrs$       rcs>eZdZfddZfddZd ddZedd ZZS) SpecificQuerySetMixincstj||d|_dS)zSet custom instance attributesFN)r__init___defer_streamfields)rargskwargsrrrrNszSpecificQuerySetMixin.__init__cst}|j|_|S)z.Ensure clones inherit custom attribute values.)r_clonerO)rclonerrrrRs zSpecificQuerySetMixin._cloneFcCs|}|rt|_nt|_|S)a This efficiently gets all the specific items for the queryset, using the minimum number of queries. When the "defer" keyword argument is set to True, only generic field values will be loaded and all specific fields will be deferred. )rRDeferredSpecificIterable_iterable_classSpecificIterable)rdeferrSrrrspecifics zSpecificQuerySetMixin.specificcCst|jttfS)zU Returns True if this queryset is already specific, False otherwise. ) issubclassrUrVrTrrrr is_specificsz!SpecificQuerySetMixin.is_specific)F) rHrIrJrNrRrXpropertyrZrLrrrrrMs    rMc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd=d(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd>d2d3Zd?d4d5Zd6d7Zd8d9Zd:d;Zds z&PageQuerySet.type_q..Zcontent_type__in)rZ get_modelsrobjectsget_for_modelsr listvalues)rriZall_subclasses content_typesrrhrtype_qs   zPageQuerySet.type_qcGs||j|S)z This filters the QuerySet to only contain pages that are an instance of the specified model(s) (including subclasses). )r)rqrrirrrtypeszPageQuerySet.typecGs||j|S)zq This filters the QuerySet to exclude any pages which are an instance of the specified model(s). )r,rqrrrrrnot_typeszPageQuerySet.not_typecGstjj|}tt|dS)Nrk)rrlrmr rnro)rrirprrr exact_type_qs zPageQuerySet.exact_type_qcGs||j|S)z This filters the QuerySet to only contain pages that are an instance of the specified model(s) (matching the model exactly, not subclasses). )r)rurrrrr exact_typeszPageQuerySet.exact_typecGs||j|S)z This filters the QuerySet to exclude any pages which are an instance of the specified model(s) (matching the model exactly, not subclasses). )r,rurrrrrnot_exact_typeszPageQuerySet.not_exact_typecCsNddlm}t}|jdD]}||j|jddO}q"|rD|StgdS)Nr)PageViewRestrictionrfTr&Zpk__in)Zwagtail.modelsrxr rlselect_relatedallr(rf)rrxr'Z restrictionrrr private_q s  zPageQuerySet.private_qcCs||S)z} Filters the QuerySet to only contain pages that are not in a private section and their descendants. )r,r}rrrrpublicszPageQuerySet.publiccCs||Szy Filters the QuerySet to only contain pages that are in a private section and their descendants. r)r}rrrr not_publicszPageQuerySet.not_publiccCs||Srrrrrrprivate!szPageQuerySet.privateFcCs|s"|r|jd|jS|r:|jddd}n<|jtddtd|jj t dddd jd dd }t |}t||jj }|d kr|d | }|d kr|r|jd|jS|jjj|dS)a7 Find the first ancestor that all pages in this queryset have in common. For example, consider a page hierarchy like:: - Home/ - Foo Event Index/ - Foo Event Page 1/ - Foo Event Page 2/ - Bar Event Index/ - Bar Event Page 1/ - Bar Event Page 2/ The common ancestors for some queries would be: .. code-block:: python >>> Page.objects\ ... .type(EventPage)\ ... .first_common_ancestor() >>> Page.objects\ ... .type(EventPage)\ ... .filter(title__contains='Foo')\ ... .first_common_ancestor() This method tries to be efficient, but if you have millions of pages scattered across your page tree, it will be slow. If `include_self` is True, the ancestor can be one of the pages in the queryset: .. code-block:: python >>> Page.objects\ ... .filter(title__contains='Foo')\ ... .first_common_ancestor() >>> Page.objects\ ... .filter(title__exact='Bar Event Index')\ ... .first_common_ancestor() A few invalid cases exist: when the queryset is empty, when the root Page is in the queryset and ``include_self`` is False, and when there are multiple page trees with no common root (a case Wagtail does not support). If ``strict`` is False (the default), then the first root node is returned in these cases. If ``strict`` is True, then a ``ObjectDoesNotExist`` is raised. z'Can not find ancestor of empty querysetr"T)Zflatr.) max_lengthZ output_field) parent_pathrrNzNo common ancestor found!r4)existsr@Z DoesNotExistZget_first_root_nodeZorder_byZ values_listannotaterrr;r Zdistinct posixpath commonprefixr:rlget)rZ include_selfstrictr<Zcommon_parent_pathZ extra_charsrrrfirst_common_ancestor(s64      z"PageQuerySet.first_common_ancestorcCs|D] }|qdS)zB This unpublishes all live pages in the QuerySet. N)r] unpublish)rrfrrrrs zPageQuerySet.unpublishcCs*|}d|_|j}|s |S|j|S)a7 Apply to a queryset to prevent fetching/decoding of StreamField values on evaluation. Useful when working with potentially large numbers of results, where StreamField values are unlikely to be needed. For example, when generating a sitemap or a long list of page links. T)rRrOr@Zget_streamfield_namesrW)rrSZstreamfield_namesrrrdefer_streamfieldss  zPageQuerySet.defer_streamfieldscCs|j|jddS)z\ This filters the QuerySet to only contain pages within the specified site. Try)r+Z root_page)rsiterrrin_siteszPageQuerySet.in_sitecCs&t|jd}|s"|t|jdM}|S)N)translation_keyr )r rr!)rrfr&r'rrrtranslation_of_qs zPageQuerySet.translation_of_qcCs||||S)z This filters the QuerySet to only contain pages that are translations of the specified page. If inclusive is True, the page itself is returned. )r)rrrfr&rrrtranslation_ofszPageQuerySet.translation_ofcCs||||S)a` This filters the QuerySet to only contain pages that are not translations of the specified page. Note, this will include the page itself as the page is technically not a translation of itself. If inclusive is True, we consider the page to be a translation of itself so this excludes the page from the results. )r,rrrrrnot_translation_ofszPageQuerySet.not_translation_ofcCs>ddlm}|jd}d}|jr*d}|t||ddS)a Performance optimisation for listing pages. Prefetches the active workflow states on each page in this queryset. Used by `workflow_in_progress` and `current_workflow_progress` properties on `wagtailcore.models.Page`. r.) WorkflowStateZcurrent_task_state__taskZ_workflow_statesZ_specific_workflow_statesZ_current_workflow_states)querysetZto_attr)modelsrrlZactiver{rZZprefetch_relatedr )rrZworkflow_statesZrelationrrrprefetch_workflow_statess  z%PageQuerySet.prefetch_workflow_statescCs<ddlm}|jt|jjddjttdt dddS) z Performance optimisation for listing pages. Annotates each page with the existence of an approved go live time. Used by `approved_schedule` property on `wagtailcore.models.Page`. r.)RevisionT)Zapproved_go_live_at__isnullr!r)Z object_id)Z_approved_schedule) rrrr Zpage_revisionsr,r)rr r )rrrrrannotate_approved_schedules z'PageQuerySet.annotate_approved_schedulecCs|jttjjtdddS)z Performance optimisation for listing pages. Annotates each object with whether it is a root page of any site. Used by `is_site_root` method on `wagtailcore.models.Page`. r)Zroot_page__translation_key)Z _is_site_root)rr rrlr)r rrrrannotate_site_root_statesz%PageQuerySet.annotate_site_root_stateN)FF)F)F) rHrIrJr_r]r`rarbrcrerfrgrqrsrtrurvrwr}r~rrrrrrrrrrrrrrrrr\s:  j  r\c@s2eZdZddZeeeeefdddZ dS)rVc #s|j}|jj|jd}||D]}tt}r`|D]"}fdd|D||d<qsz-SpecificIterable.__iter__..cSsg|]}|d|dgqS)r!rr)r5rrrrr8sz-SpecificIterable.__iter__..cSsi|]\}}|tj|qSr)rrlZ get_for_id)r5_r!rrrrsrzrcSsi|] }|j|qSrr )r5itemrrrr*sc3s|]}|kr|VqdSNr)r5r!)items_for_typerr ,sz,SpecificIterable.__iter__..zSpecific versions of the following items could not be found. This is most likely because a database migration has removed the relevant table or record since the item was created: {}cSsg|]}|j|j|jdqS))rdtitlers)rdrr)r5prrrr89scategoryN)r!r)rqueryZannotation_selectro _get_chunksrrnitemsappendZ model_classr@rlr)rOhasattrrextendr{Zin_bulkwarningswarnformatRuntimeWarningKeyErrorrr!setattr)rqsZ values_qsroZannotations_by_pkdataZ pks_and_typesZ pks_by_typer!rrpZ items_by_typeZ missing_pksZpksr@rZ generic_itemsr annotationvaluer)rrr__iter__sd    zSpecificIterable.__iter__)returnccsd|jst|VnNg}||jD].}||t||jkr"t|V|q"|r`t|VdSr)Z chunked_fetchtupleiterator chunk_sizerr:clear)rrZ current_chunkrrrrrOs    zSpecificIterable._get_chunksN) rHrIrJrrrrstrrrrrrrrVsOrVcseZdZfddZZS)rTc #s\tD]L}|jr|jVq tjd|jjd|jd|j d|j d t d|Vq dS)NzA specific version of the following object could not be returned because the specific model is not present on the active branch: r) rrZspecific_classZspecific_deferredrrrrHrdrrr)robjrrrrcs &z!DeferredSpecificIterable.__iter__)rHrIrJrrLrrrrrTbsrT)&rr collectionsrtypingrrrrZ django.appsrZ"django.contrib.contenttypes.modelsrZdjango.db.modelsr r r Zdjango.db.models.expressionsr r Zdjango.db.models.functionsrrrZdjango.db.models.queryrrZtreebeard.mp_treerZwagtail.models.sitesrZwagtail.search.querysetrrrMr\rVrTrrrrs&      |'Lc