B
    u9aڇ                 @   s&  d dl Z d dlmZmZ yd dlmZ W n  ek
rH   d dlmZ Y nX dddddd	d
gZdd Z	dd Z
dd Zdd Ze ZG dd	 d	ZG dd deZdZG dd deZG dd dedZG dd deeZdd ZG dd deZG dd deeZdd Zd d
 Zd!d" Zd#d$ ZdS )%    N)MappingProxyTypeDynamicClassAttribute)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/lib/python3.7/enum.py_is_descriptor   s    

r   c             C   sL   t | dkoJ| d d | dd    ko.dkn  oJ| d dkoJ| d dkS )N      ___)len)namer   r   r   
_is_dunder   s    (r   c             C   sL   t | dkoJ| d | d   ko&dkn  oJ| dd dkoJ| dd dkS )Nr   r   r      r   )r   )r   r   r   r   
_is_sunder"   s     r   c             C   s   dd }|| _ 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 Zd ZeZdS )r
   N)__name__r%   __qualname__
_auto_nullvaluer   r   r   r   r
   2   s   c                   s(   e Zd Z fddZ fddZ  ZS )	_EnumDictc                s    t    g | _g | _g | _d S )N)super__init___member_names_last_values_ignore)r!   )	__class__r   r   r.   @   s    
z_EnumDict.__init__c                sP  t |r|dkrtd|dkr.t| d| nV|dkrt|trR|dd }nt|}|| _t	|t	| j
@ }|rtd|f nt|r|d	krd
}n|| j
krtd| n|| jkrn~t|s>|| krtd|| | f t|tr&|jtkr | |dt| j
| j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 user5   _generate_next_valuer7   , z-_ignore_ cannot specify already set names: %r	__order__r3   zAttempted to reuse key: %rz%r already defined as: %rr   )r   
ValueErrorsetattr
isinstancestrreplacesplitlistr1   setr/   r   r    r   r
   r+   r*   r8   r   r0   appendr-   __setitem__)r!   keyr+   already)r2   r   r   rE   F   s>    	



"z_EnumDict.__setitem__)r(   r%   r)   r.   rE   __classcell__r   r   )r2   r   r,   9   s   r,   c                   s   e Zd Zedd Z fddZdd Zd(ddddd	d
dZdd Z fddZ	dd Z
dd Zdd Zdd Zdd Zedd Zdd Zdd Z fd d!Zddddd	d"d#Zed$d% Zed&d' Z  ZS ))r   c             C   s0   t  }| |\}}|d k	r,t|dd |d< |S )Nr5   )r,   _get_mixins_getattr)metaclsr&   bases	enum_dictmember_type
first_enumr   r   r   __prepare__}   s
    zEnumMeta.__prepare__c          	      s"    dg d  d }x|D ]} |d  q W | |\}|  |\}}}	 fdd jD }
x jD ]
} |= qrW  dd }t|
ddh@ }|rtdd	|d	 krd
 d	< t
 | || }g |_t |_|_dd | D }i |_d kr:tk	r:d}tfdd|D s:t| x6 jD ]*}|
