o Tjd@sVdZddlZddlZddlZddlZddlmZmZm Z m Z m Z m Z m Z mZmZmZmZmZedZedZedZddlmZddlmZd ed ed dfd d ZGdddeeZejj eGdddeee eZ Gdddeee eZ!Gddde eefZ"Gddde eefZ#GdddZ$GdddZ%GdddZ&dS)aContains container classes to represent different protocol buffer types. This file defines container classes which represent categories of protocol buffer field types which need extra maintenance. Currently these categories are: - Repeated scalar fields - These are all repeated fields which aren't composite (e.g. they are of simple types like int32, string, etc). - Repeated composite fields - Repeated fields which are composite. This includes groups and nested messages. N) AnyIterableIteratorListMutableMappingMutableSequenceNoReturnOptionalSequenceTypeVarUnionoverload_T_K_V)FieldDescriptor)message is_frozenmsgreturncCs|r tjdtdddSdS)NzwMutating messages or containers returned by GetOptions() is deprecated and will raise an exception in a future release.)category stacklevel)warningswarn FutureWarning)rrrf/home/dinkstrade/pdmp-scanner/venv/lib/python3.10/site-packages/google/protobuf/internal/containers.py _CheckFrozen.s rc@seZdZdZgdZdeddfddZedede fd d Z ede de e fd d Z d d Z defd dZ dedefddZdZdefddZdddZdddZdddZdddZdS) BaseContainerzBase container class._message_listener_values_frozenmessage_listenerrNcCs||_g|_d|_dS)zArgs: message_listener: A MessageListener implementation. The RepeatedScalarFieldContainer will call this object's Modified() method when it is modified. FNr )selfr$rrr__init__>s zBaseContainer.__init__keycCdSNrr%r'rrr __getitem__IzBaseContainer.__getitem__cCr(r)rr*rrrr+Mr,cC |j|S)z$Retrieves item by the specified key.r"r*rrrr+Q cC t|jS)z0Returns the number of elements in the container.lenr"r%rrr__len__Ur/zBaseContainer.__len__othercCs ||k S)z3Checks if another instance isn't equal to this one.rr%r5rrr__ne__Ys zBaseContainer.__ne__cCr0r)reprr"r3rrr__repr__` zBaseContainer.__repr__cC d|_dSNTr#r3rrr _SetFrozencr;zBaseContainer._SetFrozenBaseContainer[_T]cCt|jd|S)NzContainer is immutablerr#r3rrr_AssureWritablef zBaseContainer._AssureWritablecOs4|d|vr|d|d<|jj|i|dS)N sort_functioncmp)rCpopr"sort)r%argskwargsrrrrHjszBaseContainer.sortcCs||jdSr))rCr"reverser3rrrrKsszBaseContainer.reverserN)rr@)__name__ __module__ __qualname____doc__ __slots__rr&r intrr+slicerr4boolr7__hash__strr:r?rCrHrKrrrrr8s"     rc s\eZdZdZddgZ d-dedededdffd d Zd eddfd d Zde d eddfddZ de eddfddZ de de efddfddZdefddZd.dee defddZede d eddfddZeded e eddfd dZd/d!dZde e efddfd"d#Zdedefd$d%Z d-d&eddfd'd(Zdefd)d*Zd0d+d,ZZS)1RepeatedScalarFieldContainerzGSimple, type-checked, list-like container for holding repeated scalars. _type_checker_fieldNr$ type_checkerfieldrcst|||_||_dS)aArgs: message_listener: A MessageListener implementation. The RepeatedScalarFieldContainer will call this object's Modified() method when it is modified. type_checker: A type_checkers.ValueChecker instance to run on elements inserted into this container. N)superr&rXrY)r%r$rZr[ __class__rrr&s  z%RepeatedScalarFieldContainer.__init__valuecCs6||j|j||jjs|jdSdS)z6Appends an item to the list. Similar to list.append().N)rCr"appendrX CheckValuer!dirtyModifiedr%r_rrrr`s z#RepeatedScalarFieldContainer.appendr'cCs8||j||j||jjs|jdSdS)zEInserts the item at the specified position. Similar to list.insert().N)rCr"insertrXrar!rbrcr%r'r_rrrres z#RepeatedScalarFieldContainer.insertelem_seqcs@t|}fdd|D}|rj|jdS)zBExtends by appending the given iterable. Similar to list.extend().csg|]}j|qSr)rXra).0elemr3rr sz7RepeatedScalarFieldContainer.extend..N)rCiterr"extendr!rc)r%rg elem_seq_iter new_valuesrr3rrls  z#RepeatedScalarFieldContainer.extendr5z RepeatedScalarFieldContainer[_T]cC"||j||jdS)zAppends the contents of another repeated field of the same type to this one. We do not check the types of the individual fields. N)rCr"rlr!rcr6rrr MergeFroms z&RepeatedScalarFieldContainer.MergeFromricCroz8Removes an item from the list. Similar to list.remove().NrCr"remover!rcr%rirrrrs z#RepeatedScalarFieldContainer.removecC ||j|}|||SzDRemoves and returns an item at a given index. Similar to list.pop().rCr" __delitem__rfrrrrG  z RepeatedScalarFieldContainer.popcCr(r)rrfrrr __setitem__r,z(RepeatedScalarFieldContainer.__setitem__cCr(r)rrfrrrr|r,cCsf|t|tr#|jdurtdt|jj||j|<|j dS|j||j|<|j dS)z(Sets the item on the specified position.NzExtended slices not supported) rC isinstancerSstep ValueErrormaprXrar"r!rcrfrrrr|s  cC||j|=|jdSz+Deletes the item at the specified position.NrCr"r!rcr*rrrrzz(RepeatedScalarFieldContainer.__delitem__cCs.||urdSt||jr|j|jkS||jkS)/Compares the current instance with another one.T)r}r^r"r6rrr__eq__s    z#RepeatedScalarFieldContainer.__eq__ unused_memocCs&tt|j|j|j}|||Sr))rWcopydeepcopyr!rXrYrp)r%rclonerrr __deepcopy__s  z)RepeatedScalarFieldContainer.__deepcopy__cKs td)Nz:Can't pickle repeated scalar fields, convert to list first)pickle PickleError)r%rJrrr __reduce__sz'RepeatedScalarFieldContainer.__reduce__cCsddl}|durl|jj}|tjkr|j}nW|tjkr|j}nN|tjkr'|j }nE|tj kr0|j }n<|tj kr9|j }n3|tjkrB|j}n*|tjkrK|j}n!|tjkrT|j}n|jjtjkr^d}n|jjtjkrht}ntd|j|j|ddS)NrSzSCode should never reach here: message type detected in RepeatedScalarFieldContainerT)dtyper)numpyrYcpp_typer CPPTYPE_INT32int32 CPPTYPE_INT64int64CPPTYPE_UINT32uint32CPPTYPE_UINT64uint64CPPTYPE_DOUBLEfloat64 CPPTYPE_FLOATfloat32 CPPTYPE_BOOLrT CPPTYPE_ENUMtype TYPE_BYTES TYPE_STRINGrV SystemErrorarrayr")r%rrnprrrr __array__s6        z&RepeatedScalarFieldContainer.__array__r)rvrL)NN)rMrNrOrPrQrr&rr`rRrerrlr rprsr rGr r|rSrzrTrrrrr __classcell__rrr]rrW}sL        rWcsFeZdZdZdgZdededdffdd Zd)fd d Zd edefd d Z deddfddZ de deddfddZ de eddfddZdede efddfddZdeddfddZd*dee defdd Zede deddfd!d"Zedede eddfd#d"Zd$d"Zdee efddfd%d&Zdedefd'd(ZZS)+RepeatedCompositeFieldContainerzBSimple, list-like container for holding repeated composite fields._message_descriptorr$message_descriptorrNcst|||_dS)aNote that we pass in a descriptor instead of the generated directly, since at the time we construct a _RepeatedCompositeFieldContainer we haven't yet necessarily initialized the type that will be contained in the container. Args: message_listener: A MessageListener implementation. The RepeatedCompositeFieldContainer will call this object's Modified() method when it is modified. message_descriptor: A Descriptor instance describing the protocol type that should be present in this container. We'll use the _concrete_class field of this descriptor when the client calls add(). N)r\r&r)r%r$rr]rrr& s  z(RepeatedCompositeFieldContainer.__init__cs"t|jD]}|qdSr))r\r?r"r%valr]rrr?2s   z*RepeatedCompositeFieldContainer._SetFrozenrJcKsH||jjdi|}||j|j||jjs"|j|S)z|Adds a new element at the end of the list and returns it. Keyword arguments may be used to initialize the element. Nr) rCr_concrete_class _SetListenerr!r"r`rbrc)r%rJ new_elementrrradd7s   z#RepeatedCompositeFieldContainer.addr_cCsN||j}||j|||j||jjs%|j dSdS)z+Appends one element by copying the message.N) rCrrrr!CopyFromr"r`rbrc)r%r_rrrrr`Ds    z&RepeatedCompositeFieldContainer.appendr'cCsP||j}||j|||j|||jjs&|j dSdS)z6Inserts the item at the specified position by copying.N) rCrrrr!rr"rerbrc)r%r'r_rrrrreNs   z&RepeatedCompositeFieldContainer.insertrgcCsV||jj}|j}|j}|D]}|}||||||q|dS)z|Extends by appending the given sequence of elements of the same type as this one, copying each individual message. N) rCrrr!r"rrpr`rc)r%rg message_classlistenervaluesrrrrrrlXs    z&RepeatedCompositeFieldContainer.extendr5z#RepeatedCompositeFieldContainer[_T]cCs|||dS)zwAppends the contents of another repeated field of the same type to this one, copying each individual message. N)rCrlr6rrrrphsz)RepeatedCompositeFieldContainer.MergeFromricCrorqrrrtrrrrssruz&RepeatedCompositeFieldContainer.removervcCrwrxryrfrrrrGyr{z#RepeatedCompositeFieldContainer.popcCr(r)rrfrrrr|r,z+RepeatedCompositeFieldContainer.__setitem__cCr(r)rrfrrrr|r,cCst|jjd)Nz( object does not support item assignment) TypeErrorr^rMrfrrrr|s cCrrrr*rrrrzrz+RepeatedCompositeFieldContainer.__delitem__cCs,||urdSt||jstd|j|jkS)rTzSCan only compare repeated composite fields against other repeated composite fields.)r}r^rr"r6rrrrs  z&RepeatedCompositeFieldContainer.__eq__rLr)rMrNrOrPrQrr&r?rrr`rRrerrlr rprsr rGr r|rSrzrTrrrrr]rrs.     rc @sReZdZdZgdZdededededdf d d Zd1d d Zd2ddZde de fddZ de de fddZ ede dee fddZede dedee effddZd3ddZde de defddZde ddfddZdefd d!Zdee fd"d#Zdefd$d%Zd3de dee de fd&d'Zd4d)d*Zd1d+d,Zd1d-d.Zdefd/d0ZdS)5 ScalarMapzGSimple, type-checked, dict-like container for holding repeated scalars.) _key_checker_value_checkerr"r!_entry_descriptorr#r$ key_checker value_checkerentry_descriptorrNcC(||_||_||_||_i|_d|_dSaArgs: message_listener: A MessageListener implementation. The ScalarMap will call this object's Modified() method when it is modified. key_checker: A type_checkers.ValueChecker instance to run on keys inserted into this container. value_checker: A type_checkers.ValueChecker instance to run on values inserted into this container. entry_descriptor: The MessageDescriptor of a map entry: key and value. FN)r!rrrr"r#)r%r$rrrrrrr&  zScalarMap.__init__cCr<r=r>r3rrrr?r;zScalarMap._SetFrozenScalarMap[_K, _V]cCrANzMap is immutablerBr3rrrrCrDzScalarMap._AssureWritabler'cCsL|j|}z|j|WSty%||j}||j|<|YSwr))rrar"KeyErrorrCr DefaultValue)r%r'rrrrr+s     zScalarMap.__getitem__itemcC|j|}||jvSr)rrar"r%rrrr __contains__s  zScalarMap.__contains__cCr(r)rr*rrrgetr,z ScalarMap.getdefaultcCr(r)rr%r'rrrrrr,cCs"|j|}||jvr||S|Sr)r)r%r'r checked_keyrrrrs  r_cCs8||j|}|j|}||j|<|jdSr))rCrrarr"r!rc)r%r'r_r checked_valuerrrr|s    zScalarMap.__setitem__cCs*||j|}|j|=|jdSr)rCrrar"r!rc)r%r'rrrrrz zScalarMap.__delitem__cCr0r)r1r3rrrr4r;zScalarMap.__len__cCr0r)rkr"r3rrr__iter__r;zScalarMap.__iter__cCr0r)r8r3rrrr:r;zScalarMap.__repr__cCsB||j|}|dkrtd||jvr|||||S)Nz0The value for scalar map setdefault must be set.)rCrrarr"r|)r%r'r_rrrr setdefaults   zScalarMap.setdefaultr5cCs$||j|j|jdSr))rCr"updater!rcr6rrrrpszScalarMap.MergeFromcC|j}||_d|d<dSr)r"rr%originalrrrInvalidateIterators  zScalarMap.InvalidateIteratorscC ||j|jdSr)rCr"clearr!rcr3rrrr zScalarMap.clearcC|jjSr)rrr3rrr GetEntryClass"zScalarMap.GetEntryClassrL)rrr))r5rrN) rMrNrOrPrQrr&r?rCrrr+rTrr r rrr r|rzrRr4rrrVr:rrprrrrrrrrsB        rc @sdeZdZdZgdZdededededdf d d Zd3d d Zd4ddZde de fddZ de de fddZ e de dee fddZe de dedee effddZd5ddZde defddZde de defddZde ddfd d!Zdefd"d#Zdee fd$d%Zdefd&d'Zd5de dee de fd(d)Zd6d+d,Zd3d-d.Zd3d/d0Z defd1d2Z!dS)7 MessageMapzESimple, type-checked, dict-like container for with submessage values.)rr"r!rrr#r$rrrrNcCrr)r!rrrr"r#)r%r$rrrrrrr&3rzMessageMap.__init__cCs"d|_|jD]}|qdSr=)r#r"rr?rrrrr?Ls zMessageMap._SetFrozenMessageMap[_K, _V]cCrArrBr3rrrrCQrDzMessageMap._AssureWritabler'cCsb|j|}z|j|WSty0||j}||j||j|<|j |YSwr)) rrar"rrCrrrr!rc)r%r'rrrrr+Us       zMessageMap.__getitem__cCs||S)aaget_or_create() is an alias for getitem (ie. map[key]). Args: key: The key to get or create in the map. This is useful in cases where you want to be explicit that the call is mutating the map. This can avoid lint errors for statements like this that otherwise would appear to be pointless statements: msg.my_map[key] rr*rrr get_or_createas zMessageMap.get_or_createcCr(r)rr*rrrror,zMessageMap.getrcCr(r)rrrrrrsr,cCs||vr||S|Sr)rrrrrrzsrcCrr)rrrrrrs  zMessageMap.__contains__r_cC|td)Nz5May not set values directly, call my_map[key].foo = 5)rCrrfrrrr|szMessageMap.__setitem__cCs*||j|}|j|=|jdSr)rr*rrrrzrzMessageMap.__delitem__cCr0r)r1r3rrrr4r;zMessageMap.__len__cCr0r)rr3rrrrr;zMessageMap.__iter__cCr0r)r8r3rrrr:r;zMessageMap.__repr__cCr)NzISet message map value directly is not supported, call my_map[key].foo = 5)rCNotImplementedErrorrfrrrrszMessageMap.setdefaultr5cCs8||jD]}||vr||=||||qdSr))rCr"r)r%r5r'rrrrps  zMessageMap.MergeFromcCrr)rrrrrrrzMessageMap.InvalidateIteratorscCrr)rr3rrrrrzMessageMap.clearcCrr)rr3rrrrrzMessageMap.GetEntryClassrL)rrr))r5rrN)"rMrNrOrPrQrr&r?rCrrr+rr r rrr rTrrr|rzrRr4rrrVr:rrprrrrrrrr&sD        rc@s0eZdZdZgdZddZddZddZd S) _UnknownFieldzA parsed unknown field. _field_number _wire_type_datacCs||_||_||_dSr)r)r% field_number wire_typedatarrrr&sz_UnknownField.__init__cCs |j|jkSr))rr6rrr__lt__s z_UnknownField.__lt__cCs0||urdS|j|jko|j|jko|j|jkSr=rr6rrrrs   z_UnknownField.__eq__N)rMrNrOrPrQr&rrrrrrrs  rc@s@eZdZddZddZeddZeddZed d Zd S) UnknownFieldRefcCs||_||_dSr))_parent_index)r%parentindexrrrr&s zUnknownFieldRef.__init__cCs*|jstd|jt|jkrtddS)NzAUnknownField does not exist. The parent message might be cleared.)rrrr2r3rrr _check_validszUnknownFieldRef._check_validcC||j|jjSr))rr _internal_getrrr3rrrrzUnknownFieldRef.field_numbercCrr))rrrrrr3rrrrrzUnknownFieldRef.wire_typecCrr))rrrrrr3rrrrrzUnknownFieldRef.dataN) rMrNrOr&rpropertyrrrrrrrrs  rc@s^eZdZdZdgZddZddZddZd d Zd d Z d dZ ddZ ddZ ddZ dS)UnknownFieldSetzUnknownField containerr"cCs g|_dSr)r.r3rrrr&r;zUnknownFieldSet.__init__cCsP|jdur tdt|j}|dkr||7}|dks||kr#tdjt||S)NBUnknownFields does not exist. The parent message might be cleared.rzindex %d out of range)r"rr2 IndexErrorrr)r%rsizerrrr+s    zUnknownFieldSet.__getitem__cCr-r)r.)r%rrrrrr;zUnknownFieldSet._internal_getcCs|jdur tdt|jS)Nr)r"rr2r3rrrr4s  zUnknownFieldSet.__len__cCst|||}|j||Sr))rr"r`)r%rrr unknown_fieldrrr_adds  zUnknownFieldSet._addccs$tt|D]}t||VqdSr))ranger2r)r%irrrrszUnknownFieldSet.__iter__cCs|durdS|j|jdSr))r"rlr6rrr_extend"szUnknownFieldSet._extendcCs>||urdSt|j}|dur| S|t|j}||kSr=)listr"rHsorted)r%r5r other_valuesrrrr(s  zUnknownFieldSet.__eq__cCs,|jD] }t|jtr|jqd|_dSr))r"r}rr_clearrdrrrr5s    zUnknownFieldSet._clearN)rMrNrOrPrQr&r+rr4rrrrrrrrrrs  r)'rPcollections.abc collectionsrrrtypingrrrrrrrr r r r r rrrgoogle.protobuf.descriptorrgoogle.protobufrrTrVrrabcregisterrWrrrrrrrrrrs2 8   B #