emhass 0.11.4__py3-none-any.whl → 0.15.5__py3-none-any.whl

emhass/connection_manager.py

@@ -0,0 +1,108 @@
+import asyncio
+import logging
+
+from emhass.websocket_client import AsyncWebSocketClient, ConnectionError
+
+_global_client: AsyncWebSocketClient | None = None
+_lock = asyncio.Lock()
+
+
+async def _create_and_start_client(
+    hass_url: str, token: str, logger: logging.Logger | None
+) -> AsyncWebSocketClient:
+    """Helper to create and start a new WebSocket client."""
+    if logger:
+        logger.debug(f"Creating new WebSocket client for {hass_url}")
+    client = AsyncWebSocketClient(hass_url=hass_url, long_lived_token=token, logger=logger)
+    try:
+        await asyncio.wait_for(client.startup(), timeout=15.0)
+        if logger:
+            logger.info("WebSocket client started successfully")
+        return client
+    except TimeoutError:
+        if logger:
+            logger.error("WebSocket startup timed out")
+        raise ConnectionError("WebSocket startup timed out") from None
+    except Exception as e:
+        if logger:
+            logger.error(f"WebSocket startup failed: {e}")
+        raise
+
+
+async def _reconnect_client(client: AsyncWebSocketClient, logger: logging.Logger | None) -> None:
+    """Helper to reconnect an existing WebSocket client."""
+    if logger:
+        logger.debug("WebSocket client exists but not connected, attempting reconnect")
+    try:
+        await asyncio.wait_for(client.reconnect(), timeout=15.0)
+        if logger:
+            logger.info("WebSocket client reconnected successfully")
+    except TimeoutError:
+        if logger:
+            logger.error("WebSocket reconnect timed out")
+        raise ConnectionError("WebSocket reconnect timed out") from None
+    except Exception as e:
+        if logger:
+            logger.error(f"WebSocket reconnect failed: {e}")
+        raise
+
+
+async def get_websocket_client(
+    hass_url: str, token: str, logger: logging.Logger | None = None
+) -> AsyncWebSocketClient:
+    """
+    Get or create a global WebSocket client connection.
+
+    :param hass_url: The Home Assistant URL
+    :type hass_url: str
+    :param token: Long-lived token for authentication
+    :type token: str
+    :param logger: Logger instance for logging
+    :type logger: logging.Logger | None
+    :return: Connected AsyncWebSocketClient instance
+    :rtype: AsyncWebSocketClient
+    :raises ConnectionError: If connection cannot be established
+    """
+    global _global_client
+    async with _lock:
+        if _global_client is None:
+            # Logic extracted to helper
+            _global_client = await _create_and_start_client(hass_url, token, logger)
+        elif not _global_client.connected:
+            # Logic extracted to helper
+            try:
+                await _reconnect_client(_global_client, logger)
+            except Exception:
+                # Ensure global is reset if reconnect fails
+                _global_client = None
+                raise
+        elif logger:
+            logger.debug("Using existing connected WebSocket client")
+    return _global_client
+
+
+async def close_global_connection():
+    """
+    Close the global WebSocket connection if it exists.
+    """
+    global _global_client
+
+    async with _lock:
+        if _global_client is not None:
+            try:
+                await _global_client.shutdown()
+            except Exception:
+                # Ignore errors during shutdown
+                pass
+            finally:
+                _global_client = None
+
+
+def is_connected() -> bool:
+    """
+    Check if the global WebSocket client is connected.
+
+    :return: True if connected, False otherwise
+    :rtype: bool
+    """
+    return _global_client is not None and _global_client.connected

emhass/data/associations.csv

@@ -1,23 +1,41 @@
-config_categorie,legacy_parameter_name,parameter,list_name 
+config_categorie,legacy_parameter_name,parameter,list_name
 retrieve_hass_conf,freq,optimization_time_step
 retrieve_hass_conf,days_to_retrieve,historic_days_to_retrieve
 retrieve_hass_conf,var_PV,sensor_power_photovoltaics
