diff --git a/mo/db.py b/mo/db.py index bf4bc21bef92a25a857cf088920e911027a5e1a3..92ee8d0e3bd0c95b73fd4df0fc4857e80cc786bc 100644 --- a/mo/db.py +++ b/mo/db.py @@ -55,12 +55,15 @@ class MOEnum(str, PythonEnum): return name @classmethod - def choices(enum): + def choices(enum) -> List[Tuple[str, str]]: out = [] for item in enum: out.append((item.name, item.friendly_name())) return out + def friendly_name(self) -> str: + return str(self) + @classmethod def coerce(enum, name): if isinstance(name, enum):