| }t|tsf|f}n|}tkrz|f}|	s||}t|ds||_n6||f| }t|ds҈tkr||_n
| |_|j}||_||_|j|  x8|j D ]\}}|j|jkr|}P qW |j| ||kr>t||| ||j|< y||j|< W n tk
rl   Y nX qDW xPdD ]H}t||}t|d }t||d }|d k	rz||krzt||| qzW t d k	r|r||_!t j|_|d k	rt|t"r
|#dd$ }||jkrtd|S )Nr7   c                s   i | ]} | |qS r   r   ).0k)	classdictr   r   
<dictcomp>   s    z$EnumMeta.__new__.<locals>.<dictcomp>r3   mro zInvalid enum member name: {0}r9   __doc__zAn enumeration.c             S   s.   h | ]&}|j  D ]\}}t|tr|qqS r   )__dict__itemsr>   r   )rQ   crR   vr   r   r   	<setcomp>   s    z#EnumMeta.__new__.<locals>.<setcomp>r$   )__getnewargs_ex____getnewargs__r$   
__reduce__c             3   s   | ]}| j kV  qd S )N)rX   )rQ   m)rN   r   r   	<genexpr>   s    z#EnumMeta.__new__.<locals>.<genexpr>_value_)__repr____str__
__format__r$   r:   z#member order does not match _order_)%
setdefaultrD   poprI   
_find_new_r/   rC   r<   formatjoinr-   __new___member_names_r   _member_map__member_type_rU   _value2member_map_objectanyr'   r>   tupler   rb   _name___objclass__r.   rY   r=   r    rJ   r   __new_member__r?   r@   rA   )rK   r&   rL   rS   ignorerF   rO   rk   save_newuse_argsenum_membersr   r3   invalid_names
enum_classdynamic_attributesmethodsmember_namer+   argsenum_membercanonical_memberclass_method
obj_methodenum_method)r2   )rS   rN   r   rk      s    













zEnumMeta.__new__c             C   s   dS )NTr   )r!   r   r   r   __bool__  s    zEnumMeta.__bool__Nr   )modulequalnametypestartc            C   s*   |d kr|  | |S | j||||||dS )N)r   r   r   r   )rk   _create_)r&   r+   namesr   r   r   r   r   r   r   __call__  s    zEnumMeta.__call__c             C   s6   t |ts dd l}|dtd t || o4|j| jkS )Nr   zHusing non-Enums in containment checks will raise TypeError in Python 3.8r   )r>   r   warningswarnDeprecationWarningrs   rm   )r&   memberr   r   r   r   __contains__:  s    
zEnumMeta.__contains__c                s(   || j krtd| j t | d S )Nz%s: cannot delete Enum member.)rm   AttributeErrorr(   r-   __delattr__)r&   attr)r2   r   r   r   C  s    
zEnumMeta.__delattr__c             C   s   ddddg| j  S )Nr2   rW   __members__r%   )rl   )r!   r   r   r   __dir__K  s    
zEnumMeta.__dir__c             C   s>   t |rt|y
| j| S  tk
r8   t|d Y nX d S )N)r   r   rm   KeyError)r&   r   r   r   r   __getattr__O  s    	
zEnumMeta.__getattr__c             C   s
   | j | S )N)rm   )r&   r   r   r   r   __getitem___  s    zEnumMeta.__getitem__c                s    fdd j D S )Nc             3   s   | ]} j | V  qd S )N)rm   )rQ   r   )r&   r   r   ra   c  s    z$EnumMeta.__iter__.<locals>.<genexpr>)rl   )r&   r   )r&   r   __iter__b  s    zEnumMeta.__iter__c             C   s
   t | jS )N)r   rl   )r&   r   r   r   __len__e  s    zEnumMeta.__len__c             C   s
   t | jS )N)r   rm   )r&   r   r   r   r   h  s    zEnumMeta.__members__c             C   s
   d| j  S )Nz	<enum %r>)r(   )r&   r   r   r   rc   r  s    zEnumMeta.__repr__c                s    fddt  jD S )Nc             3   s   | ]} j | V  qd S )N)rm   )rQ   r   )r&   r   r   ra   v  s    z(EnumMeta.__reversed__.<locals>.<genexpr>)reversedrl   )r&   r   )r&   r   __reversed__u  s    zEnumMeta.__reversed__c                s0   | j di }||krtdt || d S )Nrm   zCannot reassign members.)rX   getr   r-   __setattr__)r&   r   r+   
member_map)r2   r   r   r   x  s    zEnumMeta.__setattr__c         
   C   s  | j }|d kr| fn|| f}| |\}	}
|||}t|trP|dd }t|ttfr|rt|d tr|g  }}g }xDt	|D ]8\}}|

