Python how to list an array and make them an Enum

Question

So I'm trying to achieve something like this

from enum import Enum

tabulate_formats = ['fancy_grid', 'fancy_outline', 'github', 'grid']

class TableFormat(str, Enum):
    for item in tabulate_formats:
        exec(f"{item} = '{item}'")

Though i get this error

Traceback (most recent call last):
  File "/app/src/main.py", line 25, in <module>
    class TableFormat(str, Enum):
  File "/app/src/main.py", line 26, in TableFormat
    for item in tabulate_formats:
  File "/usr/local/lib/python3.6/enum.py", line 92, in __setitem__
    raise TypeError('Attempted to reuse key: %r' % key)
TypeError: Attempted to reuse key: 'item'

How do I properly assign them into the class

Tom · Accepted Answer · 2021-10-21 01:42:04Z

1

Not sure if this would also be acceptable, but based on turning a dict into an Enum, you could do:

TableFormat = Enum('TableFormat', {i:i for i in tabulate_formats})

answered Oct 21, 2021 at 1:42

Tom

8,8402 gold badges20 silver badges44 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Ethan Furman · Accepted Answer · 2021-10-21 01:35:54Z

1

class StrEnum(str, Enum):
    @staticmethod
    def _generate_next_value_(name, *args):
        return name

TableFormat = StrEnum('TableFormat', tabulate_formats)

and

>>> list(TableFormat)
[<TableFormat.fancy_grid: 'fancy_grid'>, <TableFormat.fancy_outline: 'fancy_outline'>, <TableFormat.github: 'github'>, <TableFormat.grid: 'grid'>]

answered Oct 21, 2021 at 1:35

Ethan Furman

70.1k21 gold badges174 silver badges251 bronze badges

Collectives™ on Stack Overflow

Python how to list an array and make them an Enum

2 Answers 2

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related