Fix no energy tariff bug

Fixes bug causing error when generating a battery with no energy tariff
hif2k1 · Sep 23, 2022 · 254af28 · 254af28
1 parent 41bd4ed
commit 254af28
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/custom_components/battery_sim/config_flow.py b/custom_components/battery_sim/config_flow.py
@@ -80,7 +80,8 @@ async def async_step_connectsensors(self, user_input = None):
         if user_input is not None:
             self._data[CONF_IMPORT_SENSOR] = user_input[CONF_IMPORT_SENSOR]
             self._data[CONF_EXPORT_SENSOR] = user_input[CONF_EXPORT_SENSOR]
-            self._data[CONF_ENERGY_TARIFF] = user_input[CONF_ENERGY_TARIFF]
+            if CONF_ENERGY_TARIFF in user_input:
+                self._data[CONF_ENERGY_TARIFF] = user_input[CONF_ENERGY_TARIFF]
             self._data[SETUP_TYPE] = CONFIG_FLOW
             return self.async_create_entry(title=self._data["name"], data=self._data)