pyhOn/pyhon/parameter/enum.py

52 lines
1.7 KiB
Python
Raw Normal View History

2023-04-16 01:43:37 +02:00
from typing import Dict, Any, List
from pyhon.parameter.base import HonParameter
2023-06-28 19:02:11 +02:00
def clean_value(value: str | float) -> str:
2023-05-22 01:07:55 +02:00
return str(value).strip("[]").replace("|", "_").lower()
2023-04-16 01:43:37 +02:00
class HonParameterEnum(HonParameter):
2023-05-06 16:07:28 +02:00
def __init__(self, key: str, attributes: Dict[str, Any], group: str) -> None:
super().__init__(key, attributes, group)
2023-07-12 19:36:32 +02:00
self._default: str | float = ""
self._value: str | float = ""
2023-07-12 00:05:27 +02:00
self._values: List[str] = []
self._set_attributes()
2023-05-22 01:07:55 +02:00
if self._default and clean_value(self._default.strip("[]")) not in self.values:
2023-05-15 19:29:42 +02:00
self._values.append(self._default)
2023-04-16 01:43:37 +02:00
2023-07-12 19:36:32 +02:00
def _set_attributes(self) -> None:
2023-07-12 00:05:27 +02:00
super()._set_attributes()
2023-07-12 19:36:32 +02:00
self._default = self._attributes.get("defaultValue", "")
2023-07-12 00:05:27 +02:00
self._value = self._default or "0"
2023-07-12 19:36:32 +02:00
self._values = self._attributes.get("enumValues", [])
2023-07-12 00:05:27 +02:00
2023-04-16 01:43:37 +02:00
def __repr__(self) -> str:
return f"{self.__class__} (<{self.key}> {self.values})"
@property
def values(self) -> List[str]:
2023-05-22 01:07:55 +02:00
return [clean_value(value) for value in self._values]
2023-04-16 01:43:37 +02:00
2023-05-21 02:25:43 +02:00
@values.setter
2023-06-28 19:02:11 +02:00
def values(self, values: List[str]) -> None:
2023-05-21 02:25:43 +02:00
self._values = values
2023-06-09 02:09:20 +02:00
@property
def intern_value(self) -> str:
return str(self._value) if self._value is not None else str(self.values[0])
2023-04-16 01:43:37 +02:00
@property
def value(self) -> str | float:
2023-05-22 01:07:55 +02:00
return clean_value(self._value) if self._value is not None else self.values[0]
2023-04-16 01:43:37 +02:00
@value.setter
def value(self, value: str) -> None:
if value in self.values:
self._value = value
2023-05-21 02:25:43 +02:00
self.check_trigger(value)
2023-04-16 01:43:37 +02:00
else:
2023-07-01 14:31:37 +02:00
raise ValueError(f"Allowed values: {self._values} But was: {value}")