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 )N__get____set__
__delete__)hasattr)obj r /usr/lib64/python3.8/enum.py_is_descriptor s
r c C sL t | dkoJ| d d | dd ko.dkn oJ| d dkoJ| d dkS )N ___lennamer r r
_is_dunder s &
r c C sL t | dkoJ| d | d ko&dkn oJ| dd dkoJ| dd dkS )Nr r r r r r r r r
_is_sunder! s r c C s dd }|| _ d| _d S )Nc 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 )
__reduce_ex__
__module__)clsr% r r r _make_class_unpicklable, s r) c @ s e Zd ZeZdS )r N)__name__r' __qualname__
_auto_nullvaluer r r r r 6 s c s( e Zd Z fddZ fddZ ZS ) _EnumDictc s& t g | _g | _g | _d| _d S )NF)super__init__
_member_names _last_values_ignore _auto_calledr# __class__r r r0 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 | jd d |_d| _|j}| j | | j| t || d S )N)_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__r8 zAttempted to reuse key: %rz%r already defined as: %rr T)r
ValueErrorr4 r! setattr
isinstancestrreplacesplitlistr3 setr1 r r r r- r, r= r r2 appendr/ __setitem__)r# keyr- alreadyr6 r r rJ K sT
z_EnumDict.__setitem__)r* r' r+ r0 rJ
__classcell__r r r6 r r. = s r. c s e Zd Zedd Z fddZdd Zd.ddddd d
d Zd d
Z fddZ dd Z
dd Z dd Z dd Z
dd Zedd Zd d Zd d Z fd d!Zddddd d"d#Zd/d$d%Zd&d' Zed(d) Zed*d+ Zed,d- Z ZS )0r 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__..r8 mro z Invalid enum member name: {0}r> __doc__zAn enumeration.c S s. h | ]&}|j D ]\}}t|tr|qqS r )__dict__itemsrC r )rW crX vr r r s
z#EnumMeta.__new__..r& )__getnewargs_ex____getnewargs__r&
__reduce__c 3 s | ]}| j kV qd S N)r^ rW m)rT r r s z#EnumMeta.__new__.._value_)__repr____str__
__format__r& r? z#member order does not match _order_)$
setdefaultrI poprO
_find_new_r1 rH rA formatjoinr/ __new___member_names_ _member_map_
_member_type_r[ _value2member_map_objectanyr) rC tupler rj _name_ __objclass__r0 r_ rB r! rP r __new_member__rD rE rF )rQ r( rR rY ignorerK rU rs save_newuse_args enum_membersr r8
invalid_names
enum_classdynamic_attributesmethods member_namer- args enum_membercanonical_member class_method
obj_method enum_methodr6 )rY rT r rs s
zEnumMeta.__new__c C s dS )NTr r5 r r r __bool__3 s zEnumMeta.__bool__Nr modulequalnametypestartc C s* |d kr| | |S | j||||||dS )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 Nz3unsupported operand type(s) for 'in': '%s' and '%s')rC r r! r r+ r7 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* r/ __delattr__)r( attrr6 r r r e s
zEnumMeta.__delattr__c C s ddddg| j S )Nr7 r] __members__r' rt r5 r r r __dir__l s
zEnumMeta.__dir__c C s@ t |rt|z | j| W S tk
r: t|d Y nX d S rf )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 )Nc 3 s | ]} j | V qd S rf r rW 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 rf )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 )Nc 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 || d S )Nru zCannot reassign members.)r^ getr r/ __setattr__)r( r r-
member_mapr6 r r r s zEnumMeta.__setattr__c
C s~ | j }|d kr| fn|| f}| | |\} }
|||} t|trR|dd }t|ttfr|rt|d tr|g } }g }
t | D ]8\}}|
||||
d d }|
| | ||f q|D ].}t|tr||| }}n|\}}|| |< q| |||| }|d krPzt
djd }W n* tttfk
rN } zW 5 d }~X Y nX |d krdt| n||_|d k rz||_|S )Nr> r? r r r* )r7 rO rV rC rD rE rF rz rG enumerater: rI rs sys _getframe f_globalsr rA r r) r' r+ )r(
class_namer r r r r rQ rR r rU rY 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 )Nc s g | ]\}} |r||fqS r r )rW r r- filterr r
s z&EnumMeta._convert_..c S s | d | d fS )Nr r r tr r r z$EnumMeta._convert_..)rK 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 rR chainbaser r r rN
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 rR r rU rT r r r rO s
zEnumMeta._get_mixins_c C s | dd }|d k }|d krpdD ]H}||fD ].}t||d }|d d jtjtjhkr,|} q\q,|d k r qpq tj}|tjkrd}nd}|||fS )Nrs )r} rs FT)r rP rs rx r ) rY rT rU rs r methodpossibletargetr r r r rp C s*
zEnumMeta._find_new_)N)N) r* r' r+ classmethodrV rs r r r r r r r r r propertyr rk r r r r r staticmethodrN rO rp rM r r r6 r r s6
%
5
!
.c @ sp e Zd Zdd Zdd Zedd Zdd Zd d
Zd d Z d
d Z
dd Z dd Z e
dd Ze
dd ZdS )r 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; ExceptionrC rA 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 Nr )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>)r7 r* r{ rj r5 r r r rk s
z
Enum.__repr__c C s d| j j| jf S )N%s.%s)r7 r* r{ r5 r r r rl s z Enum.__str__c s6 fdd j D dd jD }dddg| S )Nc s2 g | ]*}|j D ] }|d dkr| jkr|qqS r r )r^ ru )rW r( rh r5 r r r s
z Enum.__dir__..c S s g | ]}|d dkr|qS r r rg r r r r s r7 r] r' )r7 r[ r^ )r# added_behaviorr r5 r r s
z Enum.__dir__c C sJ t | jtjtjfk}| jtks$|r2t}t| }n | j}| j}|||S rf ) r rl r r rv rx rD rj rm )r# format_specstr_overriddenr( valr r r rm s
zEnum.__format__c C s
t | jS rf )hashr{ r5 r r r __hash__ s z
Enum.__hash__c C s | j | jffS rf r7 rj r" r r r r& s zEnum.__reduce_ex__c C s | j S rf )r{ r5 r r r r s z Enum.namec C s | j S rf rj r5 r r r r- s z
Enum.valueN)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 ZdS )r N)r* r' r+ r r r r r s c C s | j S rf r r" r r r r s r c @ sl e Zd Zdd Zedd Zedd Zdd Zd d
Zd d Z d
d Z
dd Z dd Z dd Z
dd ZdS )r c C sd |s|d k r|S dS t |D ]:}zt|}W qXW q tk
rT td| d Y q X q d|d S )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 r )r9 )r( r- original_valuepossible_memberr r r r; s
zFlag._missing_c C sb | j |d }|d kr^t| |\}}|r:td|| jf t| }d |_||_| j ||}|S Nr )
rw r
_decomposerA r* rx rs r{ rj rn )r( r-
pseudo_memberr extra_flagsr r r r9 # s
zFlag._create_pseudo_member_c C s8 t || js&tdt|j| jjf |j| j@ |jkS r )rC r7 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 rD r{ rj rg r r r r H s z!Flag.__repr__..)r7 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__..)r7 r{ r* r rj r rr r r r r rl L s
z Flag.__str__c C s
t | jS rf )boolrj r5 r r r r Y s z
Flag.__bool__c C s" t || jstS | | j|jB S rf rC r7 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 r7 rj )r# r r invertedrh r r r
__invert__k s
zFlag.__invert__N)r* r' r+ r: r r; r9 r rk rl r r r r r r r r r r s
c @ sP e Zd Zedd Zedd Zdd Zdd Zd d
ZeZ eZ
eZ d d Z d
S )r c C s* t |ts td|| jf | |}|S r )rC intrA r* r9 )r( r-
new_memberr r r r; y s
zIntFlag._missing_c C s | j |d }|d kr|g}t| |\}}|rtt|}d| }|| j krZ||krZ|| || krjd}q*||N }q*t|D ]*}t| |}d |_||_ | j
||}q||S )Nr r ) rw r r r rI r r rs r{ rj rn )r( r- r need_to_creater r bit
flag_valuer r r r9 s(
z IntFlag._create_pseudo_member_c C s0 t || jtfstS | | j| |jB }|S rf rC r7 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; r9 r r r __ror____rand____rxor__r r r r r r t s
c C s | d S r )
bit_lengthr- r r r r s r c C s^ g }| j D ]"\}}||jkr|||jf q|rZddd |D }td| |f | S )Nz, c S s g | ]\}}d ||f qS )z%s -> %sr )rW aliasr r r r r s zunique..z duplicate values found in %r: %s)r r_ r rI rr rA ) 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 )
Nr c S s" g | ]\}}|j d k r||fqS rf r rW 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)rK reverser )rG rw r_ rI 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%