U `B@sddlZddlmZmZddddddd gZd d Zd d ZddZddZe Z GdddZ Gddde Z dZGdddeZGdddedZGdddeeZddZGdddeZGdddeeZddZdd Zd d!Zd"d#ZdS)$N)MappingProxyTypeDynamicClassAttributeEnumMetaEnumIntEnumFlagIntFlagautouniquecCst|dpt|dpt|dS)z5Returns True if obj is a descriptor, False otherwise.__get____set__ __delete__)hasattr)objr/usr/lib/python3.8/enum.py_is_descriptor s  rcCsLt|dkoJ|dd|ddko.dknoJ|ddkoJ|ddkS)z3Returns True if a __dunder__ name, False otherwise.N___lennamerrr _is_dunders &  rcCsLt|dkoJ|d|dko&dknoJ|dddkoJ|dddkS)z1Returns True if a _sunder_ name, False otherwise.rrrrrrrrr _is_sunders r cCsdd}||_d|_dS)z"Make the given class un-picklable.cSstd|dS)Nz%r cannot be pickled) TypeErrorselfprotorrr_break_on_call_reduce&sz6_make_class_unpicklable.._break_on_call_reducez N) __reduce_ex__ __module__)clsr%rrr_make_class_unpicklable$sr)c@seZdZdZeZdS)r zP Instances are replaced with an appropriate value in Enum class suites. N)__name__r' __qualname____doc__ _auto_nullvaluerrrrr ,scs,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. cs&tg|_g|_g|_d|_dS)NF)super__init__ _member_names _last_values_ignore _auto_calledr# __class__rrr1:s  z_EnumDict.__init__csdt|r|dkrtd|dkr<|jr.tdt|d|nV|dkrt|tr`|dd}nt |}||_ t |t |j @}|rtd |fnt |r|d krd }n||j krtd |n||j krnt|sR||krtd |||ft|tr:d|_|jtkr4||dt|j |jdd|_|j}|j ||j|t||dS)zChanges anything not dundered or not a descriptor. If an enum member name is used twice, an error is raised; duplicate values are not checked for. Single underscore (sunder) names are reserved. )_order__create_pseudo_member__generate_next_value_ _missing__ignore_z(_names_ are reserved for future Enum user;z4_generate_next_value_ must be defined before members_generate_next_valuer=, z-_ignore_ cannot specify already set names: %r __order__r9zAttempted to reuse key: %rz%r already defined as: %rTrN)r ValueErrorr5r!setattr isinstancestrreplacesplitlistr4setr2rrr r.r-r>rr3appendr0 __setitem__)r#keyr.alreadyr7rrrKAsB       "  z_EnumDict.__setitem__)r*r'r+r,r1rK __classcell__rrr7rr/3s r/cseZdZdZeddZfddZddZd-dddd d d d Zd dZ fddZ ddZ ddZ ddZ ddZddZeddZddZdd Zfd!d"Zdddd d d#d$Zd.d%d&Zd'd(Zed)d*Zed+d,ZZS)/rzMetaclass for EnumcCs0t}||\}}|dk r,t|dd|d<|S)Nr;)r/ _get_mixins_getattr)metaclsr(bases enum_dict member_type first_enumrrr __prepare__|s zEnumMeta.__prepare__c sdgdd}|D]}|dq||\}||\}}} fddjD} jD] } | =qldd} t| ddh@} | rtdd | d krd d <t |||}g|_ i|_ |_d d |D}i|_d kr0tk r0d}tfdd|Ds0t|jD]*}| |}t|tsX|f}n|}tkrl|f}| s||}t|ds||_n6||f|}t|dsĈtkr||_n ||_|j}||_||_|j||j D]"\} }|j|jkr|}qq|j |||kr0t|||||j |<z||j|<Wntk r^YnXq6dD]H} t|| }t| d}t|| d}|dk rh||krht|| |qhtdk r|r||_ tj |_ | dk r t| t!r| "dd#} | |j kr td|S)Nr=csi|]}||qSrr).0k) classdictrr sz$EnumMeta.__new__..r9mrozInvalid enum member name: {0}r?r,zAn enumeration.cSs.h|]&}|jD]\}}t|tr|qqSr)__dict__itemsrDr)rWcrXvrrr s   z#EnumMeta.__new__..r&)__getnewargs_ex____getnewargs__r& __reduce__c3s|]}|jkVqdSN)r]rWm)rTrr sz#EnumMeta.__new__.._value_)__repr____str__ __format__r&r@z#member order does not match _order_)$ setdefaultrJpoprO _find_new_r2rIrBformatjoinr0__new___member_names_ _member_map_ _member_type_r[_value2member_map_objectanyr)rDtuplerri_name_ __objclass__r1r^rCr!rPr__new_member__rErFrG)rQr(rRrYignorerLrUrrsave_newuse_args enum_membersrr9 invalid_names enum_classdynamic_attributesmethods member_namer.args enum_membercanonical_member class_method obj_method enum_methodr7)rYrTrrrs                          zEnumMeta.__new__cCsdS)z6 classes/types should always be True. Trr6rrr__bool__szEnumMeta.__bool__NrmodulequalnametypestartcCs*|dkr|||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: `value` will be the name of the new class. `names` should be either a string of white-space/comma delimited names (values will start at `start`), or an iterator/mapping of name, value pairs. `module` should be set to the module this class is being created in; if it is not set, an attempt to find that module will be made, but if it fails the class will not be picklable. `qualname` should be set to the actual location this class can be found at in its module; by default it is set to the global scope. If this is not correct, unpickling will fail in some circumstances. `type`, if set, will be mixed in as the first base class. Nr)rr_create_)r(r.namesrrrrrrr__call__s zEnumMeta.__call__cCs:t|ts$tdt|j|jjft||o8|j|jkSNz3unsupported operand type(s) for 'in': '%s' and '%s')rDrr!rr+r8rzrt)r(memberrrr __contains__9s zEnumMeta.__contains__cs(||jkrtd|jt|dS)Nz%s: cannot delete Enum member.)rtAttributeErrorr*r0 __delattr__)r(attrr7rrr@s  zEnumMeta.__delattr__cCsddddg|jS)Nr8r, __members__r'rsr6rrr__dir__Hs zEnumMeta.__dir__cCs@t|rt|z |j|WStk r:t|dYnXdS)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. N)rrrtKeyErrorr(rrrr __getattr__Ls  zEnumMeta.__getattr__cCs |j|Srertrrrr __getitem__\szEnumMeta.__getitem__csfddjDS)Nc3s|]}j|VqdSrerrWrr(rrrh`sz$EnumMeta.__iter__..rrrrr__iter___szEnumMeta.__iter__cCs t|jSre)rrsrrrr__len__bszEnumMeta.__len__cCs t|jS)zReturns a mapping of member name->value. This mapping lists all enum members, including aliases. Note that this is a read-only view of the internal mapping. )rrtrrrrreszEnumMeta.__members__cCs d|jS)Nz )r*rrrrrjoszEnumMeta.__repr__csfddtjDS)Nc3s|]}j|VqdSrerrrrrrhssz(EnumMeta.__reversed__..)reversedrsrrrr __reversed__rszEnumMeta.__reversed__cs0|jdi}||krtdt||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. rtzCannot reassign members.N)r]getrr0 __setattr__)r(rr. member_mapr7rrruszEnumMeta.__setattr__c Cs||j}|dkr|fn||f}||\} } |||} t|trP|dd}t|ttfr|rt|dtr|g} }g} t | D]8\}}| |||| dd}| || ||fq|D].}t|tr|||}}n|\}}|| |<q| |||| }|dkrNzt djd}Wn*tttfk rL}zW5d}~XYnX|dkrbt|n||_|dk rx||_|S)aConvenience method to create a new Enum class. `names` can be: * A string containing member names, separated either with spaces or commas. Values are incremented by 1 from `start`. * An iterable of member names. Values are incremented by 1 from `start`. * An iterable of (member name, value) pairs. * A mapping of member name -> value pairs. Nr?r@rrr*)r8rOrVrDrErFrGryrH enumerater;rJrrsys _getframe f_globalsrrBrr)r'r+)r( class_namerrrrrrQrRrrUrYoriginal_names last_valuescountrr.itemr member_valuerexcrrrrs<            zEnumMeta._create_csttj|}|rt|}n|}fdd|D}z|jdddWn$tk rn|jdddYnX||||d}t|_||j |||<|S)z[ Create a new Enum subclass that replaces a collection of global constants cs g|]\}}|r||fqSrr)rWrr.filterrr sz&EnumMeta._convert_..cSs|d|dfS)Nrrrtrrrz$EnumMeta._convert_..)rLcSs|dSNrrrrrrrr)r) varsrmodulesr^sortr!_reduce_ex_by_namer&updater)r(rrrsourcemodule_globalsmembersrrr _convert_s    zEnumMeta._convert_cOs$ddl}|jdtdd|j||S)NrzI_convert is deprecated and will be removed in 3.9, use _convert_ instead.r) stacklevel)warningswarnDeprecationWarningr)r(rkwargsrrrr_converts zEnumMeta._convertcCsP|s ttfSdd}|d}t|ts.td||p8t}|jrHtd||fS)zReturns the type for creating enum members, and the first inherited enum class. bases: the tuple of bases that was given to __new__ cSsH|D]>}|jD]2}|tkrqqd|jkrt|tr4q|SqqdS)Nrr)__mro__rwr] issubclassr)rRchainbaserrr_find_data_types   z.EnumMeta._get_mixins_.._find_data_typerzZnew enumerations should be created as `EnumName([mixin_type, ...] [data_type,] enum_type)`zCannot extend enumerations)rwrrr!rs)rRrrUrTrrrrOs   zEnumMeta._get_mixins_c Cs|dd}|dk }|dkrpdD]H}||fD].}t||d}|ddjtjtjhkr,|}q\q,|dk r qpq tj}|tjkrd}nd}|||fS)a 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__ rrN)r|rrFT)rrPrrrwr) rYrTrUrrr~methodpossibletargetrrrrros*    zEnumMeta._find_new_)N)N)r*r'r+r, classmethodrVrrrrrrrrrrrpropertyrrjrrrrr staticmethodrOrorNrrr7rrzs4    5 ! c@steZdZdZddZddZeddZdd Zd d Z d d Z ddZ ddZ ddZ eddZeddZdS)rzRGeneric enumeration. Derive from this class to define new enumerations. c Cst||kr|Sz |j|WStk r0Yn:tk rh|jD]}|j|krH|YSqHYnXzd}||}Wn*tk r}z |}d}W5d}~XYnXt ||r|St d||j f}|dkr|dkr|n|dkrtd|j |f}||_ |dS)N%r is not a valid %szDerror in %s._missing_: returned %r instead of None or a valid member) rrvrr!rtvaluesrir< ExceptionrDrBr* __context__)r(r.rrresulteve_excrrrrr2s:    z Enum.__new__c Cs6t|D](}z|dWStk r.YqXq|S)Nr)rr!)rrrr last_valuerrrr;Zs  zEnum._generate_next_value_cCstd||jfdSNr)rBr*)r(r.rrrr<cszEnum._missing_cCsd|jj|j|jfS)N <%s.%s: %r>)r8r*rzrir6rrrrjgs z Enum.__repr__cCsd|jj|jfS)N%s.%s)r8r*rzr6rrrrkksz Enum.__str__cs&fddjD}dddg|S)Ncs2g|]*}|jD]}|ddkr|jkr|qqS)rr)r]rt)rWr(rgr6rrros  z Enum.__dir__..r8r,r')r8r[)r#added_behaviorrr6rrns z Enum.__dir__cCs0|jtkrt}t|}n |j}|j}|||Sre)rurwrErirl)r# format_specr(valrrrrlws   zEnum.__format__cCs t|jSre)hashrzr6rrr__hash__sz Enum.__hash__cCs|j|jffSrer8rir"rrrr&szEnum.__reduce_ex__cCs|jS)zThe name of the Enum member.)rzr6rrrrsz Enum.namecCs|jS)zThe value of the Enum member.rir6rrrr.sz Enum.valueN)r*r'r+r,rrr;rr<rjrkrrlrr&rrr.rrrrr,s(     ) metaclassc@seZdZdZdS)rz.Enum where members are also (and must be) intsN)r*r'r+r,rrrrrscCs|jSrerr"rrrrsrc@speZdZdZddZeddZeddZdd Zd d Z d d Z ddZ ddZ ddZ ddZddZdS)rzSupport for flagsc Csd|s|dk r|SdSt|D]:}zt|}WqXWqtk rTtd|dYqXqd|dS)z Generate the next value when not given. name: the name of the member start: the initial start value or None count: the number of existing members last_value: the last value assigned or None NrzInvalid Flag value: %rr)r _high_bitrr!)rrrrrhigh_bitrrrr;s   zFlag._generate_next_value_cCs.|}|dkr|}||}|dkr*|}|Sr)r:)r(r.original_valuepossible_memberrrrr<s zFlag._missing_cCsb|j|d}|dkr^t||\}}|r:td||jft|}d|_||_|j ||}|S)zL Create a composite member iff value contains only members. Nr) rvr _decomposerBr*rwrrrzrirm)r(r. pseudo_memberr extra_flagsrrrr:s zFlag._create_pseudo_member_cCs8t||js&tdt|j|jjf|j|j@|jkSr)rDr8r!rr+rir#otherrrrrs zFlag.__contains__cCsV|j}|jdk r$d|j|j|jfSt||j\}}d|jddd|D|jfS)Nr|cSsg|]}t|jp|jqSrrErzrirfrrrrsz!Flag.__repr__..)r8rzr*rirrqr#r(r uncoveredrrrrjs z Flag.__repr__cCs|j}|jdk r d|j|jfSt||j\}}t|dkr^|djdkr^d|j|djfSd|jddd|DfSdS)Nrrrz%s.%rrcSsg|]}t|jp|jqSrrrfrrrrsz Flag.__str__..)r8rzr*rrirrqrrrrrks z Flag.__str__cCs t|jSre)boolrir6rrrrsz Flag.__bool__cCs"t||jstS||j|jBSrerDr8NotImplementedrirrrr__or__s z Flag.__or__cCs"t||jstS||j|j@Srerrrrr__and__s z Flag.__and__cCs"t||jstS||j|jASrerrrrr__xor__s z Flag.__xor__cCsNt|j|j\}}|d}|jD] }||kr"|j|j@s"||B}q"||Sr)rr8ri)r#rrinvertedrgrrr __invert__ s    zFlag.__invert__N)r*r'r+r,r;rr<r:rrjrkrrrrrrrrrrs    c@sTeZdZdZeddZeddZddZdd Zd d Z eZ eZ e Z d d Z dS)rzSupport for integer-based FlagscCs*t|tstd||jf||}|Sr)rDintrBr*r:)r(r. new_memberrrrr<s  zIntFlag._missing_cCs|j|d}|dkr|g}t||\}}|rtt|}d|}||jkrZ||krZ|||| krjd}q*||N}q*t|D]*}t||}d|_||_ |j ||}q||S)Nrr) rvrrrrJrrrrrzrirm)r(r.rneed_to_createrrbit flag_valuerrrr:s(      zIntFlag._create_pseudo_member_cCs0t||jtfstS||j||jB}|SrerDr8rrri)r#rrrrrr;szIntFlag.__or__cCs,t||jtfstS||j||j@SrerrrrrrAszIntFlag.__and__cCs,t||jtfstS||j||jASrerrrrrrFszIntFlag.__xor__cCs||j}|Srer)r#rrrrrOszIntFlag.__invert__N)r*r'r+r,rr<r:rrr__ror____rand____rxor__rrrrrrs  cCs |dS)z@returns index of highest bit, or -1 if value is zero or negativer) bit_lengthr.rrrrTsrcCs^g}|jD]"\}}||jkr|||jfq|rZddd|D}td||f|S)z?Class decorator for enumerations ensuring unique member values.z, cSsg|]\}}d||fqS)z%s -> %sr)rWaliasrrrrr`szunique..z duplicate values found in %r: %s)rr^rrJrqrB) enumeration duplicatesrr alias_detailsrrrr Xs  cCs|}|dk}|r*ddt|jD}nddt|jD}g}|D],\}}|rJ||@|krJ||||M}qJ|s||jkr||j||jddddt|d kr|dj|kr|d||fS) z#Extract all members from the value.rcSs"g|]\}}|jdk r||fqSrerrWr`rgrrrros z_decompose..cSs*g|]"\}}|jdk st|r||fqSre)r _power_of_tworrrrrvs cSs|jSrer)rgrrrrrz_decompose..T)rLreverser)rHrvr^rJrrr.rn)flagr. not_coverednegativeflags_to_checkrrrrrrres(      rcCs|dkr dS|dt|kS)NrFr)rr rrrrsr)rtypesrr__all__rrr r)rwr-r dictr/rrrrrrrrrr rrrrrrs<D5rnA #