||||d d  }|| |||f qW x6|D ].}t|tr|||  }}n|\}}|||< qW |||||}|d krVytdjd }W n* tttfk
rT } zW d d }~X Y nX |d krjt| n||_|d k	r||_|S )Nr9   r:   r   r   r(   )r2   rI   rP   r>   r?   r@   rA   rr   rB   	enumerater5   rD   rk   sys	_getframe	f_globalsr   r<   r   r'   r%   r)   )r&   
class_namer   r   r   r   r   rK   rL   r   rO   rS   original_nameslast_valuescountr   r+   itemr~   member_valuer{   excr   r   r   r     s<    
 







zEnumMeta._create_c             C   sP   | st tfS dd }| d }t|ts.td|| p8t }|jrHtd||fS )Nc             S   sH   xB| D ]:}x4|j D ]*}|tkr"qqd|jkrt|tr8q|S qW qW d S )Nrk   )__mro__rp   rX   
issubclassr   )rL   chainbaser   r   r   _find_data_type  s    


z.EnumMeta._get_mixins_.<locals>._find_data_typer   zZnew enumerations should be created as `EnumName([mixin_type, ...] [data_type,] enum_type)`zCannot extend enumerations)rp   r   r   r    rl   )rL   r   rO   rN   r   r   r   rI     s    
zEnumMeta._get_mixins_c       	      C   s   |  dd }|d k	}|d krtxVdD ]H}x8||fD ],}t||d }|d d jtjtjhkr0|}P q0W |d k	r"P q"W tj}|tjkrd}nd}|||fS )Nrk   )ru   rk   FT)r   rJ   rk   rp   r   )	rS   rN   rO   rk   rw   methodpossibletargetrx   r   r   r   rh     s(    


zEnumMeta._find_new_)N)r(   r%   r)   classmethodrP   rk   r   r   r   r   r   r   r   r   r   propertyr   rc   r   r   r   staticmethodrI   rh   rH   r   r   )r2   r   r   {   s&   
 	
5 c               @   s~   e Zd Zdd Zdd Zedd Zdd Zd	d
 Zdd Z	dd Z
dd Zdd Zedd Zedd ZedddZdS )r   c          
   C   s   t || kr|S y
| j| S  tk
r.   Y n6 tk
rb   x | j D ]}|j|krH|S qHW Y nX yd }| |}W n* tk
r } z|}d }W d d }~X Y nX t	|| r|S t
d|| jf }|d kr|d kr|n|d krtd| j|f }||_|d S )Nz%r is not a valid %szDerror in %s._missing_: returned %r instead of None or a valid member)r   ro   r   r    rm   valuesrb   r6   	Exceptionr>   r<   r(   __context__)r&   r+   r   r   resulteve_excr   r   r   rk     s6    


