26 lines
591 B
Python
26 lines
591 B
Python
from cpl_cli.abc.generate_schematic_abc import GenerateSchematicABC
|
|
|
|
|
|
class Enum(GenerateSchematicABC):
|
|
def __init__(self, *args: str):
|
|
GenerateSchematicABC.__init__(self, *args)
|
|
|
|
def get_code(self) -> str:
|
|
import textwrap
|
|
|
|
code = textwrap.dedent(
|
|
"""\
|
|
from enum import Enum
|
|
|
|
|
|
class $Name(Enum):
|
|
|
|
atr = 0
|
|
"""
|
|
)
|
|
return self.build_code_str(code, Name=self._class_name)
|
|
|
|
@classmethod
|
|
def register(cls):
|
|
GenerateSchematicABC.register(cls, "enum", ["e", "E"])
|