WV@sddlZddlmZddlmZmZdddgZddZd d Zd d Z d dZ Gddde Z dZ GdddeZGddddeZ Gdddee ZddZddZdS)N) OrderedDict)MappingProxyTypeDynamicClassAttributeEnumIntEnumuniquecCs+t|dp*t|dp*t|dS)z5Returns True if obj is a descriptor, False otherwise.__get____set__ __delete__)hasattr)objr 9/afs/.cs/s/python-3.5.2/amd64_ubu14/lib/python3.5/enum.py_is_descriptorsrcCsl|dd|ddko+dknok|dddkok|dd dkokt|dkS) z3Returns True if a __dunder__ name, False otherwise.N___r)len)namer r r _is_dunders0rcCs`|d|dkodkno_|dddko_|dddko_t|dkS)z1Returns True if a _sunder_ name, False otherwise.rrrrr)r)rr r r _is_sunders$rcCs"dd}||_d|_dS)z"Make the given class un-picklable.cSstd|dS)Nz%r cannot be pickled) TypeError)selfprotor r r_break_on_call_reduce"sz6_make_class_unpicklable.._break_on_call_reducez N) __reduce_ex__ __module__)clsr r r r_make_class_unpicklable s  r$cs:eZdZdZfddZfddZS) _EnumDictzTrack 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. cstjg|_dS)N)super__init__ _member_names)r) __class__r rr'/s z_EnumDict.__init__cst|rtdnmt|r*n^||jkrLtd|n<t|s||krxtd|||jj|tj||dS)zChanges 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. z(_names_ are reserved for future Enum usezAttempted to reuse key: %rzKey already defined as: %rN) r ValueErrorrr(rrappendr& __setitem__)rkeyvalue)r)r rr,3s    z_EnumDict.__setitem__)__name__r" __qualname____doc__r'r,r r )r)rr%(s r%c sNeZdZdZeddZfddZddZdd dd dd dd d ddZddZ fddZ ddZ ddZ ddZ ddZddZeddZd d!Zd"d#Zfd$d%Zdd dd dd dd d d&d'Zed(d)Zed*d+ZS),EnumMetazMetaclass for EnumcCstS)N)r%)metaclsr#basesr r r __prepare__TszEnumMeta.__prepare__c s^|j|\}|j|\}}}fddjD}xjD] } | =qYWt|dh@} | rtdjdj| dkrddcs z$EnumMeta.__new__..mrozInvalid enum member name: {0},r1zAn enumeration.cSs&h|]}|jD] }|qqSr )__dict__)r6bar r r ys z#EnumMeta.__new__..r!__getnewargs_ex____getnewargs__ __reduce__c3s|]}|jkVqdS)N)r<)r6m) member_typer r sz#EnumMeta.__new__.._value___repr____str__ __format__)z__getnewargs_ex__z__getnewargs__ __reduce_ex__z __reduce__)z__repr__z__str__z __format__rJ) _get_mixins_ _find_new_r(setr*formatjoinr&__new___member_names_r _member_map_ _member_type_r:_value2member_map_objectanyr$ isinstancetupler rF_name_ __objclass__r'itemsr+setattrrgetattrr__new_member__)r3r#r4r8 first_enumrPsave_newuse_argsmembersr invalid_names enum_classbase_attributesmethods member_namer.args enum_membercanonical_member class_method obj_method enum_method)r))r8rDrrPXs                             zEnumMeta.__new__cCsdS)z6 classes/types should always be True. Tr )rr r r__bool__szEnumMeta.__bool__Nmodulequalnametypestartrc CsD|dkr|j||S|j||d|d|d|d|S)aEither 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. Nrorprqrr)rP_create_)r#r.namesrorprqrrr r r__call__s zEnumMeta.__call__cCst||o|j|jkS)N)rWrYrR)r#memberr r r __contains__szEnumMeta.__contains__cs6||jkr"td|jtj|dS)Nz%s: cannot delete Enum member.)rRAttributeErrorr/r& __delattr__)r#attr)r)r rryszEnumMeta.__delattr__cCsddddg|jS)Nr)r1 __members__r")rQ)rr r r__dir__szEnumMeta.__dir__c CsOt|rt|y|j|SWn!tk rJt|dYnXdS)a5Return 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)rrxrRKeyError)r#rr r r __getattr__s   zEnumMeta.__getattr__cCs |j|S)N)rR)r#rr r r __getitem__szEnumMeta.__getitem__csfddjDS)Nc3s|]}j|VqdS)N)rR)r6r)r#r rrEsz$EnumMeta.__iter__..)rQ)r#r )r#r__iter__szEnumMeta.__iter__cCs t|jS)N)rrQ)r#r r r__len__szEnumMeta.__len__cCs t|jS)zReturns 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. )rrR)r#r r rr{szEnumMeta.__members__cCs d|jS)Nz )r/)r#r r rrG'szEnumMeta.__repr__cs fddtjDS)Nc3s|]}j|VqdS)N)rR)r6r)r#r rrE+sz(EnumMeta.__reversed__..)reversedrQ)r#r )r#r __reversed__*szEnumMeta.__reversed__csD|jjdi}||kr-tdtj||dS)zBlock 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. rRzCannot reassign members.N)r<getrxr& __setattr__)r#rr. member_map)r)r rr-s  zEnumMeta.__setattr__cCs|j}|dkr|fn ||f}|j||} t|trc|jddj}t|ttfrt|dtrddt||D}xG|D]?} t| tr| || } } n | \} } | | | value pairs. Nr; rcSs"g|]\}}||fqSr r )r6ier r r Ns z%EnumMeta._create_..rr/)r)r5rWstrreplacesplitrXlist enumeraterPsys _getframe f_globalsrxr*r$r"r0)r# class_namertrorprqrrr3r4r8itemrg member_valuerdexcr r rrs:s0 !(        zEnumMeta._create_cCs|sttfSd}}x>|D]6}|tk r!t|tr!|jr!tdq!Wt|tsvtdt|dts|d}|d}nNxK|djD]<}t|tr|dkr|}q|dkr|}qW||fS)zReturns the type for creating enum members, and the first inherited enum class. bases: the tuple of bases that was given to __new__ NzCannot extend enumerationszHnew enumerations must be created as `ClassName([mixin_type,] enum_type)`rrr)rUr issubclassrQr__mro__)r4rDr_baser r rrKis(            zEnumMeta._get_mixins_c Cs|jdd}|dk }|dkrxudD]d}xN||fD]@}t||d}|ddjtjtjhkrD|}PqDW|dk r1Pq1Wtj}|tjkrd}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__ rPNr^FT)z__new_member__z__new__)rr]rPrUr) r8rDr_rPr`methodpossibletargetrar r rrLs(       zEnumMeta._find_new_)r/r"r0r1 classmethodr5rPrnrurwryr|r~rrrpropertyr{rGrrrs staticmethodrKrLr r )r)rr2Rs& y '         '/-r2c@seZdZdZddZddZddZdd Zd d Zd d Z ddZ e ddZ e ddZ edddZdS)rzRGeneric enumeration. Derive from this class to define new enumerations. c Cst||kr|Sy||jkr3|j|SWn?tk rux*|jjD]}|j|krT|SqTWYnXtd||jfdS)Nz%r is not a valid %s)rqrTrrRvaluesrFr*r/)r#r.rvr r rrPs  z Enum.__new__cCsd|jj|j|jfS)Nz <%s.%s: %r>)r)r/rYrF)rr r rrGsz Enum.__repr__cCsd|jj|jfS)Nz%s.%s)r)r/rY)rr r rrHsz Enum.__str__cs3fddjjD}dddg|S)NcsEg|];}|jD]+}|ddkr|jkr|qqS)rr)r<rR)r6r#rC)rr rrs  z Enum.__dir__..r)r1r")r)r:)radded_behaviorr )rrr|sz Enum.__dir__cCsF|jtkr$t}t|}n|j}|j}|j||S)N)rSrUrrFrI)r format_specr#valr r rrIs   zEnum.__format__cCs t|jS)N)hashrY)rr r r__hash__sz Enum.__hash__cCs|j|jffS)N)r)rF)rrr r rr!szEnum.__reduce_ex__cCs|jS)zThe name of the Enum member.)rY)rr r rr sz Enum.namecCs|jS)zThe value of the Enum member.)rF)rr r rr.sz Enum.valueNcsttj|}|r(t|}n|}fdd|jD}|||d|}t|_|j|j|||<|S)z[ Create a new Enum subclass that replaces a collection of global constants cs+i|]!\}}|r||qSr r )r6rr.)filterr rr9#s z!Enum._convert..ro)varsrmodulesr[_reduce_ex_by_namer!updater{)r#rrorsourcemodule_globalsrbr )rr_converts   z Enum._convert)r/r"r0r1rPrGrHr|rIrr!rrr.rrr r r rrs        metaclassc@seZdZdZdS)rz.Enum where members are also (and must be) intsN)r/r"r0r1r r r rr,s cCs|jS)N)r)rrr r rr0srcCsg}xB|jjD]1\}}||jkr|j||jfqW|rdjdd|D}td||f|S)z?Class decorator for enumerations ensuring unique member values.z, cSs&g|]\}}d||fqS)z%s -> %sr )r6aliasrr r rr;s zunique..z duplicate values found in %r: %s)r{r[rr+rOr*) enumeration duplicatesrrv alias_detailsr r rr3s)r collectionsrtypesrr__all__rrrr$dictr%rrqr2intrrrr r r rs     'tg