pyhOn/pyhon/parameter/base.py

98 lines
2.9 KiB
Python
Raw Normal View History

2023-05-21 02:25:43 +02:00
from typing import Dict, Any, List, Tuple, Callable, TYPE_CHECKING
if TYPE_CHECKING:
from pyhon.rules import HonRule
2023-04-16 01:43:37 +02:00
class HonParameter:
2023-05-06 16:07:28 +02:00
def __init__(self, key: str, attributes: Dict[str, Any], group: str) -> None:
2023-04-16 01:43:37 +02:00
self._key = key
2023-07-12 00:05:27 +02:00
self._attributes = attributes
self._category: str = ""
self._typology: str = ""
self._mandatory: int = 0
2023-04-16 01:43:37 +02:00
self._value: str | float = ""
2023-05-06 16:07:28 +02:00
self._group: str = group
2023-06-28 19:02:11 +02:00
self._triggers: Dict[
str, List[Tuple[Callable[["HonRule"], None], "HonRule"]]
] = {}
2023-07-12 00:05:27 +02:00
self._set_attributes()
2023-07-12 19:36:32 +02:00
def _set_attributes(self) -> None:
2023-07-12 00:05:27 +02:00
self._category = self._attributes.get("category", "")
self._typology = self._attributes.get("typology", "")
self._mandatory = self._attributes.get("mandatory", 0)
2023-04-16 01:43:37 +02:00
@property
def key(self) -> str:
return self._key
@property
def value(self) -> str | float:
return self._value if self._value is not None else "0"
2023-05-28 06:17:43 +02:00
@value.setter
def value(self, value: str | float) -> None:
self._value = value
self.check_trigger(value)
2023-05-22 01:07:55 +02:00
@property
2023-06-09 02:09:20 +02:00
def intern_value(self) -> str:
return str(self.value)
2023-05-22 01:07:55 +02:00
2023-04-24 04:33:00 +02:00
@property
def values(self) -> List[str]:
2023-05-06 16:07:28 +02:00
return [str(self.value)]
2023-04-24 04:33:00 +02:00
2023-04-16 01:43:37 +02:00
@property
def category(self) -> str:
return self._category
@property
def typology(self) -> str:
return self._typology
@property
def mandatory(self) -> int:
return self._mandatory
2023-05-06 16:07:28 +02:00
@property
def group(self) -> str:
return self._group
2023-05-21 02:25:43 +02:00
2023-06-28 19:02:11 +02:00
def add_trigger(
2023-07-12 19:36:32 +02:00
self, value: str, func: Callable[["HonRule"], None], data: "HonRule"
2023-06-28 19:02:11 +02:00
) -> None:
2023-05-21 02:25:43 +02:00
if self._value == value:
func(data)
self._triggers.setdefault(value, []).append((func, data))
2023-06-28 19:02:11 +02:00
def check_trigger(self, value: str | float) -> None:
2023-05-21 02:25:43 +02:00
if str(value) in self._triggers:
for trigger in self._triggers[str(value)]:
func, args = trigger
func(args)
@property
2023-06-28 19:02:11 +02:00
def triggers(self) -> Dict[str, Any]:
result: Dict[str, Any] = {}
2023-05-21 02:25:43 +02:00
for value, rules in self._triggers.items():
2023-06-10 06:41:37 +02:00
for _, rule in rules:
if rule.extras:
param = result.setdefault(value, {})
for extra_key, extra_value in rule.extras.items():
param = param.setdefault(extra_key, {}).setdefault(
extra_value, {}
)
else:
param = result.setdefault(value, {})
if fixed_value := rule.param_data.get("fixedValue"):
param[rule.param_key] = fixed_value
else:
param[rule.param_key] = rule.param_data.get("defaultValue", "")
2023-05-21 02:25:43 +02:00
return result
2023-07-12 00:05:27 +02:00
2023-07-12 19:36:32 +02:00
def reset(self) -> None:
2023-07-12 00:05:27 +02:00
self._set_attributes()