U d @sUdZddlZddlZddlmZddlmZmZm Z m Z m Z mZm Z m Z mZmZmZmZeedZee eed<edZedZed ZGd d d eeefZdS) zTools for creating and manipulating SON, the Serialized Ocument Notation. Regular dictionaries can be used instead of SON objects, but not when the order of keys is important. A SON object can be used just like a normal Python dictionary. N)Mapping) AnyDictIterableIteratorListrOptionalPatternTupleTypeTypeVarUnionRE_TYPE_Key_Value_TcseZdZUdZeeed<d;eee e e fe e e e ffeddddZedeeddfd d Zed d d Ze e ddddZe ddddZdd ddZee d ddZe edddZee d ddZee d ddZee d ddZdd fd d! Ze e e d"d#d$Ze ee efee efd%d&d'Z e e e fd d(d)Z!dSONzSON data. A subclass of dict that maintains ordering of keys and provides a few extra niceties for dealing with SON. SON provides an API similar to collections.OrderedDict. _SON__keysN)datakwargsreturncKs(g|_t|||||dSN)rdict__init__update)selfrrr,/tmp/pip-unpacked-wheel-oblwsawz/bson/son.pyr;s  z SON.__init__zSON[_Key, _Value])clsargsrrcstj|f||}g|_|Sr)super__new__r)rr rinstance __class__rrr"Esz SON.__new__)rcCs:g}|jD] }|d|d||dq dd|S)N(z, )z SON([%s]))rappendjoin)rresultkeyrrr__repr__Js z SON.__repr__)r+valuercCs(||jkr|j|t|||dSr)rr(r __setitem__)rr+r-rrrr.Ps  zSON.__setitem__)r+rcCs|j|t||dSr)rremover __delitem__rr+rrrr0Us zSON.__delitem__cCst}|||Sr)rrrotherrrrcopyYs zSON.copyccs|jEdHdSrrrrrr__iter__asz SON.__iter__cCs ||jkSrr5r1rrrhas_keydsz SON.has_keycCs|Sr)r7r6rrriterkeysgsz SON.iterkeysccs|D]\}}|VqdSritems)r_vrrr itervalueskszSON.itervaluescCsdd|DS)NcSsg|] \}}|qSrr).0r<r=rrr pszSON.values..r:r6rrrvaluesosz SON.valuescsg|_tdSr)rr!clearr6r$rrrBrsz SON.clear)r+defaultrcCs,z ||WStk r&|||<YnX|SrKeyErrorrr+rCrrr setdefaultvs  zSON.setdefault)r+r rcGsbt|dkr$tdtdt|z ||}Wn&tk rV|rP|dYSYnX||=|S)Nz&pop expected at most 2 arguments, got r)len TypeErrorreprrE)rr+r r-rrrpop}s   zSON.popcCsDztt|\}}Wntk r4tdYnX||=||fS)Nzcontainer is empty)nextiterr; StopIterationrE)rkr=rrrpopitems z SON.popitem)r3rrcKs~|dkr nbt|dr0|D]\}}|||<qnt|tr2t|t|ko0t|t|kS||kS)z|Comparison to another SON is order-sensitive while comparison to a regular dictionary is order-insensitive. ) isinstancerrIlistr;to_dictr2rrr__eq__s (z SON.__eq__cCs ||k Srrr2rrr__ne__sz SON.__ne__cCs t|jSr)rIrr6rrr__len__sz SON.__len__cs ttdfdd t|S)zConvert a SON document to a normal Python dictionary instance. This is trickier than just *dict(...)* because it needs to be recursive. )r-rcsDt|trfdd|DSt|tr<fdd|DS|SdS)Ncsg|] }|qSrr)r?r=transform_valuerrr@sz8SON.to_dict..transform_value..csi|]\}}||qSrr)r?rPr=r[rr sz8SON.to_dict..transform_value..)rUrV_Mappingr;)r-r[rrr\s   z$SON.to_dict..transform_value)rrr6rr[rrWsz SON.to_dict)memorcCsZt}t|}||kr||S|||<|D]&\}}t|tsLt||}|||<q.|Sr)ridr;rUrr4deepcopy)rr_outZval_idrPr=rrr __deepcopy__s   zSON.__deepcopy__)N)N)N),__name__ __module__ __qualname____doc__rr__annotations__rr rrrrr rr r"strr,r.r0r4rr7boolr8r9r>rArBrGrrLrQrrTrXrYintrZrrWrc __classcell__rrr$rr1s:  " " *r)rgr4recollections.abcrr^typingrrrrrrr r r r r typecompilerrhrrrrrrrrs 8