U Lk7g0@s\ddlmZddlmZmZmZmZmZmZdZ dZ dZ GdddeZ Gdd d eZ d S) )OperationError)CASCADEDENY DO_NOTHINGNULLIFYPULL BaseQuerySet)QuerySetQuerySetNoCacherrrrrdcs^eZdZdZdZdZdZddZddZdd Z d d Z d d Z dfdd Z ddZ ZS)r zThe default queryset, that builds queries and handles a set of results returned from a query. Wraps a MongoDB cursor, providing :class:`~mongoengine.Document` objects as the results. TNcCsd|_|jr|St|jS)zIteration utilises a results cache which iterates the cursor in batches of ``ITER_CHUNK_SIZE``. If ``self._has_more`` the cursor hasn't been exhausted so cache then batch. Otherwise iterate the result_cache. T)_iter _has_more _iter_resultsiter _result_cacheselfrA/tmp/pip-unpacked-wheel-n1etwkgt/mongoengine/queryset/queryset.py__iter__&szQuerySet.__iter__cCs4|jdk r|jS|jr"t|t|j|_|jS)zSince __len__ is called quite frequently (for example, as part of list(qs)), we populate the result cache and cache the length. N)_lenrlistrlenrrrrr__len__5s    zQuerySet.__len__cCs@|jr dS||jdtd}t|tkr8d|d<t|S)z/Provide a string representation of the QuerySet.. queryset mid-iteration ..N$...(remaining elements truncated)...)r _populate_cacherREPR_OUTPUT_SIZErrepr)rdatarrr__repr__Ds zQuerySet.__repr__ccs^|jdkrg|_d}|t|jkr8|j|V|d7}q|jsBdSt|j|kr|qdS)zA generator for iterating over the result cache. Also populates the cache if there are more possible results to yield. Raises StopIteration when there are no more results. Nrr)rrrr)rposrrrrOs    zQuerySet._iter_resultscCs\|jdkrg|_|jsdSz"ttD]}|jt|q$Wntk rVd|_YnXdS)z{ Populates the result cache with ``ITER_CHUNK_SIZE`` more entries (until the cursor is exhausted). NF)rrrangeITER_CHUNK_SIZEappendnext StopIteration)r_rrrros  zQuerySet._populate_cacheFcs2|dkrt|S|jdkr,t||_|jS)zCount the selected elements in the query. :param with_limit_and_skip (optional): take any :meth:`limit` or :meth:`skip` that has been applied to this cursor into account when getting the count FN)supercountr)rZwith_limit_and_skip __class__rrr,s   zQuerySet.countcCs&|jdk rtd|t|j|jS)z!Convert to a non-caching querysetNzQuerySet already cached)rr _clone_intor _document _collectionrrrrno_caches zQuerySet.no_cache)F)__name__ __module__ __qualname____doc__rrrrrr#rrr,r2 __classcell__rrr-rr s  r c@s(eZdZdZddZddZddZdS) r zA non caching QuerySetcCs|t|j|jS)zConvert to a caching queryset)r/r r0r1rrrrcacheszQuerySetNoCache.cachec Csr|jr dSg}ttdD]2}z|t|Wqtk rJYqNYqXqt|tkrbd|d<|t|S)z2Provides the string representation of the QuerySetrrrr) r r%r r'r(r)rrewindr!)rr"r*rrrr#s zQuerySetNoCache.__repr__cCs|}|jr|}||S)N)r cloner9)rZquerysetrrrrs zQuerySetNoCache.__iter__N)r3r4r5r6r8r#rrrrrr sr N)Zmongoengine.errorsrZmongoengine.queryset.baserrrrrr__all__r r&r r rrrrs