daviddoji/example-code-2e
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ch25: changed MicroEnum to use __class_getitem__

Browse Source
This commit is contained in:
Luciano Ramalho
2021-04-08 14:43:15 -03:00
parent c6a1da4882
commit bbc664308a
2 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
25-class-metaprog/tinyenums/microenum.py
View File

@@ -54,12 +54,10 @@ class MicroEnumMeta(type):
def __prepare__(name, bases, **kwargs): def __prepare__(name, bases, **kwargs):
return AutoFillDict() return AutoFillDict()
def __getitem__(cls, key):
class MicroEnum(metaclass=MicroEnumMeta):
def __class_getitem__(cls, key):
for k, v in cls.__dict__.items(): for k, v in cls.__dict__.items():
if v == key: if v == key:
return k return k
raise KeyError(key) raise KeyError(key)
class MicroEnum(metaclass=MicroEnumMeta):
pass

1
25-class-metaprog/tinyenums/nanoenum.py
View File

@@ -25,7 +25,6 @@ Testing ``KeyIsValueDict``::
class KeyIsValueDict(dict): class KeyIsValueDict(dict):
def __missing__(self, key): def __missing__(self, key):
if key.startswith('__') and key.endswith('__'): if key.startswith('__') and key.endswith('__'):
raise KeyError(key) raise KeyError(key)