Delete enums.py

danielgtaylor · Jul 12, 2020 · 907c91b · 907c91b
1 parent 83e47b1
commit 907c91b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 150 deletions.
diff --git a/betterproto/enums.py b/betterproto/enums.py
diff --git a/src/betterproto/enum.py b/src/betterproto/enum.py
@@ -45,7 +45,10 @@ def __new__(
         value_mapping: Dict[Any, EnumMember] = {}
         member_mapping: Dict[str, EnumMember] = {}
         member_names: List[str] = []
-        value_cls = IntEnumMember if int in bases else EnumMember
+        try:
+            value_cls = IntEnumMember if IntEnum in bases else EnumMember
+        except NameError:
+            value_cls = EnumMember
 
         for key, value in tuple(attrs.items()):
             is_descriptor = _is_descriptor(value)
@@ -165,4 +168,8 @@ def patched_instance_check(self: _EnumMeta, instance: Any) -> bool:
     return type.__instancecheck__(self, instance)
 
 
+class MyEnum(IntEnum):
+    pass
+
+
 _EnumMeta.__instancecheck__ = patched_instance_check  # fake it till you make it