+retrieve_hass_conf,var_PV_forecast,sensor_power_photovoltaics_forecast
 retrieve_hass_conf,var_load,sensor_power_load_no_var_loads
 retrieve_hass_conf,load_negative,load_negative
 retrieve_hass_conf,set_zero_min,set_zero_min
 retrieve_hass_conf,var_replace_zero,sensor_replace_zero,list_sensor_replace_zero
 retrieve_hass_conf,var_interp,sensor_linear_interp,list_sensor_linear_interp
+retrieve_hass_conf,use_influxdb,use_influxdb
+retrieve_hass_conf,influxdb_host,influxdb_host
+retrieve_hass_conf,influxdb_port,influxdb_port
+retrieve_hass_conf,influxdb_username,influxdb_username
+retrieve_hass_conf,influxdb_password,influxdb_password
+retrieve_hass_conf,influxdb_database,influxdb_database
+retrieve_hass_conf,influxdb_measurement,influxdb_measurement
+retrieve_hass_conf,influxdb_retention_policy,influxdb_retention_policy
+retrieve_hass_conf,influxdb_use_ssl,influxdb_use_ssl
+retrieve_hass_conf,influxdb_verify_ssl,influxdb_verify_ssl
 retrieve_hass_conf,method_ts_round,method_ts_round
 retrieve_hass_conf,continual_publish,continual_publish
+retrieve_hass_conf,use_websocket,use_websocket
 params_secrets,time_zone,time_zone
 params_secrets,lat,Latitude
 params_secrets,lon,Longitude
 params_secrets,alt,Altitude
 optim_conf,costfun,costfun
 optim_conf,logging_level,logging_level
+optim_conf,set_use_pv,set_use_pv
+optim_conf,set_use_adjusted_pv,set_use_adjusted_pv
+optim_conf,adjusted_pv_regression_model,adjusted_pv_regression_model
+optim_conf,adjusted_pv_solar_elevation_threshold,adjusted_pv_solar_elevation_threshold
+optim_conf,adjusted_pv_model_max_age,adjusted_pv_model_max_age
 optim_conf,set_use_battery,set_use_battery
 optim_conf,num_def_loads,number_of_deferrable_loads
 optim_conf,P_deferrable_nom,nominal_power_of_deferrable_loads,list_nominal_power_of_deferrable_loads
+optim_conf,minimum_power_of_deferrable_loads,minimum_power_of_deferrable_loads
 optim_conf,def_total_hours,operating_hours_of_each_deferrable_load,list_operating_hours_of_each_deferrable_load
 optim_conf,treat_def_as_semi_cont,treat_deferrable_load_as_semi_cont,list_treat_deferrable_load_as_semi_cont
 optim_conf,set_def_constant,set_deferrable_load_single_constant,list_set_deferrable_load_single_constant
@@ -32,6 +50,8 @@ optim_conf,prod_sell_price,photovoltaic_production_sell_price
 optim_conf,set_total_pv_sell,set_total_pv_sell
 optim_conf,lp_solver,lp_solver
 optim_conf,lp_solver_path,lp_solver_path
+optim_conf,lp_solver_timeout,lp_solver_timeout
+optim_conf,num_threads,num_threads
 optim_conf,set_nocharge_from_grid,set_nocharge_from_grid
 optim_conf,set_nodischarge_to_grid,set_nodischarge_to_grid
 optim_conf,set_battery_dynamic,set_battery_dynamic
@@ -40,9 +60,16 @@ optim_conf,battery_dynamic_min,battery_dynamic_min
 optim_conf,weight_battery_discharge,weight_battery_discharge
 optim_conf,weight_battery_charge,weight_battery_charge
 optim_conf,weather_forecast_method,weather_forecast_method
+optim_conf,open_meteo_cache_max_age,open_meteo_cache_max_age
 optim_conf,def_start_timestep,start_timesteps_of_each_deferrable_load,list_start_timesteps_of_each_deferrable_load
 optim_conf,def_end_timestep,end_timesteps_of_each_deferrable_load,list_end_timesteps_of_each_deferrable_load
 optim_conf,list_hp_periods,load_peak_hour_periods
