Home > other >  Python dynamic enum
Python dynamic enum

Time:06-09

I would like to create dynamic enums in python loaded from a SQL Table. The output of SQL will be a list of tuplets, which with I want to fill the attributes of the enum.

Lets say I receive this list:

lst = [('PROCESS_0', 0, "value", 123, False), ('PROCESS_1',1,"anothervalue", 456, True)]

I now want to fill the values in the enum below:

class Jobs(IntEnum):
    def __new__(cls, value: int, label: str, heartbeat: int = 60, heartbeat_required: bool = False):
        obj = int.__new__(cls, value)
        obj._value_ = value
        obj.label = label
        obj.heartbeat = heartbeat
        obj.heartbeat_required = heartbeat_required
        return obj

The first variable in the tuple should be the variable name of the enum, I have solved this with:

locals()['Test'] = (0, '', 789, False)

But this only works for single values, it seems that I can not run a for loop within enum. When using a for loop like this:

 for i in lst:
    locals()[i[0]] = (i[1], i[2], i[3])

Python sends this error TypeError: Attempted to reuse key: 'i' which propably comes from enums only having constants.

Is there any (possibly elegant) solution for this?

Many thanks in advance!

CodePudding user response:

You need to use _ignore_ = "i". Something like:

class Jobs(IntEnum):

    _ignore_ = "i"

    def __new__(cls, value, label, heartbeat=60, heartbeat_required=False):
        obj = int.__new__(cls, value)
        obj._value_ = value
        obj.label = label
        obj.heartbeat = heartbeat
        obj.heartbeat_required = heartbeat_required
        return obj

    for i in lst:
        locals()[i[0]] = i[1:]
  • Related