zEnum.__new__c          	   C   s8   x2t |D ]"}y|d S  tk
r*   Y q
X q
W |S d S )Nr   )r   r    )r   r   r   r   
last_valuer   r   r   r5   6  s    
zEnum._generate_next_value_c             C   s   t d|| jf d S )Nz%r is not a valid %s)r<   r(   )r&   r+   r   r   r   r6   ?  s    zEnum._missing_c             C   s   d| j j| j| jf S )Nz<%s.%s: %r>)r2   r(   rs   rb   )r!   r   r   r   rc   C  s    zEnum.__repr__c             C   s   d| j j| jf S )Nz%s.%s)r2   r(   rs   )r!   r   r   r   rd   G  s    zEnum.__str__c                s&    fdd j  D }dddg| S )Nc                s2   g | ]*}|j D ]}|d  dkr| jkr|qqS )r   r   )rX   rm   )rQ   r&   r`   )r!   r   r   
<listcomp>L  s   
z Enum.__dir__.<locals>.<listcomp>r2   rW   r%   )r2   rU   )r!   added_behaviorr   )r!   r   r   J  s    
zEnum.__dir__c             C   s0   | j tkrt}t| }n| j }| j}|||S )N)rn   rp   r?   rb   re   )r!   format_specr&   valr   r   r   re   S  s    

zEnum.__format__c             C   s
   t | jS )N)hashrs   )r!   r   r   r   __hash__b  s    zEnum.__hash__c             C   s   | j | jffS )N)r2   rb   )r!   r"   r   r   r   r$   e  s    zEnum.__reduce_ex__c             C   s   | j S )N)rs   )r!   r   r   r   r   o  s    z	Enum.namec             C   s   | j S )N)rb   )r!   r   r   r   r+   t  s    z
Enum.valueNc                s   t tj| }rt n| fdd D }y|jdd d W n$ tk
rp   |jdd d Y nX | |||d} t| _|| j	 | ||< | S )Nc                s    g | ]} |r|| fqS r   r   )rQ   r   )filtersourcer   r   r     s   z!Enum._convert.<locals>.<listcomp>c             S   s   | d | d fS )Nr   r   r   )tr   r   r   <lambda>      zEnum._convert.<locals>.<lambda>)rF   c             S   s   | d S )Nr   r   )r   r   r   r   r     r   )r   )
varsr   moduleskeyssortr    _reduce_ex_by_namer$   updater   )r&   r   r   r   r   module_globalsmembersr   )r   r   r   _converty  s    

zEnum._convert)N)r(   r%   r)   rk   r5   r   r6   rc   rd   r   re   r   r$   r   r   r+   r   r   r   r   r   r     s   (		
)	metaclassc               @   s   e Zd ZdS )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 Zd Zdd Zedd Zedd Zdd Zd	d
 Zdd Z	dd Z
dd Zdd Zdd Zdd ZdS )r   c          	   C   sd   |s|d k	r|S dS xBt |D ]6}yt|}P W q tk
rR   td| d Y qX qW d|d  S )Nr   zInvalid Flag value: %rr   )r   	_high_bitr   r    )r   r   r   r   r   high_bitr   r   r   r5     s    	zFlag._generate_next_value_c             C   s.   |}|dk r| }|  |}|dk r*| }|S )Nr   )r4   )r&   r+   original_valuepossible_memberr   r   r   r6     s    
zFlag._missing_c             C   sb   | j |d }|d kr^t| |\}}|r:td|| jf t| }d |_||_| j 	||}|S )Nz%r is not a valid %s)
ro   r   
_decomposer<   r(   rp   rk   rs   rb   rf   )r&   r+   pseudo_memberr   extra_flagsr   r   r   r4     s    
zFlag._create_pseudo_member_c             C   s8   t || js&dd l}|dtd dS |j| j@ |jkS )Nr   zHusing non-Flags in containment checks will raise TypeError in Python 3.8r   F)r>   r2   r   r   r   rb   )r!   otherr   r   r   r   r     s    zFlag.__contains__c             C   sV   | j }| jd k	r$d|j| j| jf S t|| j\}}d|jddd |D | jf S )Nz<%s.%s: %r>|c             S   s   g | ]}t |jp|jqS r   )r?   rs   rb   )rQ   r`   r   r   r   r     s    z!Flag.__repr__.<locals>.<listcomp>)r2   rs   r(   rb   r   rj   )r!   r&   r   	uncoveredr   r   r   rc     s    
zFlag.__repr__c             C   s   | j }| jd k	r d|j| jf S t|| j\}}t|dkr^|d jd kr^d|j|d jf S d|jddd |D f S d S )Nz%s.%sr   r   z%s.%rr   c             S   s   g | ]}t |jp|jqS r   )r?   rs   rb   )rQ   r`   r   r   r   r     s    z Flag.__str__.<locals>.<listcomp>)r2   rs   r(   r   rb   r   rj   )r!   r&   r   r   r   r   r   rd     s    
zFlag.__str__c             C   s
   t | jS )N)boolrb   )r!   r   r   r   r     s    zFlag.__bool__c             C   s"   t || jstS | | j|jB S )N)r>   r2   NotImplementedrb   )r!   r   r   r   r   __or__  s    zFlag.__or__c             C   s"   t || jstS | | j|j@ S )N)r>   r2   r   rb   )r!   r   r   r   r   __and__  s    zFlag.__and__c             C   s"   t || jstS | | j|jA S )N)r>   r2   r   rb   )r!   r   r   r   r   __xor__  s    zFlag.__xor__c             C   sR   t | j| j\}}| d}x*| jD ] }||kr$|j| j@ s$||B }q$W | |S )Nr   )r   r2   rb   )r!   r   r   invertedr`   r   r   r   
