graphql-python/graphene
1
1
Fork 0
mirror of https://github.com/graphql-python/graphene.git synced 2025-02-18 12:30:37 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fixed Python 3.5 issues with Enum

Browse Source
This commit is contained in:
Syrus Akbary 2017-07-12 21:41:57 -07:00
parent 6321c52bd2
commit 3fbc3281a2
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
graphene/types/enum.py
View File

@ -27,6 +27,10 @@ class EnumOptions(BaseOptions):
class EnumMeta(SubclassWithMeta_Meta): class EnumMeta(SubclassWithMeta_Meta):
def __new__(cls, name, bases, classdict, **options):
enum = PyEnum(cls.__name__, OrderedDict(classdict, __eq__=eq_enum))
return SubclassWithMeta_Meta.__new__(cls, name, bases, OrderedDict(classdict, __enum__=enum), **options)
def get(cls, value): def get(cls, value):
return cls._meta.enum(value) return cls._meta.enum(value)
@ -52,7 +56,7 @@ class Enum(six.with_metaclass(EnumMeta, UnmountedType, BaseType)):
@classmethod @classmethod
def __init_subclass_with_meta__(cls, enum=None, **options): def __init_subclass_with_meta__(cls, enum=None, **options):
_meta = EnumOptions(cls) _meta = EnumOptions(cls)
_meta.enum = enum or PyEnum(cls.__name__, OrderedDict(cls.__dict__, __eq__=eq_enum)) _meta.enum = enum or cls.__enum__
for key, value in _meta.enum.__members__.items(): for key, value in _meta.enum.__members__.items():
setattr(cls, key, value) setattr(cls, key, value)
super(Enum, cls).__init_subclass_with_meta__(_meta=_meta, **options) super(Enum, cls).__init_subclass_with_meta__(_meta=_meta, **options)