Fix input sensors (#329)

bieniu · Oct 24, 2022 · cef335b · cef335b
1 parent 7e615fc
commit cef335b
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/python_scripts/shellies_discovery.py b/python_scripts/shellies_discovery.py
@@ -1664,7 +1664,6 @@ def mqtt_publish(topic, payload, retain, json=False):
     relay_sensors = {
         SENSOR_POWER: OPTIONS_SENSOR_RELAY_POWER,
         SENSOR_ENERGY: OPTIONS_SENSOR_RELAY_ENERGY,
-        SENSOR_INPUT_0: OPTIONS_SENSOR_INPUT_0,
     }
     relay_binary_sensors = {
         SENSOR_OVERPOWER: OPTIONS_SENSOR_OVERPOWER,
@@ -1682,6 +1681,7 @@ def mqtt_publish(topic, payload, retain, json=False):
     binary_sensors = {
         SENSOR_FIRMWARE_UPDATE: OPTIONS_SENSOR_FIRMWARE_UPDATE,
         SENSOR_OVERTEMPERATURE: OPTIONS_SENSOR_OVERTEMPERATURE,
+        SENSOR_INPUT_0: OPTIONS_SENSOR_INPUT_0,
     }
     buttons = {
         BUTTON_UPDATE_FIRMWARE: OPTIONS_BUTTON_UPDATE_FIRMWARE,
@@ -1696,7 +1696,6 @@ def mqtt_publish(topic, payload, retain, json=False):
     relay_sensors = {
         SENSOR_POWER: OPTIONS_SENSOR_RELAY_POWER,
         SENSOR_ENERGY: OPTIONS_SENSOR_RELAY_ENERGY,
-        SENSOR_INPUT_0: OPTIONS_SENSOR_INPUT_0,
     }
     sensors = {
         SENSOR_RSSI: OPTIONS_SENSOR_RSSI,
@@ -1709,6 +1708,7 @@ def mqtt_publish(topic, payload, retain, json=False):
     binary_sensors = {
         SENSOR_FIRMWARE_UPDATE: OPTIONS_SENSOR_FIRMWARE_UPDATE,
         SENSOR_OVERTEMPERATURE: OPTIONS_SENSOR_OVERTEMPERATURE,
+        SENSOR_INPUT_0: OPTIONS_SENSOR_INPUT_0,
     }
     ext_temp_sensors = 1
     buttons = {
@@ -1804,7 +1804,6 @@ def mqtt_publish(topic, payload, retain, json=False):
     inputs_types = [VALUE_BUTTON_LONG_PRESS, VALUE_BUTTON_SHORT_PRESS]
     relay_binary_sensors = {
         SENSOR_OVERPOWER: OPTIONS_SENSOR_OVERPOWER,
-        SENSOR_INPUT_0: OPTIONS_SENSOR_INPUT_0,
         "longpush": {},
         "shortpush": {},
     }
@@ -1815,7 +1814,10 @@ def mqtt_publish(topic, payload, retain, json=False):
         SENSOR_SSID: OPTIONS_SENSOR_SSID,
         SENSOR_UPTIME: OPTIONS_SENSOR_UPTIME,
     }
-    binary_sensors = {SENSOR_FIRMWARE_UPDATE: OPTIONS_SENSOR_FIRMWARE_UPDATE}
+    binary_sensors = {
+        SENSOR_FIRMWARE_UPDATE: OPTIONS_SENSOR_FIRMWARE_UPDATE,
+        SENSOR_INPUT_0: OPTIONS_SENSOR_INPUT_0,
+    }
     buttons = {
         BUTTON_UPDATE_FIRMWARE: OPTIONS_BUTTON_UPDATE_FIRMWARE,
         BUTTON_RESTART: OPTIONS_BUTTON_RESTART,