wip: python: Allow assignment of enum fields from enums with same name

python/test/test_scheme.py

@@ -10,6 +10,7 @@
 import copy
 import datetime
 import decimal
+import enum
 import pytest
 import struct
 
@@ -658,6 +659,16 @@ def test_enum():
     assert m.SCHEME['e8'].from_string('1') == m.e8.A
     assert m.SCHEME['e8'].from_string('A') == m.e8.A
 
+    class e8(enum.Enum):
+        A = 1
+        B = 2
+        Other = 30
+
+    with pytest.raises(ValueError): m.e64 = e8.A
+    with pytest.raises(ValueError): m.e8 = e8.Other
+    m.e8 = e8.A
+    assert m.e8.name == 'A'
+
 def test_enum_eq():
     SCHEME = """yamls://
 - enums:

python/tll/scheme.pyx

@@ -598,6 +598,12 @@ cdef class FEnum(FBase):
     cdef convert(FEnum self, v):
         if isinstance(v, str):
             return self.enum_class.__members__[v]
+        elif isinstance(v, enum.Enum):
+            if isinstance(v, self.enum_class):
+                return v
+            if self.enum_class.__name__ != v.__class__.__name__:
+                raise ValueError(f"Enum name of '{v}' does not match field enum name '{self.enum_class.__name__}'")
+            return self.enum_class(v.value)
         return self.enum_class(v)
     cdef from_string(FEnum self, str s):
         v = self.enum_class.__members__.get(s, None)