U Û~fhUã@s°UdZddlmZddlZddlZddlmZddlm Z m Z m Z m Z mZm Z mZmZmZmZmZmZee d¡ƒZded<ed ƒZed ƒZed ƒZGd d „d e eefƒZdS)zðTools 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. é)Ú annotationsN)ÚMapping) ÚAnyÚDictÚIterableÚIteratorrÚOptionalÚPatternÚTupleÚTypeÚTypeVarÚUnionÚcastÚzType[Pattern[Any]]ÚRE_TYPEÚ_KeyÚ_ValueÚ_TcsˆeZdZUdZded<dOddddœd d „Zd ddd d œ‡fdd„ Zddœdd„Zddddœdd„Zdddœdd„Z d dœdd„Z ddœdd „Z dd!dœd"d#„Z ddœd$d%„Z d&dœd'd(„Zd)dœd*d+„Zddœ‡fd,d-„ Zdddd.œd/d0„Zdd1d1d2œd3d4„Zd5dœd6d7„ZdPd8ddd9œd:d;„ZdQddd?„Zdd!d@œdAdB„Zdd!d@œdCdD„ZdEdœdFdG„ZdHdœdIdJ„ZdKd dLœdMdN„Z‡ZS)RÚSONz¾SON 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. z list[Any]Ú _SON__keysNzEOptional[Union[Mapping[_Key, _Value], Iterable[Tuple[_Key, _Value]]]]rÚNone)ÚdataÚkwargsÚreturncKs(g|_t |¡| |¡| |¡dS©N)rÚdictÚ__init__Úupdate)Úselfrr©rú,/tmp/pip-unpacked-wheel-36gvocj8/bson/son.pyr<s  z SON.__init__zType[SON[_Key, _Value]]zSON[_Key, _Value])ÚclsÚargsrrcstƒj|f|ž|Ž}g|_|Sr)ÚsuperÚ__new__r)r!r"rÚinstance©Ú __class__rr r$Fsz SON.__new__Ústr)rcCs:g}|jD] }| d|›d||›d¡q dd |¡S)Nú(z, ú)z SON([%s]))rÚappendÚjoin)rÚresultÚkeyrrr Ú__repr__Ks z SON.__repr__rr)r.ÚvaluercCs(||jkr|j |¡t |||¡dSr)rr+rÚ __setitem__)rr.r0rrr r1Qs  zSON.__setitem__)r.rcCs|j |¡t ||¡dSr)rÚremoverÚ __delitem__©rr.rrr r3Vs zSON.__delitem__cCstƒ}| |¡|Sr)rr©rÚotherrrr ÚcopyZs zSON.copyzIterator[_Key]ccs|jEdHdSr©r©rrrr Ú__iter__bsz SON.__iter__ÚboolcCs ||jkSrr8r4rrr Úhas_keyesz SON.has_keycCs| ¡Sr)r:r9rrr Úiterkeyshsz SON.iterkeyszIterator[_Value]ccs| ¡D]\}}|VqdSr©Úitems)rÚ_Úvrrr Ú itervalueslszSON.itervaluesz list[_Value]cCsdd„| ¡DƒS)NcSsg|] \}}|‘qSrr)Ú.0r@rArrr Ú qszSON.values..r>r9rrr Úvaluespsz SON.valuescsg|_tƒ ¡dSr)rr#Úclearr9r&rr rFssz SON.clear)r.ÚdefaultrcCs,z ||WStk r&|||<YnX|Sr©ÚKeyError©rr.rGrrr Ú setdefaultws  zSON.setdefaultzUnion[_Value, _T])r.r"rcGsbt|ƒdkr$tdtdt|ƒƒƒ‚z ||}Wn&tk rV|rP|dYS‚YnX||=|S)Néz&pop expected at most 2 arguments, got r)ÚlenÚ TypeErrorÚreprrI)rr.r"r0rrr Úpop~s   zSON.popzTuple[_Key, _Value]cCsFztt| ¡ƒƒ\}}Wntk r6tdƒd‚YnX||=||fS)Nzcontainer is empty)ÚnextÚiterr?Ú StopIterationrI)rÚkrArrr ÚpopitemŠs z SON.popitemz Optional[Any])r6rrcKsz|dkr n^t|dƒr0| ¡D]\}}|||<qn8t|dƒrR|D]}||||<q>n|D]\}}|||<qV|rv| |¡dS)Nr?Úkeys)Úhasattrr?r)rr6rrTrArrr r’s     z SON.updatezOptional[Union[_Value, _T]]zUnion[_Value, _T, None]cCs(z ||WStk r"|YSXdSrrHrJrrr Úget¢s zSON.get)r6rcCsDt|tƒr2t|ƒt|ƒko0t| ¡ƒt| ¡ƒkStt| ¡|kƒS)z|Comparison to another SON is order-sensitive while comparison to a regular dictionary is order-insensitive. )Ú isinstancerrMÚlistr?rr;Úto_dictr5rrr Ú__eq__ªs (z SON.__eq__cCs ||k Srrr5rrr Ú__ne__²sz SON.__ne__ÚintcCs t|jƒSr)rMrr9rrr Ú__len__µsz SON.__len__údict[_Key, _Value]cs&dddœ‡fdd„ ‰tdˆt|ƒƒƒS)z¡Convert a SON document to a normal Python dictionary instance. This is trickier than just *dict(...)* because it needs to be recursive. r)r0rcsDt|tƒr‡fdd„|DƒSt|tƒr<‡fdd„| ¡DƒS|SdS)Ncsg|] }ˆ|ƒ‘qSrr)rCrA©Útransform_valuerr rDÁsz8SON.to_dict..transform_value..csi|]\}}|ˆ|ƒ“qSrr)rCrTrArarr Ú Ãsz8SON.to_dict..transform_value..)rYrZÚ_Mappingr?)r0rarr rb¿s   z$SON.to_dict..transform_valuer`)rrr9rrar r[¸sz SON.to_dictzdict[int, SON[_Key, _Value]])ÚmemorcCsZtƒ}t|ƒ}||kr||S|||<| ¡D]&\}}t|tƒsLt ||¡}|||<q.|Sr)rÚidr?rYrr7Údeepcopy)rreÚoutZval_idrTrArrr Ú __deepcopy__És   zSON.__deepcopy__)N)N)N)Ú__name__Ú __module__Ú __qualname__Ú__doc__Ú__annotations__rr$r/r1r3r7r:r<r=rBrErFrKrPrUrrXr\r]r_r[riÚ __classcell__rrr&r r2s4 þ  ÿr)rmÚ __future__rr7ÚreÚcollections.abcrrdÚtypingrrrrrr r r r r rÚtypeÚcompilerrnrrrrrrrr Ús  8