Home Assistant Git Exporter

config/custom_components/ecoflow_cloud/devices/registry.py

@@ -0,0 +1,32 @@
+from custom_components.ecoflow_cloud.config_flow import EcoflowModel
+from custom_components.ecoflow_cloud.devices import BaseDevice, DiagnosticDevice
+from custom_components.ecoflow_cloud.devices.delta2 import Delta2
+from custom_components.ecoflow_cloud.devices.delta_mini import DeltaMini
+from custom_components.ecoflow_cloud.devices.delta_pro import DeltaPro
+from custom_components.ecoflow_cloud.devices.river2 import River2
+from custom_components.ecoflow_cloud.devices.river2_max import River2Max
+from custom_components.ecoflow_cloud.devices.river2_pro import River2Pro
+from custom_components.ecoflow_cloud.devices.river_max import RiverMax
+from custom_components.ecoflow_cloud.devices.river_pro import RiverPro
+from custom_components.ecoflow_cloud.devices.delta_max import DeltaMax
+from custom_components.ecoflow_cloud.devices.delta2_max import Delta2Max
+from custom_components.ecoflow_cloud.devices.powerstream import PowerStream
+from custom_components.ecoflow_cloud.devices.glacier import Glacier
+from custom_components.ecoflow_cloud.devices.wave2 import Wave2
+
+devices: dict[str, BaseDevice] = {
+    EcoflowModel.DELTA_2.name: Delta2(),
+    EcoflowModel.RIVER_2.name: River2(),
+    EcoflowModel.RIVER_2_MAX.name: River2Max(),
+    EcoflowModel.RIVER_2_PRO.name: River2Pro(),
+    EcoflowModel.DELTA_PRO.name: DeltaPro(),
+    EcoflowModel.RIVER_MAX.name: RiverMax(),
+    EcoflowModel.RIVER_PRO.name: RiverPro(),
+    EcoflowModel.DELTA_MINI.name: DeltaMini(),
+    EcoflowModel.DELTA_MAX.name: DeltaMax(),
+    EcoflowModel.DELTA_2_MAX.name: Delta2Max(),
+    EcoflowModel.POWERSTREAM.name: PowerStream(),
+    EcoflowModel.GLACIER.name: Glacier(),
+    EcoflowModel.WAVE_2.name: Wave2(),
+    EcoflowModel.DIAGNOSTIC.name: DiagnosticDevice()
+}