Add isinstance check for Enum in default resolver

This commit is contained in:
Kalle Jepsen 2017-10-04 11:48:01 +02:00
parent 2cc701f444
commit d58126facb

View File

@ -1,5 +1,11 @@
from .enum import PyEnum
def attr_resolver(attname, default_value, root, info, **args): def attr_resolver(attname, default_value, root, info, **args):
return getattr(root, attname, default_value) value = getattr(root, attname, default_value)
if isinstance(value, PyEnum):
return value.value
return value
def dict_resolver(attname, default_value, root, info, **args): def dict_resolver(attname, default_value, root, info, **args):