Initial Commit

custom_components/powercalc/strategy/selector.py

@@ -0,0 +1,46 @@
+from homeassistant.helpers.typing import ConfigType
+
+from custom_components.powercalc.const import (
+    CONF_COMPOSITE,
+    CONF_FIXED,
+    CONF_LINEAR,
+    CONF_LUT,
+    CONF_MODE,
+    CONF_MULTI_SWITCH,
+    CONF_PLAYBOOK,
+    CONF_WLED,
+    CalculationStrategy,
+)
+from custom_components.powercalc.errors import UnsupportedStrategyError
+from custom_components.powercalc.power_profile.power_profile import PowerProfile
+
+STRATEGY_CONFIG_MAP = {
+    CONF_LINEAR: CalculationStrategy.LINEAR,
+    CONF_FIXED: CalculationStrategy.FIXED,
+    CONF_MULTI_SWITCH: CalculationStrategy.MULTI_SWITCH,
+    CONF_PLAYBOOK: CalculationStrategy.PLAYBOOK,
+    CONF_WLED: CalculationStrategy.WLED,
+    CONF_LUT: CalculationStrategy.LUT,
+    CONF_COMPOSITE: CalculationStrategy.COMPOSITE,
+}
+
+
+def detect_calculation_strategy(
+    config: ConfigType,
+    power_profile: PowerProfile | None,
+) -> CalculationStrategy:
+    """Select the calculation strategy."""
+    config_mode = config.get(CONF_MODE)
+    if config_mode:
+        return CalculationStrategy(config_mode)
+
+    for config_key, strategy in STRATEGY_CONFIG_MAP.items():
+        if config_key in config:
+            return strategy
+
+    if power_profile:
+        return power_profile.calculation_strategy
+
+    raise UnsupportedStrategyError(
+        "Cannot select a strategy, supply it in the config. See the readme",
+    )