from homeassistant.components.sensor import SensorDeviceClass, SensorStateClass
from homeassistant.const import Platform, UnitOfPower, UnitOfElectricPotential
from homeassistant.components.switch import SwitchDeviceClass

DEVICE_MAPPING = {
    "default": {
        "rationale": ["off", "on"],
        "queries": [{}],
        "centralized": [],
        "calculate": {
            "get": [
                {
                    "lvalue": "[b7_vbattery]",
                    "rvalue": "[b7_vbatt] / 1000.0"
                },
            ],
        },
        "entities": {
            Platform.SWITCH: {
                "power": {
                    "device_class": SwitchDeviceClass.SWITCH,
                },
                "light": {
                    "device_class": SwitchDeviceClass.SWITCH,
                },
                "wisdom_wind": {
                    "device_class": SwitchDeviceClass.SWITCH,
                }
            },
            Platform.SENSOR: {
                "error_code": {
                    "device_class": SensorDeviceClass.ENUM
                },
                "b7_left_status": {
                    "device_class": SensorDeviceClass.ENUM,
                    "translation_key": "left_status",
                },
                "b7_right_status": {
                    "device_class": SensorDeviceClass.ENUM,
                    "translation_key": "right_status",
                },
                "b7_vbattery":{
                    "device_class": SensorDeviceClass.VOLTAGE,
                    "unit_of_measurement": UnitOfElectricPotential.VOLT,
                    "state_class": SensorStateClass.MEASUREMENT,
                    "translation_key": "battery_voltage",
                }
            },
            Platform.SELECT: {
                "wind_pressure": {
                    "options": {
                        "off": {"wind_pressure": "0"},
                        "low": {"wind_pressure": "1"},
                        "medium": {"wind_pressure": "2"},
                        "high": {"wind_pressure": "3"},
                        "extreme": {"wind_pressure": "4"},
                    }
                },
                "gear": {
                    "options": {
                        "off": {"gear": 0},
                        "low": {"gear": 1},
                        "medium": {"gear": 2},
                        "high": {"gear": 3},
                        "extreme": {"gear": 4},
                    }
                },
            },
        }
    }
}