emhass 0.10.6__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

@@ -0,0 +1,98 @@
+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
+optim_conf,def_start_penalty,set_deferrable_startup_penalty,list_set_deferrable_startup_penalty
+optim_conf,delta_forecast,delta_forecast_daily
+optim_conf,load_forecast_method,load_forecast_method
+optim_conf,load_cost_forecast_method,load_cost_forecast_method
+optim_conf,load_cost_hp,load_peak_hours_cost
+optim_conf,load_cost_hc,load_offpeak_hours_cost
+optim_conf,prod_price_forecast_method,production_price_forecast_method
+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
+optim_conf,battery_dynamic_max,battery_dynamic_max
+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
+plant_conf,inverter_model,pv_inverter_model,list_pv_inverter_model
+plant_conf,surface_tilt,surface_tilt,list_surface_tilt
+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
+plant_conf,eta_disch,battery_discharge_efficiency
+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,battery_stress_cost,battery_stress_cost
+plant_conf,battery_stress_segments,battery_stress_segments

emhass/data/config_defaults.json

@@ -0,0 +1,120 @@
+{
+  "logging_level": "INFO",
+  "costfun": "profit",
+  "optimization_time_step": 30,
+  "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": 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.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],
+  "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],
+  "load_peak_hour_periods": {
+    "period_hp_1": [
+      {
+        "start": "02:54"
+      },
+      {
+        "end": "15:24"
+      }
+    ],
+    "period_hp_2": [
+      {
+        "start": "17:24"
+      },
+      {
+        "end": "20:24"
+      }
+    ]
+  },
+  "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],
+  "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,
+  "battery_charge_power_max": 1000,
+  "battery_discharge_efficiency": 0.95,
+  "battery_charge_efficiency": 0.95,
+  "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_stress_cost": 0.0,
+  "battery_stress_segments": 10
+}