3
\F @ s> d dl Z d dlmZmZ d dlmZ d dlmZ yd dl m
Z
W n e k
r` d dl m
Z
Y nX dddd d
d d gZ
d
d Zdd Zdd Zdd Ze ZG dd d ZG dd deZdZG dd deZG dd dedZG dd deeZd d Z G d d d eZ G d d
d
ee Z d d! Z d"d Z d#d$ Z!d%d& Z"dS )' N)MappingProxyTypeDynamicClassAttribute)reduce)or_) OrderedDictEnumMetaEnumIntEnumFlagIntFlagautouniquec C s t | dp t | dp t | dS )N__get____set__
__delete__)hasattr)obj r /usr/lib64/python3.6/enum.py_is_descriptor s
r c C sT | d d | dd ko"dkn oR| dd dkoR| dd dkoRt | dkS ) N __ _ r )len)namer r r
_is_dunder s (r c C sL | d | d kodkn oJ| dd dkoJ| dd dkoJt | dkS )Nr r 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) TypeError)selfprotor r r _break_on_call_reduce- 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 3 s c s( e Zd Z fddZ fddZ ZS ) _EnumDictc s t j g | _g | _d S )N)super__init__
_member_names _last_values)r$ ) __class__r r r1 A s
z_EnumDict.__init__c s t |r.|d krtd|dkrt| d| nt|rD|dkrd}n|| jkr\td| nxt|s|| krtd || | f t|tr|j t
kr| j |d
t | j| j
d d |_ |j }| jj| | j
j| t j|| d S ) N_order__create_pseudo_member__generate_next_value_ _missing_z(_names_ are reserved for future Enum use_generate_next_value __order__zAttempted to reuse key: %rz%r already defined as: %rr )r5 r6 r7 r8 )r"
ValueErrorsetattrr r2 r# r
isinstancer r. r- r9 r r3 appendr0 __setitem__)r$ keyr. )r4 r r r? F s,
" z_EnumDict.__setitem__)r+ r( r, r1 r?
__classcell__r r )r4 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#Zed$d% Zed&d' Z ZS ))r c C s0 t }| j|\}}|d k r,t|dd |d< |S )Nr7 )r/ _get_mixins_getattr)metaclsr) bases enum_dict member_type
first_enumr r r __prepare__r s
zEnumMeta.__prepare__c s | j |\}| j |\}}} fdd jD }x jD ]
} | = q>W jdd }
t|dh@ } | r~tdjdj| d krd d< t j | || } g | _
t | _ | _
d d
| j D }
i | _d krtk rd}tfdd|D st| x6 jD ]*}|| }t|ts0|f}n|}tkrD|f}|sf|| }t|ds||_n6|| f| }t|dstkr||_n
| |_|j}||_| |_|j| x8| j j D ] \} }|j|jkr|}P qW | j
j| ||
krt | || || j |< y|| j|< W n t k
r6 Y nX qW xPdD ]H} t | | }t | d }t || d }|d k rD||krDt | | | qDW t d k r|r|| _ t j | _ |
d k rt|
t!r|
j"ddj# }
|
| j
krt d| S )Nc s i | ]} | |qS r r ).0k) classdictr r
s z$EnumMeta.__new__..r5 mroz Invalid enum member name: {0},__doc__zAn enumeration.c S s. h | ]&}|j j D ]\}}t|tr|qqS r )__dict__itemsr= r )rJ crK vr r r s z#EnumMeta.__new__..r' __getnewargs_ex____getnewargs__
__reduce__c 3 s | ]}| j kV qd S )N)rQ )rJ m)rG r r s z#EnumMeta.__new__.._value___repr____str__
__format__ z#member order does not match _order_)rV rW r' rX )r\ r] r^ r' )$rB
_find_new_r2 popsetr; formatjoinr0 __new___member_names_r _member_map_
_member_type_rN _value2member_map_objectanyr* r= tupler r[ _name_ __objclass__r1 rR r> r< r# rC r __new_member__strreplacesplit)rD r) rE rL rH re save_newuse_args enum_membersr r5
invalid_names
enum_classdynamic_attributesmethods member_namer. args enum_membercanonical_member class_method
obj_method enum_method)r4 )rL rG r re | s
zEnumMeta.__new__c C s dS )NTr )r$ r r r __bool__ s zEnumMeta.__bool__Nr )modulequalnametypestartc C s* |d kr| j | |S | j||||||dS )N)r r r r )re _create_)r) r. namesr r r r r r r __call__ s zEnumMeta.__call__c C s t || o|j| jkS )N)r= rm rg )r) memberr r r __contains__) s zEnumMeta.__contains__c s( || j krtd| j t j| d S )Nz %s: cannot delete Enum member.)rg AttributeErrorr+ r0 __delattr__)r) attr)r4 r r r , s
zEnumMeta.__delattr__c C s ddddg| j S )Nr4 rP __members__r( )rf )r$ r r r __dir__4 s
zEnumMeta.__dir__c C s> t |rt|y
| j| S tk
r8 t|d Y nX d S )N)r r rg KeyError)r) r r r r __getattr__8 s
zEnumMeta.__getattr__c C s
| j | S )N)rg )r) r r r r __getitem__H s zEnumMeta.__getitem__c s fdd j D S )Nc 3 s | ]} j | V qd S )N)rg )rJ r )r) r r rZ L s z$EnumMeta.__iter__..)rf )r) r )r) r __iter__K s zEnumMeta.__iter__c C s
t | jS )N)r rf )r) r r r __len__N s zEnumMeta.__len__c C s
t | jS )N)r rg )r) r r r r Q s zEnumMeta.__members__c C s
d| j S )Nz )r+ )r) r r r r\ [ s zEnumMeta.__repr__c s fddt jD S )Nc 3 s | ]} j | V qd S )N)rg )rJ r )r) r r rZ _ s z(EnumMeta.__reversed__..)reversedrf )r) r )r) r __reversed__^ s zEnumMeta.__reversed__c s0 | j jdi }||kr tdt j|| d S )Nrg zCannot reassign members.)rQ getr r0 __setattr__)r) r r.
member_map)r4 r r r a s zEnumMeta.__setattr__c C s | j }|d kr| fn|| f}| j|\} }
|j||} t|trP|jddj }t|ttfr|rt|d tr|g } }g }
xDt | D ]8\}}|
j
||||
d d }|
j | |j ||f qW x6|D ].}t|tr||| }}n|\}}|| |< qW |j |||| }|d krTyt
jdjd }W n( ttfk
rR } zW Y d d }~X nX |d krht| n||_|d k r~||_|S )NrO r_ r r r+ )r4 rB rI r= rp rq rr rl list enumerater7 r> re sys _getframe f_globalsr r; r* r( r, )r)
class_namer r r r r rD rE r rH rL original_names last_valuescountr r. itemrz member_valuerw excr r r r n s<
zEnumMeta._create_c C s | s t tfS d }}x,| D ]$}|tk rt|tr|jrtdqW t|tsTtdt| d tst| d }| d }n8x6| d jD ](}t|tr|d kr|}q|d kr|}qW ||fS )NzCannot extend enumerationszHnew enumerations must be created as `ClassName([mixin_type,] enum_type)`r r r! )rj r
issubclassrf r# __mro__)rE rG rH baser r r rB s(
zEnumMeta._get_mixins_c C s | j dd }|d k }|d krtxVdD ]H}x8||fD ],}t||d }|d d jtjtjhkr0|}P q0W |d k r"P q"W tj}|tjkrd}nd}|||fS )Nre ro FT)ro re )r rC re rj r ) rL rG rH re rs methodpossibletargetrt r r r r` s(
zEnumMeta._find_new_)N)r+ r( r, classmethodrI re r r r r r r r r r propertyr r\ r r r staticmethodrB r` rA r r )r4 r r p s&
5 -c @ s~ e Zd Zdd Zdd Zedd Zdd Zd d
Zd d Z d
d Z
dd Z dd Z e
dd Ze
dd ZedddZdS )r c C sj t || kr|S y|| jkr&| j| S W n6 tk
r^ x | jj D ]}|j|krD|S qDW Y nX | j|S )N)r ri r# rg valuesr[ r8 )r) r. r r r r re s
z Enum.__new__c C s8 x2t |D ]"}y|d S tk
r* Y q
X q
W |S d S )Nr )r r# )r r r r
last_valuer r r r7 s
zEnum._generate_next_value_c C s t d|| jf d S )Nz%r is not a valid %s)r; r+ )r) r. r r r r8 " s zEnum._missing_c C s d| j j| j| jf S )Nz <%s.%s: %r>)r4 r+ rm r[ )r$ r r r r\ & s z
Enum.__repr__c C s d| j j| jf S )Nz%s.%s)r4 r+ rm )r$ r r r r] * s z Enum.__str__c s& fdd j j D }dddg| S )Nc s2 g | ]*}|j D ] }|d dkr| jkr|qqS )r r )rQ rg )rJ r) rY )r$ r r
/ s
z Enum.__dir__..r4 rP r( )r4 rN )r$ added_behaviorr )r$ r r - s
z Enum.__dir__c C s0 | j tkrt}t| }n | j }| j}|j||S )N)rh rj rp r[ r^ )r$ format_specr) valr r r r^ 6 s
zEnum.__format__c C s
t | jS )N)hashrm )r$ r r r __hash__E s z
Enum.__hash__c C s | j | jffS )N)r4 r[ )r$ r% r r r r' H s zEnum.__reduce_ex__c C s | j S )N)rm )r$ r r r r R s z Enum.namec C s | j S )N)r[ )r$ r r r r. W s z
Enum.valueNc
s t tj| }r t n| fddj D }y|jdd d W n$ tk
rp |jdd d Y nX | |||d} t| _|j| j | ||< | S )Nc s g | ]} |r|| fqS r r )rJ r )filtersourcer r r q s z!Enum._convert..c S s | d | d fS )Nr r r )tr r r v s z Enum._convert..)r@ c S s | d S )Nr r )r r r r r y s )r )
varsr moduleskeyssortr# _reduce_ex_by_namer' updater )r) r r r r module_globalsmembersr )r r r _convert\ s
z
Enum._convert)N)r+ r( r, re r7 r r8 r\ r] r r^ r r' r r r. r r r r r r s
) metaclassc @ s e Zd ZdS )r N)r+ r( r, r r r r r s c C s | j S )N)r )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 xBt |D ]6}yt|}P W q tk
rR td| d Y q X q W d|d S )Nr zInvalid Flag value: %rr )r _high_bit Exceptionr# )r r r r r high_bitr r r r7 s zFlag._generate_next_value_c C s. |}|dk r| }| j |}|dk r*| }|S )Nr )r6 )r) r. original_valuepossible_memberr r r r8 s
zFlag._missing_c C sb | j j|d }|d kr^t| |\}}|r:td|| jf tj| }d |_||_| j j ||}|S )Nz%r is not a valid %s)
ri r
_decomposer; r+ rj re rm r[
setdefault)r) r.
pseudo_memberr extra_flagsr r r r6 s
zFlag._create_pseudo_member_c C s" t || jstS |j| j@ |jkS )N)r= r4 NotImplementedr[ )r$ otherr r r r s zFlag.__contains__c C sV | j }| jd k r$d|j| j| jf S t|| j\}}d|jdjdd |D | jf S )Nz <%s.%s: %r>|c S s g | ]}t |jp|jqS r )rp rm r[ )rJ rY r r r r s z!Flag.__repr__..)r4 rm r+ r[ r rd )r$ r) r uncoveredr r r r\ 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|jdjdd |D f S d S )Nz%s.%sr r z%s.%rr c S s g | ]}t |jp|jqS r )rp rm r[ )rJ rY r r r r s z Flag.__str__..)r4 rm r+ r r[ r rd )r$ r) r r r r r r] s
z Flag.__str__c C s
t | jS )N)boolr[ )r$ r r r r s z
Flag.__bool__c C s" t || jstS | j| j|jB S )N)r= r4 r r[ )r$ r r r r __or__ s z Flag.__or__c C s" t || jstS | j| j|j@ S )N)r= r4 r r[ )r$ r r r r __and__ s z Flag.__and__c C s" t || jstS | j| j|jA S )N)r= r4 r r[ )r$ r r r r __xor__ s z Flag.__xor__c sD t jj\ } fddjD }tt|jd}j|S )Nc s& g | ] }| kr|j j @ r|qS r )r[ )rJ rY )r r$ r r r s z#Flag.__invert__..r )r r4 r[ r _or_)r$ r inverted_membersinvertedr )r r$ r
__invert__ s zFlag.__invert__N)r+ r( r, r7 r r8 r6 r r\ r] 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 | j|}|S )Nz%r is not a valid %s)r= intr; r+ r6 )r) r.
new_memberr r r r8 s
zIntFlag._missing_c C s | j j|d }|d kr|g}t| |\}}xL|rvt|}d| }|| j kr\||kr\|j| || krld}q,||N }q,W x6t|D ]*}tj| |}d |_||_ | j j
||}qW |S )Nr r ) ri r r r r> r r re rm r[ r )r) r. r need_to_creater r bit
flag_valuer r r r6 s&
z IntFlag._create_pseudo_member_c C s0 t || jtfstS | j| j| j|jB }|S )N)r= r4 r r r[ )r$ r resultr r r r s zIntFlag.__or__c C s, t || jtfstS | j| j| j|j@ S )N)r= r4 r r r[ )r$ r r r r r # s zIntFlag.__and__c C s, t || jtfstS | j| j| j|jA S )N)r= r4 r r r[ )r$ r r r r r ( s zIntFlag.__xor__c C s | j | j }|S )N)r4 r[ )r$ r r r r r 1 s zIntFlag.__invert__N)
r+ r( r, r r8 r6 r r r __ror____rand____rxor__r r r r r r s c C s | j d S )Nr )
bit_length)r. r r r r 6 s r c C sb g }x0| j j D ]"\}}||jkr|j||jf qW |r^djdd |D }td| |f | S )Nz, c S s g | ]\}}d ||f qS )z%s -> %sr )rJ aliasr r r r r B s zunique..z duplicate values found in %r: %s)r rR r r> rd r; ) enumeration
duplicatesr r
alias_detailsr r r r
: s
c C s |}|dk }|r*dd t | jj D }ndd t | jj D }g }x4|D ],\}}|rL||@ |krL|j| || M }qLW | r|| jkr|j| j| |jdd dd t|d kr|d j|kr|jd ||fS )
Nr c S s" g | ]\}}|j d k r||fqS )N)r )rJ rT rY r r r r R s z _decompose..c S s* g | ]"\}}|j d k s t|r||fqS )N)r
_power_of_two)rJ rT rY r r r r Y s c S s | j S )N)r[ )rY r r r r d s z _decompose..T)r@ reverser )r ri rR r> r r r. ra )flagr. not_coverednegativeflags_to_checkr r r r r r r G s$
r c C s | dk r dS | dt | kS )Nr Fr )r )r. r r r r j s r )#r typesr r functoolsr operatorr r _collectionsr ImportError collections__all__r r r" r* rj r- r dictr/ r r r r r r r
r r r
r r r r r r s@ 3 mA
#