o
0i @ s d dl Z d dlmZmZ g dZdd Zdd Zdd Zd
d Zd d
Z e
Z G dd dZ G dd de
ZdZG dd deZG dd dedZG dd deeZdd ZG dd d eZG d d d eeZd d Zd!d" Zd#d$ ZdS )% N)MappingProxyTypeDynamicClassAttribute)EnumMetaEnumIntEnumFlagIntFlagautouniquec C s t | dpt | dpt | dS )z?
Returns True if obj is a descriptor, False otherwise.
__get____set__
__delete__)hasattr)obj r +/opt/alt/python310/lib64/python3.10/enum.py_is_descriptor s
r c C sL t | dko%| dd | dd kodkn o%| d dko%| d dkS )z=
Returns True if a __dunder__ name, False otherwise.
N ___lennamer r r
_is_dunder s &
r c C sL t | dko%| d | d kodkn o%| dd dko%| 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^ d| f }t |}t ||kr-||r-|||d dgkr-|d dks+|d dkr-dS dS )Nz_%s__r r r r TF)r
startswith)cls_namer patternpat_lenr r r _is_private, s
r% c C s dd }|| _ d| _dS )z,
Make the given class un-picklable.
c S s t d| )Nz%r cannot be pickled) TypeErrorselfprotor r r _break_on_call_reduce> 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 D s r 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 r6 R s
z_EnumDict.__init__c s t | j|rddl}|jd|f tdd t|r_|dvr!td|dkr3| jr,td t | d
| n|d kr^t
|t rE| d d
}nt|}|| _t|t| j@ }|r^td|f n[t|rj|dkrid}nP|| jv rutd| || jv r{n?t|s|| v rtd|| | f t
|tr|jtkr| |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.
r Nz@private variables, such as %r, will be normal attributes in 3.11r
stacklevel)_order__create_pseudo_member__generate_next_value_ _missing__ignore_z(_names_ are reserved for future Enum userB z4_generate_next_value_ must be defined before members_generate_next_valuerD , z-_ignore_ cannot specify already set names: %r __order__r@ zAttempted to reuse key: %rz%r already defined as: %rr T) r% _cls_namewarningswarnDeprecationWarningr
ValueErrorr: r& setattr
isinstancestrreplacesplitlistr9 setr7 r r r r3 r2 rE r r8 appendr5 __setitem__)r( keyr3 rJ alreadyr< r r rV Y sj
z_EnumDict.__setitem__)r/ r, r0 r1 r6 rV
__classcell__r r r< r r4 K s r4 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$Zd-d%d&Zed'd( Zed)d* Zed+d, Z ZS ).r z
Metaclass for Enum
c K sD | || t }||_| ||\}}|d ur t|dd |d< |S )NrB )_check_for_existing_membersr4 rI _get_mixins_getattr)metaclsr- baseskwds enum_dict member_type
first_enumr r r __prepare__ s zEnumMeta.__prepare__c s dg d d }|D ]}|d q| ||\}| |\}} }
fddjD } jD ]} | = q7dd }
t| ddh@ }|rWtdd |d vr_d
d < t
j | ||fi |}g |_ i |_
|_d d | D }i |_d
vrՈturd}tfdd|D sdv rt| n6d }|D ]+}|jD ] tu rqt fdd|D rd} n
d jv rd} nq|d ur nq|rt| jD ]}| | }t|ts|f}n|}tu r|f}|
s||}t|ds||_n ||g|R }t|dstu r||_n| |_|j}||_||_|j | |j
D ]\} }|j|jkr=|} nq.|j | ||vrPt ||| ||j
|< z||j|< W q t yf Y qw dD ]+} | v rrqit || }t | d }t || d }|d ur||u rt || | qit!d ur| r||_"t!j |_ |
d urt|
t#r|
$dd% }
|
|j krt d|S )NrD c s i | ]}| | qS r r ).0k) classdictr r
s z$EnumMeta.__new__..r@ mro z Invalid enum member name: {0}rF r1 zAn enumeration.c S s. h | ]}|j D ] \}}t|tr |q qS r )__dict__itemsrO r )rd cre vr r r s z#EnumMeta.__new__..r+ )__getnewargs_ex____getnewargs__r+
__reduce__c 3 | ]}| j v V qd S Nrj rd m)ra r r z#EnumMeta.__new__..__new__c 3 rr rs rt ru )baser r rw rx FT_value_)__repr____str__
__format__r+ rG z#member order does not match _order_)&
setdefaultrU popr[
_find_new_r7 rT rM formatjoinr5 ry _member_names_ _member_map_
_member_type_rh _value2member_map_objectanyr. __mro__rj rO tupler r{ _name_ __objclass__r6 rk rN r& r\ r __new_member__rP rQ rR ) r] r- r^ rf r_ ignorerW rb ry save_newuse_args enum_membersr r@
invalid_names
enum_classdynamic_attributesmethodssabotagechain member_namer3 args enum_membercanonical_member class_method
obj_method enum_methodr< )rz rf ra r ry s
zEnumMeta.__new__c C s dS )z6
classes/types should always be True.
Tr r; r r r __bool__a s zEnumMeta.__bool__Nr modulequalnametypestartc C s* |du r
| | |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 )ry _create_)r- r3 namesr r r r r r r __call__g s zEnumMeta.__call__c C sR t |ts dd l}|jdtdd tdt|j| jjf t || o(|j | j
v S )Nr zpin 3.12 __contains__ will no longer raise TypeError, but will return True if
obj is a member or a member's valuer r> 3unsupported operand type(s) for 'in': '%s' and '%s') rO r rJ rK rL r& r r0 r= r r )r- r rJ r r r __contains__ s
zEnumMeta.__contains__c s( || j v r td| j t | d S )Nz %s: cannot delete Enum member.)r AttributeErrorr/ r5 __delattr__)r- attrr< r r r s
zEnumMeta.__delattr__c C s g d| j S )N)r= r1 __members__r, r r; r r r __dir__ s zEnumMeta.__dir__c C s6 t |rt|z| j| W S ty t|dw )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 r KeyErrorr- r r r r __getattr__ s
zEnumMeta.__getattr__c C s
| j | S rs r r r r r __getitem__
zEnumMeta.__getitem__c s fdd j D S )z6
Returns members in definition order.
c 3 | ]} j | V qd S rs r rd r r- r r rw rx z$EnumMeta.__iter__..r r r r r __iter__ s zEnumMeta.__iter__c C
t | jS rs )r r r r r r __len__ r zEnumMeta.__len__c C r )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 r r r r r r s
zEnumMeta.__members__c C s
d| j S )Nz )r/ r r r r r| r zEnumMeta.__repr__c s fddt jD S )z>
Returns members in reverse definition order.
c 3 r rs r r r r r rw rx z(EnumMeta.__reversed__..)reversedr r r r r __reversed__ s zEnumMeta.__reversed__c s0 | j di }||v rtdt || 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.
r zCannot reassign members.N)rj getr r5 __setattr__)r- r r3
member_mapr< r r r s zEnumMeta.__setattr__c
C sf | j }|du r
| fn|| f}| | |\} }
|||} t|tr)|dd }t|ttfra|rat|d tra|g } }g }
t | D ] \}}|
||||
dd }|
| | ||f qD|D ]}t|trr||| }}n|\}}|| |< qc| |||| }|du rz
t
djd }W n tttfy Y nw |du rt| n||_|dur||_|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.
NrF rG r r r/ )r= r[ rc rO rP rQ rR r rS enumeraterB rU ry sys _getframe f_globalsr rM r r. r, r0 )r-
class_namer r r r r r] r^ r rb rf original_names last_valuescountr r3 itemr member_valuer r r r r s>
zEnumMeta._create_c s t tj| }|rt |}n|} fdd| D }z
|jdd d W n ty6 |jdd d Y nw | |||d} t| _|| j | ||< | S )z[
Create a new Enum subclass that replaces a collection of global constants
c s g | ] \}} |r||fqS r r )rd r r3 filterr r
) s z&EnumMeta._convert_..c S s | d | d fS )Nr r r tr r r / s z$EnumMeta._convert_..)rW c S s | d S Nr r r r r r r 2 s )r )
varsr modulesrk sortr& _reduce_ex_by_namer+ updater )r- r r r sourcemodule_globalsmembersr r r _convert_ s"
zEnumMeta._convert_c C s< |D ]}|j D ]}t|tr|jrtd| |jf qqd S )Nz %s: cannot extend enumeration %r)r
issubclassr r r& r/ )r r^ r rz r r r rZ 9 s
z$EnumMeta._check_for_existing_membersc sT |st tfS fdd}|d }t|tstd||p t }|jr&td||fS )z
Returns the type for creating enum members, and the first inherited
enum class.
bases: the tuple of bases that was given to __new__
c s t }| D ]:}d }|jD ]2}|tu rq t|tr&|jtur%||j nq d|jv r:t|tr1q ||p6| n|p=|}q qt|dkrNt d |f |rT|
S d S )Nry r z%r: too many data types: %r) rT r r r r r addrj r r& r )r^
data_typesr candidaterz r r r _find_data_typeN s.
z.EnumMeta._get_mixins_.._find_data_typer zZnew enumerations should be created as `EnumName([mixin_type, ...] [data_type,] enum_type)`zCannot extend enumerations)r r r r& r )r r^ r rb ra r r r r[ C s
zEnumMeta._get_mixins_c C s | dd}|du}|du r8dD ]$}||fD ]}t||d}|ddjtjtjhvr-|} nq|dur4 nqtj}|tju r@d}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__
ry N)r ry FT)r r\ ry r r ) rf ra rb ry r methodpossibletargetr r r r r r s.
zEnumMeta._find_new_rs ) r/ r, r0 r1 classmethodrc ry r r r r r r r r r propertyr r| r r r r staticmethodrZ r[ r rY r r r< r r s8
.%
5!
.r 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 || u r|S z| j| W S ty Y n ty2 | j D ]
}|j|kr/| Y S q"Y nw z d }| |}W n tyR } z
|}d }W Y d }~nd }~ww z6t || r`|W d }d }S t
d|| j f }|d u rs|d u rs||d u rtd| j |f }t |t
s||_
|d }d }w )N%r is not a valid %szDerror in %s._missing_: returned %r instead of None or a valid member)r r r r& r valuesr{ rC ExceptionrO rM r0 r/ __context__)r- r3 memberexcresulteve_excr r r ry sP
z Enum.__new__c C s2 t |D ]}z|d W S ty Y qw |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 )r r& )r r r r
last_valuer r r rB s zEnum._generate_next_value_c C s d S rs r )r- r3 r r r rC s zEnum._missing_c C s d| j j| j| jf S )N <%s.%s: %r>)r= r/ r r{ r; r r r r| s z
Enum.__repr__c C s d| j j| jf S )N%s.%s)r= r/ r r; r r r r} s z Enum.__str__c s4 fdd j D dd jD }g d| S )z<
Returns all members and all public methods
c s2 g | ]}|j D ]}|d dkr| jvr|qqS r r )rj r )rd r- rv r; r r r s z Enum.__dir__..c S s g | ]
}|d dkr|qS r r ru r r r r )r= r1 r, )r= rh rj )r( added_behaviorr r; r r s
z Enum.__dir__c C sJ t | jtjtjfv}| jtu s|rt}t| }n| j}| j}|||S )z\
Returns format using actual value type unless __str__ has been overridden.
) r r} r r r r rP r{ r~ )r( format_specstr_overriddenr- valr r r r~ s
zEnum.__format__c C r rs )hashr r; r r r __hash__ r z
Enum.__hash__c C s | j | jffS rs r= r{ r' r r r r+ s zEnum.__reduce_ex__c C | j S )z The name of the Enum member.)r r; r r r r z Enum.namec C r )z The value of the Enum member.r{ r; r r r r3 ! r z
Enum.valueN)r/ r, r0 r1 ry rB r rC r| r} r r~ r r+ r r r3 r r r r r s .
r ) metaclassc @ s e Zd ZdZdS )r z.Enum where members are also (and must be) intsN)r/ r, r0 r1 r r r r r ' s r c C r rs 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 sX |s
|dur|S dS t |D ]}zt|}W n ty% td| dw d|d S )r Nr zInvalid Flag value: %rr )r _high_bitr r& )r r r r r high_bitr r r rB 3 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 )rA )r- r3 original_valuepossible_memberr r r rC F s
zFlag._missing_c C sb | j |d}|du r/t| |\}}|r td|| jf t| }d|_||_| j ||}|S )L
Create a composite member iff value contains only members.
Nr )
r r
_decomposerM r0 r ry r r{ r )r- r3
pseudo_memberr extra_flagsr r r rA S s
zFlag._create_pseudo_member_c C s8 t || jstdt|j| jjf |j| j@ |jkS )zP
Returns True if self has at least the same flags set as other.
r )rO r= r& r r0 r{ r( otherr r r r g s zFlag.__contains__c C sV | j }| jd urd|j| j| jf S t|| j\}}d|jddd |D | jf S )Nr |c S g | ]
}t |jp
|jqS r rP r r{ ru r r r r x r z!Flag.__repr__..)r= r r/ r{ r r r( r- r uncoveredr r r r| q s
z
Flag.__repr__c C s| | j }| jd urd|j| jf S t|| j\}}t|dkr/|d jd u r/d|j|d jf S d|jddd |D f S )Nr r r z%s.%rr
c S r r r ru r r r r r z Flag.__str__..)r= r r/ r r{ r r r r r r r} | s
z Flag.__str__c C r rs )boolr{ r; r r r r r z
Flag.__bool__c C s" t || jstS | | j|jB S rs rO r= NotImplementedr{ r r r r __or__ z Flag.__or__c C s" t || jstS | | j|j@ S rs r r r r r __and__ r z Flag.__and__c C s" t || jstS | | j|jA S rs r r r r r __xor__ r z Flag.__xor__c C sN t | j| j\}}| d}| jD ]}||vr!|j| j@ s!||B }q| |S r )r r= r{ )r( r r invertedrv r r r
__invert__ s
zFlag.__invert__N)r/ r, r0 r1 rB r rC rA r r| r} r r r r r r r r r r . s
r 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 |tstd|| jf | |}|S )r r )rO intrM r0 rA )r- r3
new_memberr r r rC s
zIntFlag._missing_c C s | j |d}|du rU|g}t| |\}}|r;t|}d| }|| j vr-||vr-|| || kr5d}n||N }|st|D ]}t| |}d|_||_ | j
||}q?|S )r Nr r ) r r r r rU r r ry r r{ r )r- r3 r need_to_creater r
bit
flag_valuer r r rA s(
z IntFlag._create_pseudo_member_c C s0 t || jtfs
tS | | j| |jB }|S rs rO r= r r r{ )r( r r r r r r s zIntFlag.__or__c C s, t || jtfs
tS | | j| |j@ S rs r r r r r r zIntFlag.__and__c C s, t || jtfs
tS | | j| |jA S rs r r r r r r r! zIntFlag.__xor__c C s | | j }|S rs r )r( r r r r r s zIntFlag.__invert__N)r/ r, r0 r1 r rC rA r r r __ror____rand____rxor__r r r r r r s
r c C s | d S )zJ
returns index of highest bit, or -1 if value is zero or negative
r )
bit_length)r3 r r r r s r c C s^ g }| j D ]\}}||jkr|||jf q|r-ddd |D }td| |f | S )zI
Class decorator for enumerations ensuring unique member values.
z, c S s g | ]
\}}d ||f qS )z%s -> %sr )rd aliasr r r r r r zunique..z duplicate values found in %r: %s)r rk r rU r rM ) enumeration
duplicatesr r
alias_detailsr r r r
s
r
c C s |}|dk }g }| D ]}|j }|r!||@ |kr!|| || M }q
|sG|}|rGdt| }|| jv r@|| j| || M }|| M }|s(|sV|| jv rV|| j| |jdd dd t|dkrq|d j |krq|d ||fS )z-
Extract all members from the value.
r r c S r rs r )rv r r r r s z _decompose..T)rW reverser )r3 rU r r r r r ) flagr3 not_coverednegativer r r tmpr r r r r s0
r )r typesr r __all__r r r r% r. r r2 r dictr4 r r r r r r r r r r
r r r r r s4
T vI