mirror of
https://github.com/moparisthebest/SickRage
synced 2024-11-14 13:25:11 -05:00
27 lines
801 B
Python
27 lines
801 B
Python
|
def Enum(field, enum, key_func=None):
|
||
|
|
"""
|
||
|
|
Enum is an adapter to another field: it will just change its display
|
||
|
|
attribute. It uses a dictionnary to associate a value to another.
|
||
|
|
|
||
|
|
key_func is an optional function with prototype "def func(key)->key"
|
||
|
|
which is called to transform key.
|
||
|
|
"""
|
||
|
|
display = field.createDisplay
|
||
|
|
if key_func:
|
||
|
|
def createDisplay():
|
||
|
|
try:
|
||
|
|
key = key_func(field.value)
|
||
|
|
return enum[key]
|
||
|
|
except LookupError:
|
||
|
|
return display()
|
||
|
|
else:
|
||
|
|
def createDisplay():
|
||
|
|
try:
|
||
|
|
return enum[field.value]
|
||
|
|
except LookupError:
|
||
|
|
return display()
|
||
|
|
field.createDisplay = createDisplay
|
||
|
|
field.getEnum = lambda: enum
|
||
|
|
return field
|
||
|
|
|