+optim_conf,model_type,model_type
+optim_conf,var_model,var_model
+optim_conf,sklearn_model,sklearn_model
+optim_conf,num_lags,num_lags
+optim_conf,split_date_delta,split_date_delta
+optim_conf,perform_backtest,perform_backtest
 plant_conf,P_from_grid_max,maximum_power_from_grid
 plant_conf,P_to_grid_max,maximum_power_to_grid
 plant_conf,module_model,pv_module_model,list_pv_module_model
@@ -52,6 +79,12 @@ plant_conf,surface_azimuth,surface_azimuth,list_surface_azimuth
 plant_conf,modules_per_string,modules_per_string,list_modules_per_string
 plant_conf,strings_per_inverter,strings_per_inverter,list_strings_per_inverter
 plant_conf,inverter_is_hybrid,inverter_is_hybrid
+plant_conf,inverter_ac_output_max,inverter_ac_output_max
+plant_conf,inverter_ac_input_max,inverter_ac_input_max
+plant_conf,inverter_efficiency_dc_ac,inverter_efficiency_dc_ac
+plant_conf,inverter_efficiency_ac_dc,inverter_efficiency_ac_dc
+plant_conf,inverter_stress_cost,inverter_stress_cost
+plant_conf,inverter_stress_segments,inverter_stress_segments
 plant_conf,compute_curtailment,compute_curtailment
 plant_conf,Pd_max,battery_discharge_power_max
 plant_conf,Pc_max,battery_charge_power_max
@@ -60,4 +93,6 @@ plant_conf,eta_ch,battery_charge_efficiency
 plant_conf,Enom,battery_nominal_energy_capacity
 plant_conf,SOCmin,battery_minimum_state_of_charge
 plant_conf,SOCmax,battery_maximum_state_of_charge
-plant_conf,SOCtarget,battery_target_state_of_charge
+plant_conf,SOCtarget,battery_target_state_of_charge
+plant_conf,battery_stress_cost,battery_stress_cost
+plant_conf,battery_stress_segments,battery_stress_segments

emhass/data/config_defaults.json

@@ -2,52 +2,60 @@
   "logging_level": "INFO",
   "costfun": "profit",
   "optimization_time_step": 30,
-  "historic_days_to_retrieve": 2,
+  "historic_days_to_retrieve": 9,
   "method_ts_round": "nearest",
   "continual_publish": false,
   "data_path": "default",
   "set_total_pv_sell": false,
   "lp_solver": "default",
   "lp_solver_path": "empty",
+  "lp_solver_timeout": 45,
+  "num_threads": 0,
   "set_nocharge_from_grid": false,
   "set_nodischarge_to_grid": true,
   "set_battery_dynamic": false,
   "battery_dynamic_max": 0.9,
   "battery_dynamic_min": -0.9,
-  "weight_battery_discharge": 1.0,
-  "weight_battery_charge": 1.0,
+  "weight_battery_discharge": 0.0,
+  "weight_battery_charge": 0.0,
   "sensor_power_photovoltaics": "sensor.power_photovoltaics",
+  "sensor_power_photovoltaics_forecast": "sensor.p_pv_forecast",
   "sensor_power_load_no_var_loads": "sensor.power_load_no_var_loads",
-  "sensor_replace_zero": [
-    "sensor.power_photovoltaics" 
-  ],
+  "sensor_replace_zero": ["sensor.power_photovoltaics", "sensor.p_pv_forecast"],
   "sensor_linear_interp": [
     "sensor.power_photovoltaics",
     "sensor.power_load_no_var_loads"
   ],
+  "use_websocket": false,
+  "use_influxdb": false,
+  "influxdb_host": "localhost",
+  "influxdb_port": 8086,
+  "influxdb_username": "",
+  "influxdb_password": "",
+  "influxdb_database": "homeassistant",
+  "influxdb_measurement": "W",
+  "influxdb_retention_policy": "autogen",
+  "influxdb_use_ssl": false,
+  "influxdb_verify_ssl": false,
   "load_negative": false,
   "set_zero_min": true,
   "number_of_deferrable_loads": 2,
