Add dict_or_attr resolver and set it as default

This commit is contained in:
Jonathan Kim 2018-01-05 17:44:12 +00:00 committed by Jonathan Kim
parent d728b84e48
commit a78d2fe0ad

View File

@ -6,7 +6,13 @@ def dict_resolver(attname, default_value, root, info, **args):
return root.get(attname, default_value)
default_resolver = attr_resolver
def dict_or_attr_resolver(attname, default_value, root, info, **args):
if isinstance(root, dict):
return dict_resolver(attname, default_value, root, info, **args)
return attr_resolver(attname, default_value, root, info, **args)
default_resolver = dict_or_attr_resolver
def set_default_resolver(resolver):