Files
homeassistant_config/config/custom_components/ecoflow_cloud/devices/registry.py
2024-05-31 13:07:35 +02:00

33 lines
1.7 KiB
Python

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()
}