U
e5d @ s d dl Z d dlmZmZ ddddddd gZd
d Zd d
Zdd Zdd Ze Z
G dd dZ G dd de Z
dZG dd deZG dd dedZG dd deeZdd ZG dd deZG d d deeZd d Zd d Zd d! Zd"d# ZdS )$ N)MappingProxyTypeDynamicClassAttributeEnumMetaEnumIntEnumFlagIntFlagautouniquec C s t | dp t | dp t | dS )z?
Returns True if obj is a descriptor, False otherwise.
__get____set__
__delete__)hasattr)obj r /usr/lib64/python3.8/enum.py_is_descriptor s
r c C sL t | dkoJ| dd | dd ko.dkn oJ| d dkoJ| d dkS )z=
Returns True if a __dunder__ name, False otherwise.
N ___lennamer r r
_is_dunder s &
r c C sL t | dkoJ| d | d ko&dkn oJ| dd dkoJ| dd dkS )z;
Returns True if a _sunder_ name, False otherwise.
r r r r r r r r r
_is_sunder! s r c C s dd }|| _ d| _dS )z,
Make the given class un-picklable.
c S s t d| d S )Nz%r cannot be pickled) TypeErrorselfprotor r r _break_on_call_reduce0 s z6_make_class_unpicklable.._break_on_call_reducez N)
__reduce_ex__
__module__)clsr% r r r _make_class_unpicklable, s r) c @ s e Zd ZdZeZdS )r zP
Instances are replaced with an appropriate value in Enum class suites.
N)__name__r' __qualname____doc__
_auto_nullvaluer r r r r 6 s c s, e Zd ZdZ fddZ fddZ ZS ) _EnumDictz
Track 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.
c s& t g | _g | _g | _d| _d S )NF)super__init__
_member_names _last_values_ignore _auto_calledr# __class__r r r1 D s
z_EnumDict.__init__c sd t |r|dkrtd|dkr<| jr.tdt| d| nV|dkrt|tr`|dd }nt |}|| _
t |t | j @ }|rtd |f nt
|r|d
krd }n|| j krtd | n|| j
krnt|sR|| krtd
|| | f t|tr:|jtkr4| |dt| j | jdd |_d| _|j}| j | | j| t || dS )z
Changes 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__r9 zAttempted to reuse key: %rz%r already defined as: %rr NT)r
ValueErrorr5 r! setattr
isinstancestrreplacesplitlistr4 setr2 r r r r. r- r> r r3 appendr0 __setitem__)r# keyr. alreadyr7 r r rK K sT
z_EnumDict.__setitem__)r* r' r+ r, r1 rK
__classcell__r r r7 r r/ = s r/ c s e Zd ZdZedd Z fddZdd Zd/dddd d
d d Zd
d Z fddZ
dd Z dd Z dd Z
dd Zdd Zedd Zd d Zd d Z fd!d"Zdddd d
d#d$Zd0d%d&Zd'd( Zed)d* Zed+d, Zed-d. Z Z S )1r z
Metaclass for Enum
c C s> | || t }| ||\}}|d k r:t|dd |d< |S )Nr; )_check_for_existing_membersr/ _get_mixins_getattr)metaclsr( bases enum_dict member_type
first_enumr r r __prepare__ s zEnumMeta.__prepare__c s dg d d }|D ]} |d q | ||\}| |\}}} fdd jD }
jD ]
} | = qn dd } t|
ddh@ }
|
rtdd |
d krd
d < t
| || }g |_ i |_
|_d d | D }i |_d
kr2tk r2d}tfdd|D s2t| jD ]*}|
| }t|tsZ|f}n|}tkrn|f}| s||}t|ds||_n6||f| }t|dsƈtkr||_n
| |_|j}||_||_|j| |j
D ]"\} }|j|jkr|} q q|j | ||kr2t ||| ||j
|< z||j|< W n t k
r` Y nX q8dD ]V} | kr|qjt || }t | d }t || d }|d k rj||krjt || | qjt d k r|r||_ t j |_ | d k rt| t!r| "dd# } | |j krt d|S )Nr= c s i | ]}| | qS r r ).0k) classdictr r
s z$EnumMeta.__new__..r9 mro z Invalid enum member name: {0}r? r, zAn enumeration.c S s. h | ]&}|j D ]\}}t|tr|qqS r )__dict__itemsrD r )rX crY vr r r s
z#EnumMeta.__new__..r& )__getnewargs_ex____getnewargs__r&
__reduce__c 3 s | ]}| j kV qd S N)r^ rX m)rU r r s z#EnumMeta.__new__.._value_)__repr____str__
__format__r& r@ z#member order does not match _order_)$
setdefaultrJ poprP
_find_new_r2 rI rB formatjoinr0 __new___member_names_ _member_map_
_member_type_r\ _value2member_map_objectanyr) rD tupler rj _name_ __objclass__r1 r_ rC r! rQ r __new_member__rE rF rG )rR r( rS rZ ignorerL rV rs save_newuse_args enum_membersr r9
invalid_names
enum_classdynamic_attributesmethods member_namer. args enum_membercanonical_member class_method
obj_method enum_methodr7 )rZ rU r rs s
zEnumMeta.__new__c C s dS )z6
classes/types should always be True.
Tr r6 r r r __bool__3 s zEnumMeta.__bool__Nr modulequalnametypestartc C s* |dkr| | |S | j||||||dS )a!
Either 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 )rs _create_)r( r. namesr r r r r r r __call__9 s zEnumMeta.__call__c C s: t |ts$tdt|j| jjf t || o8|j| jkS )N3unsupported operand type(s) for 'in': '%s' and '%s')rD r r! r r+ r8 r{ ru )r( memberr r r __contains__^ s
zEnumMeta.__contains__c s( || j krtd| j t | d S )Nz %s: cannot delete Enum member.)ru AttributeErrorr* r0 __delattr__)r( attrr7 r r r e s
zEnumMeta.__delattr__c C s ddddg| j S )Nr8 r, __members__r' rt r6 r r r __dir__l s
zEnumMeta.__dir__c C s@ t |rt|z | j| W S tk
r: t|dY nX dS )a=
Return 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)r r ru KeyErrorr( r r r r __getattr__r s zEnumMeta.__getattr__c C s
| j | S rf ru r r r r __getitem__ s zEnumMeta.__getitem__c s fdd j D S )z6
Returns members in definition order.
c 3 s | ]} j | V qd S rf r rX r r( r r ri s z$EnumMeta.__iter__..r r r r r __iter__ s zEnumMeta.__iter__c C s
t | jS rf )r rt r r r r __len__ s zEnumMeta.__len__c C s
t | jS )z
Returns 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.
)r ru r r r r r s zEnumMeta.__members__c C s
d| j S )Nz )r* r r r r rk s zEnumMeta.__repr__c s fddt jD S )z>
Returns members in reverse definition order.
c 3 s | ]} j | V qd S rf r r r r r ri s z(EnumMeta.__reversed__..)reversedrt r r r r __reversed__ s zEnumMeta.__reversed__c s0 | j di }||kr tdt || dS )a
Block 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.
ru zCannot reassign members.N)r^ getr r0 __setattr__)r( r r.
member_mapr7 r r r s zEnumMeta.__setattr__c
C s~ | j }|dkr| fn|| f}| | |\} }
|||} t|trR|dd }t|ttfr|rt|d tr|g } }g }
t | D ]8\}}|
||||
dd }|
| | ||f q|D ].}t|tr||| }}n|\}}|| |< q| |||| }|dkrPzt
djd }W n* tttfk
rN } zW 5 d}~X Y nX |dkrdt| n||_|dk rz||_|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 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@ r r r* )r8 rP rW rD rE rF rG rz rH enumerater; rJ rs sys _getframe f_globalsr rB r r) r' r+ )r(
class_namer r r r r rR rS r rV rZ original_names last_valuescountr r. itemr member_valuer excr r r r s<
zEnumMeta._create_c s t tj| }|r t |}n|} fdd| D }z|jdd d W n$ tk
rn |jdd d Y nX | |||d} t| _|| j | ||< | S )z[
Create a new Enum subclass that replaces a collection of global constants
c s g | ]\}} |r||fqS r r )rX r r. filterr r
s z&EnumMeta._convert_..c S s | d | d fS )Nr r r tr r r z$EnumMeta._convert_..)rL c S s | d S Nr r r r r r r r )r )
varsr modulesr_ sortr! _reduce_ex_by_namer& updater )r( r r r sourcemodule_globalsmembersr r r _convert_ s
zEnumMeta._convert_c O s$ dd l }|jdtdd | j||S )Nr zI_convert is deprecated and will be removed in 3.9, use _convert_ instead.r )
stacklevel)warningswarnDeprecationWarningr )r( r kwargsr r r r _convert s zEnumMeta._convertc C s< |D ]2}|j D ]&}t|tr|jrtd| |jf qqd S )Nz %s: cannot extend enumeration %r)__mro__
issubclassr rt r! r* )r rS chainbaser r r rO
s
z$EnumMeta._check_for_existing_membersc sT |s t tfS fdd}|d }t|ts2td||p._find_data_typer zZnew enumerations should be created as `EnumName([mixin_type, ...] [data_type,] enum_type)`zCannot extend enumerations)rx r r r! rt )r rS r rV rU r r r rP s
zEnumMeta._get_mixins_c C s | 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__
rs N)r} rs FT)r rQ rs rx r ) rZ rU rV rs r methodpossibletargetr r r r rp C s*
zEnumMeta._find_new_)N)N) r* r' r+ r, classmethodrW rs r r r r r r r r r propertyr rk r r r r r staticmethodrO rP rp rN r r r7 r r s8
%
5
!
.c @ st e Zd ZdZdd Zdd Zedd Zdd Zd
d Z d d
Z
dd Z dd Z dd Z
edd Zedd ZdS )r zV
Generic enumeration.
Derive from this class to define new enumerations.
c
C s t || kr|S z | j| W S tk
r0 Y n: tk
rh | j D ]}|j|krH| Y S qHY nX zd }| |}W n* tk
r } z |}d }W 5 d }~X Y nX zdt || r|W TS t
d|| j f }|d kr|d kr|n |d kr td| j |f }||_ |W 5 d }d }X d S )N%r is not a valid %szDerror in %s._missing_: returned %r instead of None or a valid member)
r rw r r! ru valuesrj r< ExceptionrD rB r* __context__)r( r. r r resulteve_excr r r rs w s@
z Enum.__new__c C s6 t |D ](}z|d W S tk
r. Y qX q|S )
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
r N)r r! )r r r r
last_valuer r r r; s zEnum._generate_next_value_c C s d S rf r )r( r. r r r r< s zEnum._missing_c C s d| j j| j| jf S )N <%s.%s: %r>)r8 r* r{ rj r6 r r r rk s
z
Enum.__repr__c C s d| j j| jf S )N%s.%s)r8 r* r{ r6 r r r rl s z Enum.__str__c s6 fdd j D dd jD }dddg| S )z<
Returns all members and all public methods
c s2 g | ]*}|j D ] }|d dkr| jkr|qqS r r )r^ ru )rX r( rh r6 r r r s
z Enum.__dir__..c S s g | ]}|d dkr|qS r r rg r r r r s r8 r, r' )r8 r\ r^ )r# added_behaviorr r6 r r s
z Enum.__dir__c C sJ t | jtjtjfk}| jtks$|r2t}t| }n | j}| j}|||S )z\
Returns format using actual value type unless __str__ has been overridden.
) r rl r r rv rx rE rj rm )r# format_specstr_overriddenr( valr r r rm s
zEnum.__format__c C s
t | jS rf )hashr{ r6 r r r __hash__ s z
Enum.__hash__c C s | j | jffS rf r8 rj r" r r r r& s zEnum.__reduce_ex__c C s | j S )z The name of the Enum member.)r{ r6 r r r r s z Enum.namec C s | j S )z The value of the Enum member.rj r6 r r r r. s z
Enum.valueN)r* r' r+ r, rs r; r r< rk rl r rm r r& r r r. r r r r r q s -
) metaclassc @ s e Zd ZdZdS )r z.Enum where members are also (and must be) intsN)r* r' r+ r, r r r r r s c C s | j S rf r r" r r r r s r c @ sp e Zd ZdZdd Zedd Zedd Zdd Zd
d Z d d
Z
dd Z dd Z dd Z
dd Zdd ZdS )r z
Support for flags
c C sd |s|dk r|S dS t |D ]:}zt|}W qXW q tk
rT td| dY q X q d|d S )r Nr zInvalid Flag value: %rr )r _high_bitr r! )r r r r r high_bitr r r r; s
zFlag._generate_next_value_c C s. |}|dk r| }| |}|dk r*| }|S )V
Returns member (possibly creating it) if one can be found for value.
r )r: )r( r. original_valuepossible_memberr r r r< s
zFlag._missing_c C sb | j |d}|dkr^t| |\}}|r:td|| jf t| }d|_||_| j ||}|S )L
Create a composite member iff value contains only members.
Nr )
rw r
_decomposerB r* rx rs r{ rj rn )r( r.
pseudo_memberr extra_flagsr r r r: # s
zFlag._create_pseudo_member_c C s8 t || js&tdt|j| jjf |j| j@ |jkS )zP
Returns True if self has at least the same flags set as other.
r )rD r8 r! r r+ rj r# otherr r r r 7 s zFlag.__contains__c C sV | j }| jd k r$d|j| j| jf S t|| j\}}d|jddd |D | jf S )Nr |c S s g | ]}t |jp|jqS r rE r{ rj rg r r r r H s z!Flag.__repr__..)r8 r{ r* rj r rr r# r( r uncoveredr r r rk A s
z
Flag.__repr__c C s | j }| jd k r d|j| jf S t|| j\}}t|dkr^|d jd kr^d|j|d jf S d|jddd |D f S d S )Nr r r z%s.%rr c S s g | ]}t |jp|jqS r r rg r r r r V s z Flag.__str__..)r8 r{ r* r rj r rr r r r r rl L s
z Flag.__str__c C s
t | jS rf )boolrj r6 r r r r Y s z
Flag.__bool__c C s" t || jstS | | j|jB S rf rD r8 NotImplementedrj r r r r __or__\ s z Flag.__or__c C s" t || jstS | | j|j@ S rf r r r r r __and__a s z Flag.__and__c C s" t || jstS | | j|jA S rf r r r r r __xor__f s z Flag.__xor__c C sN t | j| j\}}| d}| jD ] }||kr"|j| j@ s"||B }q"| |S r )r r8 rj )r# r r invertedrh r r r
__invert__k s
zFlag.__invert__N)r* r' r+ r, r; r r< r: r rk rl r r r r r r r r r r s
c @ sT e Zd ZdZedd Zedd Zdd Zdd Zd
d Z eZ
eZ e Z d d
Z
dS )r z)
Support for integer-based Flags
c C s* t |ts td|| jf | |}|S )r r )rD intrB r* r: )r( r.
new_memberr r r r< y s
zIntFlag._missing_c C s | j |d}|dkr|g}t| |\}}|rtt|}d| }|| j krZ||krZ|| || krjd}q*||N }q*t|D ]*}t| |}d|_||_ | j
||}q||S )r Nr r ) rw r r r rJ r r rs r{ rj rn )r( r. r need_to_creater r bit
flag_valuer r r r: s(
z IntFlag._create_pseudo_member_c C s0 t || jtfstS | | j| |jB }|S rf rD r8 r r rj )r# r r r r r r s zIntFlag.__or__c C s, t || jtfstS | | j| |j@ S rf r r r r r r s zIntFlag.__and__c C s, t || jtfstS | | j| |jA S rf r r r r r r s zIntFlag.__xor__c C s | | j }|S rf r )r# r r r r r s zIntFlag.__invert__N)r* r' r+ r, r r< r: r r r __ror____rand____rxor__r r r r r r t s
c C s | d S )zJ
returns index of highest bit, or -1 if value is zero or negative
r )
bit_lengthr. r r r r s r c C s^ g }| j D ]"\}}||jkr|||jf q|rZddd |D }td| |f | S )zI
Class decorator for enumerations ensuring unique member values.
z, c S s g | ]\}}d ||f qS )z%s -> %sr )rX aliasr r r r r s zunique..z duplicate values found in %r: %s)r r_ r rJ rr rB ) enumeration
duplicatesr r
alias_detailsr r r r
s
c C s |}|dk }|r*dd t | j D }ndd t | j D }g }|D ],\}}|rJ||@ |krJ|| || M }qJ|s|| jkr|| j| |jdd dd t|d kr|d j|kr|d ||fS )
z-
Extract all members from the value.
r c S s" g | ]\}}|j d k r||fqS rf r rX ra rh r r r r s
z _decompose..c S s* g | ]"\}}|j d k s t|r||fqS rf )r
_power_of_twor r r r r s
c S s | j S rf r )rh r r r r r z _decompose..T)rL reverser )rH rw r_ rJ r r r. ro )flagr. not_coverednegativeflags_to_checkr r r r r r r s(
r c C s | dk r dS | dt | kS )Nr Fr )r r r r r r s r )r typesr r __all__r r r r) rx r- r dictr/ r r r r r r r r r r
r r r r r r s>
L i vI%