__invert__  s    
zFlag.__invert__N)r(   r%   r)   r5   r   r6   r4   r   rc   rd   r   r   r   r   r   r   r   r   r   r     s   

c               @   sP   e Zd Zedd Zedd Zdd Zdd Zd	d
 ZeZ	eZ
eZdd ZdS )r	   c             C   s*   t |tstd|| jf | |}|S )Nz%r is not a valid %s)r>   intr<   r(   r4   )r&   r+   
new_memberr   r   r   r6     s    

zIntFlag._missing_c             C   s   | j |d }|d kr|g}t| |\}}xL|rvt|}d| }|| j kr\||kr\|| || krld}q,||N }q,W x6t|D ]*}t| |}d |_||_	| j 
||}qW |S )Nr   r   )ro   r   r   r   rD   r   r   rk   rs   rb   rf   )r&   r+   r   need_to_creater   r   bit
flag_valuer   r   r   r4      s&    


zIntFlag._create_pseudo_member_c             C   s0   t || jtfstS | | j| |jB }|S )N)r>   r2   r   r   rb   )r!   r   r   r   r   r   r   >  s    zIntFlag.__or__c             C   s,   t || jtfstS | | j| |j@ S )N)r>   r2   r   r   rb   )r!   r   r   r   r   r   D  s    zIntFlag.__and__c             C   s,   t || jtfstS | | j| |jA S )N)r>   r2   r   r   rb   )r!   r   r   r   r   r   I  s    zIntFlag.__xor__c             C   s   |  | j }|S )N)r2   rb   )r!   r   r   r   r   r   R  s    zIntFlag.__invert__N)r(   r%   r)   r   r6   r4   r   r   r   __ror____rand____rxor__r   r   r   r   r   r	     s   c             C   s   |   d S )Nr   )
bit_length)r+   r   r   r   r   W  s    r   c             C   sb   g }x0| j  D ]"\}}||jkr|||jf qW |r^ddd |D }td| |f | S )Nz, c             S   s   g | ]\}}d ||f qS )z%s -> %sr   )rQ   aliasr   r   r   r   r   c  s    zunique.<locals>.<listcomp>z duplicate values found in %r: %s)r   rY   r   rD   rj   r<   )enumeration
duplicatesr   r   alias_detailsr   r   r   r   [  s    
c             C   s   |}|dk }|r*dd t | j D }ndd t | j D }g }x4|D ],\}}|rL||@ |krL|| || M }qLW |s|| jkr|| j|  |jdd dd t|d	kr|d j|kr|d ||fS )
Nr   c             S   s"   g | ]\}}|j d k	r||fqS )N)r   )rQ   r[   r`   r   r   r   r   s  s   z_decompose.<locals>.<listcomp>c             S   s*   g | ]"\}}|j d k	st|r||fqS )N)r   _power_of_two)rQ   r[   r`   r   r   r   r   z  s   c             S   s   | j S )N)rb   )r`   r   r   r   r     r   z_decompose.<locals>.<lambda>T)rF   reverser   )rB   ro   rY   rD   r   r   r+   rg   )flagr+   not_coverednegativeflags_to_checkr   r   r   r   r   r   r   h  s$    

r   c             C   s   | dk rdS | dt |  kS )Nr   Fr   )r   )r+   r   r   r   r     s    r   )r   typesr   r   _collectionsr   ImportErrorcollections__all__r   r   r   r'   rp   r*   r
   dictr,   r   r   r   r   r   r   r   r	   r   r   r   r   r   r   r   r   <module>   s<   ?    qA#