U Lk7g=@sddlZddlmZddlmZddlmZmZdZddZ dd Z Gd d d e Z Gd d d e ZGdddeZGdddZGdddeZdS)N)DBRef) _import_class) DoesNotExistMultipleObjectsReturned)BaseDict StrictDictBaseListEmbeddedDocumentList LazyReferencecsfdd}|S)z;Decorator that ensures _mark_as_changed method gets called.cs|f||}||SN_mark_as_changed)selfargskwargsresult parent_methodC/tmp/pip-unpacked-wheel-n1etwkgt/mongoengine/base/datastructures.pywrappersz(mark_as_changed_wrapper..wrapperrrrrrrmark_as_changed_wrappers rcsfdd}|S)zPDecorator that ensures _mark_as_changed method gets called with the key argumentcs8|r||ks|||dkr&||||f||S)Nrr )rkeyrrrrrr s z,mark_key_as_changed_wrapper..wrapperrrrrrmark_key_as_changed_wrappers rcseZdZdZdZdZdZfddZdddZfdd Z d d Z d d Z e e jZe e jZe e jZee jZee jZee jZee jZee jZdddZZS)rz+A special dict so we can watch any changes.FNcs4td}t||rt||_||_t|dSN BaseDocument)r isinstanceweakrefproxy _instance_namesuper__init__)r dict_itemsinstancenamer __class__rrr#0s   zBaseDict.__init__cCs*z ||WStk r$|YSXdSr ) __getitem__KeyErrorrrdefaultrrrget8s z BaseDict.getcst|}td}t||r2|jdkr2|j|_nt|trvt|tsvt|d|jd|}t|||j|_nBt|t rt|t st |d|jd|}t|||j|_|S)NEmbeddedDocument.) r"r)rrr dictrr! __setitem__listrrrvaluer.r'rrr)?s   zBaseDict.__getitem__cCsd|_d|_|SNFr% _dereferencedrrrr __getstate__OszBaseDict.__getstate__cCs|}|Sr rrstaterrr __setstate__TszBaseDict.__setstate__cCs<t|jdr8|r*|j|jd|n|j|jdSNr r/)hasattrr r r!rrrrrr as zBaseDict._mark_as_changed)N)N)__name__ __module__ __qualname____doc__r7r r!r#r-r)r9r<rr0r1 __delattr__ __delitem__rpopclearupdatepopitem setdefaultr __classcell__rrr'rr)s$           rcseZdZdZdZdZdZfddZfddZfdd Z d d Z d d Z fddZ e ejZe ejZe ejZe ejZe ejZe ejZe ejZe ejZe ejZe ejZdddZZS)rz+A special list so we can watch any changes.FNcsHtd}t||r2t|tjr&||_n t||_||_t|dSr) rrr ProxyTypesr rr!r"r#)rZ list_itemsr%r&rr'rrr#ps   zBaseList.__init__cst|tr|dkrt||}t|}t|tr8|Std}t||r^|jdkr^|j|_nt|trt|t st |d|j d|}t |||j|_nBt|t rt|t st |d|j d|}t |||j|_|S)Nrr.r/)rintlenr"r)slicerr r0rr!r1r2rr3r'rrr)|s"     zBaseList.__getitem__c#stEdHdSr )r"__iter__r8r'rrrPszBaseList.__iter__cCsd|_d|_|Sr5r6r8rrrr9szBaseList.__getstate__cCs|}|Sr rr:rrrr<szBaseList.__setstate__cs.|}t|trd}t||}|||Sr )rrOr"r1r )rrr4Z changed_keyrr'rrr1s   zBaseList.__setitem__cCsHt|jdrD|dk r6|j|jd|t|n|j|jdSr=)r>r r r!rNr?rrrr s "zBaseList._mark_as_changed)N)r@rArBrCr7r r!r#r)rPr9r<r1rr2appendextendinsertrFremovereversesortrE__iadd____imul__r rKrrr'rris*            rc@sleZdZeddZeddZddZddZd d Zd d Z d dZ ddZ ddZ ddZ ddZdS)r cCs:|D],\}}t||}||krt||krdSqdS)zoReturn True if a given embedded doc matches all the filter kwargs. If it doesn't return False. FT)itemsgetattrstr)clsZ embedded_docrrZexpected_valueZdoc_valrrrZ __match_alls  z EmbeddedDocumentList.__match_allcss|Sfdd|DS)z2Return embedded docs that match the filter kwargs.csg|]}|r|qSr) _EmbeddedDocumentList__match_all).0docr\rrr s z7EmbeddedDocumentList.__only_matches..r)r\Z embedded_docsrrr`rZ__only_matchessz#EmbeddedDocumentList.__only_matchescKs|||}t||j|jS)a Filters the list by only including embedded documents with the given keyword arguments. This method only supports simple comparison (e.g. .filter(name='John Doe')) and does not support operators like __gte, __lte, __icontains like queryset.filter does :param kwargs: The keyword arguments corresponding to the fields to filter on. *Multiple arguments are treated as if they are ANDed together.* :return: A new ``EmbeddedDocumentList`` containing the matching embedded documents. Raises ``AttributeError`` if a given keyword is not a valid field for the embedded document class. #_EmbeddedDocumentList__only_matchesr r r!rrvaluesrrrfilters zEmbeddedDocumentList.filterc s.|||fdd|D}t||j|jS)a Filters the list by excluding embedded documents with the given keyword arguments. :param kwargs: The keyword arguments corresponding to the fields to exclude on. *Multiple arguments are treated as if they are ANDed together.* :return: A new ``EmbeddedDocumentList`` containing the non-matching embedded documents. Raises ``AttributeError`` if a given keyword is not a valid field for the embedded document class. csg|]}|kr|qSrr)r^itemexcluderrrasz0EmbeddedDocumentList.exclude..rbrdrrhrris zEmbeddedDocumentList.excludecCst|S)z The number of embedded documents in the list. :return: The length of the list, equivalent to the result of ``len()``. rNr8rrrcountszEmbeddedDocumentList.countcKsL|||}t|dkr(td|jnt|dkrDtdt||dS)a Retrieves an embedded document determined by the given keyword arguments. :param kwargs: The keyword arguments corresponding to the fields to search on. *Multiple arguments are treated as if they are ANDed together.* :return: The embedded document matched by the given keyword arguments. Raises ``DoesNotExist`` if the arguments used to query an embedded document returns no results. ``MultipleObjectsReturned`` if more than one result is returned. rz!%s matching query does not exist.z%d items returned, instead of 1)rcrNrr!rrdrrrr-s    zEmbeddedDocumentList.getcCst|dkr|dSdS)zVReturn the first embedded document in the list, or ``None`` if empty. rNrjr8rrrfirsts zEmbeddedDocumentList.firstcKs>|j}|jj|jj}|j|j|f||j|jdS)a Creates a new instance of the EmbeddedDocument and appends it to this EmbeddedDocumentList. .. note:: the instance of the EmbeddedDocument is not automatically saved to the database. You still need to call .save() on the parent Document. :param values: A dictionary of values for the embedded document. :return: The new embedded document instance. )r!r _fieldsfieldZdocument_type_objrQ)rrer&Z EmbeddedClassrrrcreate s zEmbeddedDocumentList.createcOs|jj||dS)z Saves the ancestor document. :param args: Arguments passed up to the ancestor Document's save method. :param kwargs: Keyword arguments passed up to the ancestor Document's save method. N)r save)rrrrrrrr1s zEmbeddedDocumentList.savecCs,t|}|D]}|j|j|q t|S)a Deletes the embedded documents from the database. .. note:: The embedded document changes are not automatically saved to the database after calling this method. :return: The number of entries deleted. )r2r r!rTrN)rrergrrrdelete<s zEmbeddedDocumentList.deletecKsHt|dkrdSt|}|D]"}|D]\}}t|||q(qt|S)a Updates the embedded documents with the given replacement values. This function does not support mongoDB update operators such as ``inc__``. .. note:: The embedded document changes are not automatically saved to the database after calling this method. :param update: A dictionary of update values to apply to each embedded document. :return: The number of entries updated. r)rNr2rYsetattr)rrHrergkvrrrrHLs zEmbeddedDocumentList.updateN)r@rArB classmethodr]rcrfrirkr-rmrqrrrsrHrrrrr s   r c@seZdZdZddddddhZiZdd Zd d Zd d ZddZ d'ddZ d(ddZ ddZ ddZ ddZddZddZdd Zd!d"Zd#d$Zed%d&ZdS))rrr-rF iteritemsrYkeysrqcKs"|D]\}}t|||qdSr )rYrt)rrrurvrrrr#hszStrictDict.__init__cCsD||jkrd|n|}z t||WStk r>t|YnXdSN _reserved_)_special_fieldsrZAttributeErrorr*r?rrrr)ls  zStrictDict.__getitem__cCs"||jkrd|n|}t|||Srz)r|rt)rrr4rrrr1sszStrictDict.__setitem__cCs t||Sr r>r?rrr __contains__wszStrictDict.__contains__NcCs(z ||WStk r"|YSXdSr )r*r+rrrr-zs zStrictDict.getcCs4|||}zt||Wntk r.YnX|Sr )r-delattrr})rrr,rvrrrrFs  zStrictDict.popccs|D]}|||fVqdSr rr?rrrrxszStrictDict.iteritemscsfddtDS)Ncsg|]}||fqSrrr^rur8rrrasz$StrictDict.items..iterr8rr8rrYszStrictDict.itemscCst|Sr rr8rrriterkeysszStrictDict.iterkeyscCs tt|Sr )r2rr8rrrryszStrictDict.keyscsfddjDS)Nc3s|]}t|r|VqdSr r~)r^rr8rr s z&StrictDict.__iter__..) __slots__r8rr8rrPszStrictDict.__iter__cCstt|Sr )rNr2rYr8rrr__len__szStrictDict.__len__cCst|t|kSr )r2rYrotherrrr__eq__szStrictDict.__eq__cCs ||k Sr rrrrr__ne__szStrictDict.__ne__csPtfdd|Dt}|jkrFGfddd}|j|<j|S)Nc3s$|]}|jkrd|n|VqdS)r{N)r|r)r\rrrsz$StrictDict.create..cseZdZZddZdS)z-StrictDict.create..SpecificStrictDictcSsdddd|DS)Nz{%s}, css"|]\}}d|d|VqdS)"z": Nr)r^rurvrrrrszIStrictDict.create..SpecificStrictDict.__repr__..)joinrYr8rrr__repr__s z6StrictDict.create..SpecificStrictDict.__repr__N)r@rArBrrr)allowed_keys_tuplerrSpecificStrictDictsr)tuple frozenset_classes)r\Z allowed_keysrr)rr\rrqs   zStrictDict.create)N)N)r@rArBrr|rr#r)r1rr-rFrxrYrryrPrrrrwrqrrrrrcs&  rcsPeZdZdZdddZeddZdfdd Zd d Zd d Z ddZ Z S)r ) _cached_doc passthrough document_typeFcCs6|jr |r0|jjj|jd|_|js0td||jS)N)pkz)Trying to dereference unknown document %s)rrZobjectsr-rr)rforcerrrfetchs   zLazyReference.fetchcCs|jSr )idr8rrrrszLazyReference.pkNcs*||_||_||_t|j|dSr )rrrr"r#Z_get_collection_name)rrrZ cached_docrr'rrr#szLazyReference.__init__cCs|js t|}||Sr )rr*rrr&documentrrrr)szLazyReference.__getitem__cCsDt|dst|}z ||WStk r>tYnXdS)Nr)object__getattribute__r}rr*rrrr __getattr__s  zLazyReference.__getattr__cCsd|jd|jdS)Nz)rrr8rrrrszLazyReference.__repr__)F)NF) r@rArBrrpropertyrr#r)rrrKrrr'rr s   r )rZbsonrZmongoengine.commonrZmongoengine.errorsrr__all__rrr0rr2rr rr rrrrs     @W$R