-  "nominal_power_of_deferrable_loads": [
-    3000.0,
-    750.0
-  ],
-  "operating_hours_of_each_deferrable_load": [
-    4,
-    0
-  ],
-  "weather_forecast_method": "scrapper",
+  "nominal_power_of_deferrable_loads": [3000.0, 750.0],
+  "minimum_power_of_deferrable_loads": [0.0, 0.0],
+  "operating_hours_of_each_deferrable_load": [4, 0],
+  "weather_forecast_method": "open-meteo",
+  "open_meteo_cache_max_age": 30,
   "load_forecast_method": "naive",
+  "model_type": "load_forecast",
+  "var_model": "sensor.power_load_no_var_loads",
+  "sklearn_model": "KNeighborsRegressor",
+  "num_lags": 48,
+  "split_date_delta": "48h",
+  "perform_backtest": false,
   "delta_forecast_daily": 1,
   "load_cost_forecast_method": "hp_hc_periods",
-  "start_timesteps_of_each_deferrable_load": [
-    0,
-    0
-  ],
-  "end_timesteps_of_each_deferrable_load": [
-    0,
-    0
-  ],
+  "start_timesteps_of_each_deferrable_load": [0, 0],
+  "end_timesteps_of_each_deferrable_load": [0, 0],
   "load_peak_hour_periods": {
     "period_hp_1": [
       {
@@ -66,43 +74,37 @@
       }
     ]
   },
-  "treat_deferrable_load_as_semi_cont": [
-    true,
-    true
-  ],
-  "set_deferrable_load_single_constant": [
-    false,
-    false
-  ],
-  "set_deferrable_startup_penalty": [
-    0.0,
-    0.0
-  ],
+  "treat_deferrable_load_as_semi_cont": [true, true],
+  "set_deferrable_load_single_constant": [false, false],
+  "set_deferrable_startup_penalty": [0.0, 0.0],
   "load_peak_hours_cost": 0.1907,
   "load_offpeak_hours_cost": 0.1419,
   "production_price_forecast_method": "constant",
   "photovoltaic_production_sell_price": 0.1419,
   "maximum_power_from_grid": 9000,
   "maximum_power_to_grid": 9000,
+  "set_use_pv": false,
+  "set_use_adjusted_pv": false,
+  "adjusted_pv_regression_model": "LassoRegression",
+  "adjusted_pv_solar_elevation_threshold": 10,
+  "adjusted_pv_model_max_age": 24,
   "pv_module_model": [
     "CSUN_Eurasia_Energy_Systems_Industry_and_Trade_CSUN295_60M"
   ],
   "pv_inverter_model": [
     "Fronius_International_GmbH__Fronius_Primo_5_0_1_208_240__240V_"
   ],
-  "surface_tilt": [
-    30
-  ],
-  "surface_azimuth": [
-    205
-  ],
-  "modules_per_string": [
-    16
-  ],
-  "strings_per_inverter": [
-    1
-  ],
+  "surface_tilt": [30],
+  "surface_azimuth": [205],
+  "modules_per_string": [16],
+  "strings_per_inverter": [1],
   "inverter_is_hybrid": false,
+  "inverter_ac_output_max": 5000,
+  "inverter_ac_input_max": 5000,
+  "inverter_efficiency_dc_ac": 1.0,
+  "inverter_efficiency_ac_dc": 1.0,
+  "inverter_stress_cost": 0.0,
+  "inverter_stress_segments": 10,
   "compute_curtailment": false,
   "set_use_battery": false,
   "battery_discharge_power_max": 1000,
@@ -112,5 +114,7 @@
   "battery_nominal_energy_capacity": 5000,
   "battery_minimum_state_of_charge": 0.3,
   "battery_maximum_state_of_charge": 0.9,
-  "battery_target_state_of_charge": 0.6
-}
+  "battery_target_state_of_charge": 0.6,
+  "battery_stress_cost": 0.0,
+  "battery_stress_segments": 10
+}