o j0n@sdZddlZgdZedejddZzeWn ey&ddZYnwGdd d e Z d d Z d d Z ddZ ddZGdddeZdZGdddeZiZded<ddZeed<[ddZeed<[ddZeed<[ddZeed<[d d!Zeed!<[ed"krd#d$Zeed$<[n$d%d&Zeed&<[d'd(Zeed(<[d)d*Zeed*<[d+d,Zeed,<[d-d.Zeed.<[d/d0Z e ed0<[ d1d2Z!e!ed2<[!d3d4Z"e"ed4<["e d5d6Z#e#ed6<[#e d7d8Z$e$ed8<[$ed9e feZ[Gd:d;d;e%eZ&dzPython EnumerationsN)EnumIntEnumunique%s.%scCs|D]}|r dSqdS)NTF)iterableelementrr/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/impacket/dcerpc/v5/enum.pyanys r c@s4eZdZdZd ddZd ddZddZd d ZdS) _RouteClassAttributeToGetattrajRoute attribute access on a class to __getattr__. This is a descriptor, used to define attributes that act differently when accessed through an instance and through a class. Instance access remains normal, but access to an attribute through a class will be routed to the class's __getattr__ method; this is done by raising AttributeError. NcCs ||_dSN)fget)selfrrrr __init__& z&_RouteClassAttributeToGetattr.__init__cCs|durt||Sr )AttributeErrorr)rinstance ownerclassrrr __get__)s z%_RouteClassAttributeToGetattr.__get__cCtd)Nzcan't set attributer)rrvaluerrr __set__.z%_RouteClassAttributeToGetattr.__set__cCr)Nzcan't delete attributer)rrrrr __delete__1rz(_RouteClassAttributeToGetattr.__delete__r )__name__ __module__ __qualname____doc__rrrrrrrr r s    r cCst|dpt|dpt|dS)z5Returns True if obj is a descriptor, False otherwise.rrr)hasattr)objrrr _is_descriptor5s r"cCsT|dd|ddkodkno)|dddko)|dddko)t|dkS) z3Returns True if a __dunder__ name, False otherwise.Nr___lennamerrr _is_dunder=s( r-cCsL|d|dko dkno%|dddko%|dddko%t|dkS)z1Returns True if a _sunder_ name, False otherwise.rr&rr#r)r+rrr _is_sunderEs  r0cCsdd}||_d|_dS)z"Make the given class un-picklable.cSs td|)Nz%r cannot be pickled) TypeErrorrrrr _break_on_call_reduceOs z6_make_class_unpicklable.._break_on_call_reducez N) __reduce__r)clsr3rrr _make_class_unpicklableMs r6cs,eZdZdZfddZfddZZS) _EnumDictzTrack enum member order and ensure member names are not reused. EnumMeta will use the names found in self._member_names as the enumeration member names. cstt|g|_dSr )superr7r _member_namesr2 __class__rr r\s z_EnumDict.__init__cstdkr |dkr dSt|rtdt|rn!||jvr"td|t|s8||vr2td|||j|tt | ||dS)aChanges anything not dundered or not a descriptor. If a descriptor is added with the same name as an enum member, the name is removed from _member_names (this may leave a hole in the numerical sequence of values). If an enum member name is used twice, an error is raised; duplicate values are not checked for. Single underscore (sunder) names are reserved. Note: in 3.x __order__ is simply discarded as a not necessary piece leftover from 2.x @ __order__Nz(_names_ are reserved for future Enum usezAttempted to reuse key: %rzKey already defined as: %r) pyverr0 ValueErrorr-r9r1r"appendr8r7 __setitem__)rkeyrr:rr rA`s   z_EnumDict.__setitem__)rrrrrrA __classcell__rrr:r r7Us r7cseZdZdZeddZfddZd'ddZd d Zfd d Z d dZ e ddZ ddZ ddZddZddZddZddZfddZd'dd Zed!d"Zed#krbed$d%ZZSed&d%ZZS)(EnumMetazMetaclass for EnumcCstSr )r7)metaclsr5basesrrr __prepare__zEnumMeta.__prepare__c sttur}t|D]\}}||<q||\}}|||\} } } tfddjD} jD]} | =q8ddurTjtdkrQd}n!d}nd=d}tdkrr dd fd d | D}|7t | t d g@}|rt d d |ftt||||g_i_|_i_|tur|jddurt| durj} D]{}| |}t|ts|f}n|}|tur|f}| r|s| }t|ds||_n| g|R}t|ds|||_|j}||_|_|j|jD]\} }|j|jkr|}nqj ||j|<z|j|<Wqt!y2Yqw|sLdd t"fdd jDdddD_t#durXt$dt#j%dD]$} t&| }t&|| d}t&|| d}|dur}||ur}t$| |qZtdkrt't(rt$dt&t(dntdkrt't(rdD] }t$|t&t(|qt#dur| rt$djdt$dt#jdS)Nc3s|] }||fVqdSr r).0k) classdictrr sz#EnumMeta.__new__..r=r<FT, csg|]}|vr|qSrrrIr,)r=rr sz$EnumMeta.__new__..mrozInvalid enum member name(s): %s, __getnewargs___value_cSsg|]}|dqS)rr)rIerrr rPscsg|] }|j|fqSr _member_map_rO) enum_classrr rPscSs |djS)Nr/rT)trrr s z"EnumMeta.__new__..)rB)__repr____str__ __format__@__cmp__)__le____lt____gt____ge____eq____ne____hash____member_new____new__))typedictr7items _get_mixins_ _find_new_r9getr>replacesplitsetr?joinr8rDri_member_names_rW _member_type__value2member_map_object__dict__r6 isinstancetupler rT_name_ __objclass__rrr@r1sortedrsetattrrSgetattr issubclassint)rEr5rFrK original_dictrJv member_type first_enumrisave_newuse_argsmembersr,order_specifiedaliases invalid_names member_namerargs enum_membercanonical_member class_method obj_method enum_methodmethodr:)r=rKrXr ris                      zEnumMeta.__new__NcCs&|dur |||S|j||||dS)aEither returns an existing member, or creates a new enum class. This method is used both when an enum class is given a value to match to an enumeration member (i.e. Color(3)) and for the functional API (i.e. Color = Enum('Color', names='red green blue')). When used for the functional API: `module`, if set, will be stored in the new class' __module__ attribute; `type`, if set, will be mixed in as the first base class. Note: if `module` is not set this routine will attempt to discover the calling module by walking the frame stack; if this is unsuccessful the resulting class will not be pickleable. N)modulerj)ri_create_)r5rnamesrrjrrr __call__+s zEnumMeta.__call__cCst||o |j|jvSr )ryr,rW)r5memberrrr __contains__@szEnumMeta.__contains__cs,||jvr td|jtt||dS)Nz%s: cannot delete Enum member.)rWrrr8rD __delattr__)r5attrr:rr rCs zEnumMeta.__delattr__cCsgd|jS)N)r;r __members__rrtr2rrr __dir__KszEnumMeta.__dir__cCs |jS)zReturns a mapping of member name->value. This mapping lists all enum members, including aliases. Note that this is a copy of the internal mapping. )rWcopyr5rrr rOs zEnumMeta.__members__cCs4t|rt|z|j|WStyt|w)a5Return the enum member matching `name` We use __getattr__ instead of descriptors or inserting into the enum class' __dict__ in order to support `name` and `value` being both properties for enum members (which live in the class' __dict__) and enum members themselves. )r-rrWKeyErrorr5r,rrr __getattr__Ys   zEnumMeta.__getattr__cCs |j|Sr rVrrrr __getitem__irzEnumMeta.__getitem__csfddjDS)Nc3|]}j|VqdSr rVrOrrr rLmz$EnumMeta.__iter__..rrrrr __iter__lszEnumMeta.__iter__csfddtjDS)Nc3rr rVrOrrr rLprz(EnumMeta.__reversed__..)reversedrtrrrr __reversed__ozEnumMeta.__reversed__cC t|jSr )r*rtrrrr __len__rrzEnumMeta.__len__cCs d|jS)Nz )rrrrr r\urzEnumMeta.__repr__cs4|jdi}||vrtdtt|||dS)zBlock attempts to reassign Enum members. A simple assignment to the class namespace only changes one of the several possible ways to get an Enum member from the Enum class, resulting in an inconsistent Enumeration. rWzCannot reassign members.N)rxrorr8rD __setattr__)r5r,r member_mapr:rr rxszEnumMeta.__setattr__c Cs.|j}|dur |f}n||f}|||}g}t|tr$|dd}t|ttfr;t|dtr;ddt|D}|D]} t| trL| || } } n| \} } | || <| | q=t| tsfd ||d<| ||||} |durz t djd }Wn ttfyYnw|durt| | S|| _| S) a}Convenience method to create a new Enum class. `names` can be: * A string containing member names, separated either with spaces or commas. Values are auto-numbered from 1. * An iterable of member names. Values are auto-numbered from 1. * An iterable of (member name, value) pairs. * A mapping of member name -> value. NrMrNrcSsg|] \}}||dfqS)r/r)rIirUrrr rPz%EnumMeta._create_..r=rr)r;rGrystrrprqrzlist enumerater@rsri_sys _getframe f_globalsrr?r6r) r5 class_namerrrjrErFrKr=itemr member_valuerXrrr rs<      zEnumMeta._create_cCs|rtdur ttfSd}}|D]}|tur"t|tr"|jr"tdqt|ts,tdt|dts?|d}|d}||fS|djD]}t|trR|durQ|}qD|durX|}qD||fS)zReturns the type for creating enum members, and the first inherited enum class. bases: the tuple of bases that was given to __new__ NzCannot extend enumerationszHnew enumerations must be created as `ClassName([mixin_type,] enum_type)`rr.)rrwrrtr1__mro__)rFrrbaserrr rms4   zEnumMeta._get_mixins_r<c Cs|dd}|r dStdd}ttd}tdur|}ntjd}dD]J}||fD]=}z|j|} WnttfyBt||d} Ynw| d|||fvre|dkrW| |d<dSt| tra| |} | }nq(|durlnq"tj }|tj urxd} nd} |d| fS) Returns the __new__ to be used for creating the enum members. classdict: the class dictionary given to __new__ member_type: the data type whose __new__ will be used by default first_enum: enumeration to check for an overriding __new__ riN)NTTrhrirh)NFTFT) rorrwrrxrrry staticmethodrri) rKrrriN__new__O__new__E__new__rpossibletargetrrrr rnsJ          zEnumMeta._find_new_c Cs|dd}|du}|dur8dD]$}||fD]}t||d}|ddjtjtjfvr-|}nq|dur4nqtj}|tjur@d}nd}|||fS)rriNrFT)rorrirwr) rKrrrirrrrrrrr rn#s.     )NNN)rrrr classmethodrGrirrrrpropertyrrrrrrr\rrrrmr>rnrCrrr:r rDs6        5 -9rDzNGeneric enumeration. Derive from this class to define new enumerations. rcCsxt||ur |j}z ||jvr|j|WSWnty2|jD] }|j|kr/|YSq"Ynwtd||jf)Nz%s is not a valid %s)rjrrvr1rWvaluesr?r)r5rrrrr ri[s      ricCsd|jj|j|jfS)Nz <%s.%s: %r>)r;rr{rTr2rrr r\qsr\cCsd|jj|jfS)Nr)r;rr{r2rrr r]wsr]cCsdd|jjD}gd|S)NcSsg|] }|ddkr|qS)rr&r)rImrrr rP}rz__dir__..)r;rrr,r)r;rx)radded_behaviorrrr r|s rcCs0|jtur t}t|}n|j}|j}|||Sr )rurwrrr^)r format_specr5valrrr r^s   r^r_cCs"t||jur||ur dSdStS)Nrr.)rjr;NotImplementedr1rrotherrrr r`s r`cCtd|jj|jjf)Nzunorderable types: %s() <= %s()r1r;rrrrr rarracCr)Nzunorderable types: %s() < %s()rrrrr rbrrbcCr)Nzunorderable types: %s() >= %s()rrrrr rdrrdcCr)Nzunorderable types: %s() > %s()rrrrr rcrrccCst||jur ||uStSr rjr;rrrrr rerecCst||jur ||uStSr rrrrr rfrrfcCs|jfSr rYr2rrr rSrrScCrr )hashr{r2rrr rgrrgcC|jSr )r{r2rrr r,rHr,cCrr rYr2rrr rrHrrc@seZdZdZdS)rz.Enum where members are also (and must be) intsN)rrrrrrrr rsrcCs^g}|jD]\}}||jkr|||jfq|r-ddd|D}td||f|S)zIClass decorator that ensures only unique members exist in an enumeration.rRcSsg|] \}}d||fqS)z%s -> %sr)rIaliasr,rrr rPrzunique..zduplicate names found in %r: %s)rrlr,r@rsr?) enumeration duplicatesr,rduplicate_namesrrr rs  r)(rsysr__all__float version_infor>r NameErrorrwr r"r-r0r6rkr7rrjrDtemp_enum_dictrir\r]rr^r`rarbrdrcrerfrSrgr,rrrrrrrr s   /T