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� de�ZdZG dd� de�ZG dd� ded�ZG dd� dee�Zd d � Z G d d � d e�Z G d d
� d
ee �Z d d!� Z d"d � Z d#d$� Z!d%d&� Z"dS )'� N)�MappingProxyType�DynamicClassAttribute)�reduce)�or_)� OrderedDict�EnumMeta�Enum�IntEnum�Flag�IntFlag�auto�uniquec C s t | d�p t | d�p t | d�S )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)�self�protor r r �_break_on_call_reduce- s z6_make_class_unpicklable.<locals>._break_on_call_reducez <unknown>)�
__reduce_ex__�
__module__)�clsr& r r r �_make_class_unpicklable+ s r* c @ s e Zd ZeZdS )r N)�__name__r( � __qualname__�
_auto_null�valuer r r r r 3 s c s( e Zd Z� fdd�Z� fdd�Z� 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��|dkr�t| d|� n�t|�rD|dkr�d}n�|| jkr\td| ��nxt|�s�|| kr�td || | f ��t|t�r�|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" �
ValueError�setattrr 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� fdd�Zdd� Zd(ddddd �d
d �Zd d
� Z� fdd�Z 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� kr�d� d< t� j | ||� �} g | _
t � | _ �| _
d d
� | j� D �}
i | _d � k�r�tk �rd}t�fdd�|D ���st| � �x6� jD �]*}|| }t|t��s0|f}n|}�tk�rD|f}|�sf|| �}t|d��s�||_n6|| f|�� }t|d��s��tk�r�||_n
�|� |_|j}||_| |_|j|� x8| j j� D ] \} }|j|jk�r�|}P �q�W | j
j|� ||
k�rt | ||� || j |< y|| j|< W n t k
�r6 Y nX �qW xPdD ]H} t | | �}t �| d �}t || d �}|d k �rD||k�rDt | | |� �qDW t d k �r�|�r�|| _ t j | _ |
d k �r�t|
t!��r�|
j"dd�j#� }
|
| j
k�r�t d��| S )Nc s i | ]}� | |�qS r r )�.0�k)� classdictr r �
<dictcomp>� s z$EnumMeta.__new__.<locals>.<dictcomp>r5 �mroz Invalid enum member name: {0}�,�__doc__zAn enumeration.c S s. h | ]&}|j j� D ]\}}t|t�r|�qqS r )�__dict__�itemsr= r )rJ �crK �vr r r � <setcomp>� s z#EnumMeta.__new__.<locals>.<setcomp>r' �__getnewargs_ex__�__getnewargs__�
__reduce__c 3 s | ]}|� j kV qd S )N)rQ )rJ �m)rG r r � <genexpr>� s z#EnumMeta.__new__.<locals>.<genexpr>�_value_�__repr__�__str__�
__format__� z#member order does not match _order_)rV rW r' rX )r\ r] r^ r' )$rB �
_find_new_r2 �pop�setr; �format�joinr0 �__new__�_member_names_r � _member_map_�
_member_type_rN �_value2member_map_�object�anyr* r= �tupler r[ �_name_� __objclass__r1 rR r> r< r# rC r �__new_member__�str�replace�split)rD r) rE rL rH re �save_new�use_args� enum_membersr r5 �
invalid_names�
enum_class�dynamic_attributes�methods� member_namer. �args� enum_member�canonical_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 )�module�qualname�type�startc C s* |d kr| j | |�S | j||||||d�S )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__.<locals>.<genexpr>)rf )r) r )r) r �__iter__K s zEnumMeta.__iter__c C s
t | j�S )N)r rf )r) r r r �__len__N s zEnumMeta.__len__c C s
t | j�S )N)r rg )r) r r r r� Q s zEnumMeta.__members__c C s
d| j S )Nz <enum %r>)r+ )r) r r r r\ [ s zEnumMeta.__repr__c s � fdd�t � j�D �S )Nc 3 s | ]}� j | V qd S )N)rg )rJ r )r) r r rZ _ s z(EnumMeta.__reversed__.<locals>.<genexpr>)�reversedrf )r) r )r) r � __reversed__^ s zEnumMeta.__reversed__c s0 | j jdi �}||kr td��t� 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|t�rP|jdd�j� }t|ttf�r�|r�t|d t�r�|g } }g }
xDt | �D ]8\}}|
j
||||
d d � �}|
j |� |j ||f� q�W x6|D ].}t|t�r�||| }}n|\}}|| |< q�W |j |||| �}|d k�rTyt
jd�jd }W n( ttfk
�rR } zW Y d d }~X nX |d k�rht|� 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_values�countr r. �itemrz � member_valuerw �excr r r r� n s<
zEnumMeta._create_c C s� | s t tfS d }}x,| D ]$}|tk rt|t�r|jrtd��qW t|t�sTtd��t| d t�st| d }| d }n8x6| d jD ](}t|t�r�|d kr�|}q�|d kr�|}q�W ||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}|tjkr�d}nd}|||fS )Nre ro FT)ro re )r� rC re rj r ) rL rG rH re rs �method�possible�targetrt 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� �Zeddd��ZdS )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 �
<listcomp>/ s
z Enum.__dir__.<locals>.<listcomp>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 | j�S )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|�� �fdd��j� D �}y|jdd� d� W n$ tk
rp |jdd� d� Y nX | |||d�} t| _|j| j � | ||< | S )Nc s g | ]}� |�r|�| f�qS r r )rJ r )�filter�sourcer r r� q s z!Enum._convert.<locals>.<listcomp>c S s | d | d fS )Nr r r )�tr r r �<lambda>v s z Enum._convert.<locals>.<lambda>)r@ c S s | d S )Nr r )r� r r r r� y s )r� )
�varsr� �modules�keys�sortr# �_reduce_ex_by_namer' �updater� )r) r r� r� r� �module_globals�membersr )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_value�possible_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 || j�stS |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|j��qS r )rp rm r[ )rJ rY r r r r� � s z!Flag.__repr__.<locals>.<listcomp>)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|j��qS r )rp rm r[ )rJ rY r r r r� � s z Flag.__str__.<locals>.<listcomp>)r4 rm r+ r� r[ r rd )r$ r) r� r� r r r r] � s
z Flag.__str__c C s
t | j�S )N)�boolr[ )r$ r r r r� � s z
Flag.__bool__c C s" t || j�stS | j| j|jB �S )N)r= r4 r� r[ )r$ r� r r r �__or__� s z Flag.__or__c C s" t || j�stS | j| j|j@ �S )N)r= r4 r� r[ )r$ r� r r r �__and__� s z Flag.__and__c C s" t || j�stS | j| j|jA �S )N)r= r4 r� r[ )r$ r� r r r �__xor__� s z Flag.__xor__c sD t �j�j�\� }� �fdd��jD �}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__.<locals>.<listcomp>r )r� r4 r[ r �_or_)r$ r� �inverted_members�invertedr )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 |t�s 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
||�}q�W |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 || jtf�stS | j| j| j|�jB �}|S )N)r= r4 r� r� r[ )r$ r� �resultr r r r� s zIntFlag.__or__c C s, t || jtf�stS | j| j| j|�j@ �S )N)r= r4 r� r� r[ )r$ r� r r r r� # s zIntFlag.__and__c C s, t || jtf�stS | 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.<locals>.<listcomp>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||f�qS )N)r )rJ rT rY r r r r� R s z _decompose.<locals>.<listcomp>c S s* g | ]"\}}|j d k s t|�r||f�qS )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.<locals>.<lambda>T)r@ �reverser )r� ri rR r> r� r r. ra )�flagr. � not_covered�negative�flags_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 �<module> s@ 3 mA
#