Resolved: Create an Enum using a custom mix-in type

Question:

In the relevant section of the Python Enum documentation, it is mentioned that mix-in types can be used to ensure items belong to that type and act as objects of that type, like in the example:

class IntEnum(int, Enum):
    pass

However, the documentation does not give any examples on how to use a custom class as a mix-in type, and I’ve failed miserably at that. Is it even possible? And if so, how do I do it?
The code I’m trying to write is

@dataclass<br />
class Field:<br />
    dtype: str<br />
    name: str</p>
<p>@dataclass<br />
class FwfField(Field):<br />
    width: int</p>
<p>    def __post_init__(self):<br />
        if self.width <= 0:
            raise ValueError("Field width must be positive.")

class CnefeSchema(FwfField, Enum):
    state_code = ("int", "Código da UF", 2)
    ...
[/code]

</p>



When I import <code>CnefeSchema</code> from another file or run it, I get<br> 


<p>[code]---------------------------------------------------------------------------<br />
AttributeError                            Traceback (most recent call last)<br />
Input In [1], in <cell line: 1>()<br />
----> 1 from src.conf.schemas import CnefeSchema</p>
<p>File src/conf/schemas.py:25, in <module><br />
     21     pass<br />
     24 # TODO automatically read schema from Layout<br />
---> 25 class CnefeSchema(EnumFwfField, Enum):<br />
     27     @classmethod<br />
     28     def parse(cls) -> tuple[list, dict, dict]:<br />
     29         """<br />
     30         Get field widths, dtypes, and columns from schema.<br />
     31<br />
   (...)<br />
     34         as `pandas.read_fwf`.<br />
     35         """</p>
<p>File /usr/lib/python3.10/enum.py:298, in EnumMeta.__new__(metacls, cls, bases, classdict, **kwds)<br />
    296 enum_member._name_ = member_name<br />
    297 enum_member.__objclass__ = enum_class<br />
--> 298 enum_member.__init__(*args)<br />
    299 # If another member with the same value was already defined, the<br />
    300 # new member becomes an alias to the existing one.<br />
    301 for name, canonical_member in enum_class._member_map_.items():</p>
<p>File <string>:4, in __init__(self, dtype, name, width)</p>
<p>File /usr/lib/python3.10/types.py:187, in DynamicClassAttribute.__set__(self, instance, value)<br />
    185 def __set__(self, instance, value):<br />
    186     if self.fset is None:<br />
--> 187         raise AttributeError("can't set attribute")<br />
    188     self.fset(instance, value)</p>
<p>AttributeError: can't set attribute

I’ve already tried:

Defining another class, EnumFwfField, and using it as a mix-in, i.e.

class EnumFwfField(FwfField, Enum):
    pass

class CnefeSchema(EnumFwfField, Enum):
    ...

but I get the same error as above;

Setting the Enum fields (in this example, state_code) to FwfField, i.e.

class CnefeSchema(FwfField, Enum):
    state_code = FwfField("int", "Código da UF", 2)
    ...

but then I get

Traceback (most recent call last):
  File "src/conf/schemas.py", line 25, in <module>
    class CnefeSchema(EnumFwfField, Enum):
  File "/usr/lib/python3.10/enum.py", line 298, in __new__
    enum_member.__init__(*args)
TypeError: FwfField.__init__() missing 2 required positional arguments: 'name' and 'width'

Answer:

The issue you are having is because in your above code you have a name attribute in your Field dataclass, but Enum will not let you set a name attribute.
Rename that field to something else, for example 'dname', and it should work.
(The error message in 3.11 is more informative.)

If you have better answer, please add a comment about this, thank you!

Resolved: Create an Enum using a custom mix-in type

Resolved: linq2db throws exception when filtering by nested collection

Resolved: Table data is coming as empty in React

Resolved: In a Pinescript v5 Strategy why is my bool not working?

Leave A Reply