U Lk7g®ã@s6ddlmZddlmZdZGdd„dƒZdd„ZdS) é)Úpartial)ÚQuerySet)Úqueryset_managerÚQuerySetManagerc@s*eZdZdZdZeZddd„Zdd„ZdS)raç The default QuerySet Manager. Custom QuerySet Manager functions can extend this class and users can add extra queryset functionality. Any custom manager methods must accept a :class:`~mongoengine.Document` class as its first argument, and a :class:`~mongoengine.queryset.QuerySet` as its second argument. The method function should return a :class:`~mongoengine.queryset.QuerySet` , probably the same one that was passed in, but modified in some way. NcCs|r ||_dS)N)Ú get_queryset)ÚselfZ queryset_func©rú@/tmp/pip-unpacked-wheel-n1etwkgt/mongoengine/queryset/manager.pyÚ__init__szQuerySetManager.__init__cCsv|dk r |S|j d|j¡}||| ¡ƒ}|jrr|jjj}|dkrN| |¡}n$|dkrd| ||¡}nt|j||ƒ}|S)zfDescriptor for instantiating a new QuerySet object when Document.objects is accessed. NÚqueryset_classéé)Z_metaÚgetÚdefaultZ_get_collectionrÚ__code__Ú co_argcountr)rÚinstanceÚownerr ZquerysetÚ arg_countrrr Ú__get__s  zQuerySetManager.__get__)N) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrr rrrrr rs   rcCst|ƒS)aÁDecorator that allows you to define custom QuerySet managers on :class:`~mongoengine.Document` classes. The manager must be a function that accepts a :class:`~mongoengine.Document` class as its first argument, and a :class:`~mongoengine.queryset.QuerySet` as its second argument. The method function should return a :class:`~mongoengine.queryset.QuerySet`, probably the same one that was passed in, but modified in some way. )r)Úfuncrrr r2srN)Ú functoolsrZmongoengine.queryset.querysetrÚ__all__rrrrrr Ús  *