deploy

2024-05-11 06:59:15 +08:00
commit 9cc7d172cc
16 changed files with 1576 additions and 0 deletions
@@ -0,0 +1,198 @@
+from homeassistant.components.sensor import (
+    DOMAIN as SENSOR_DOMAIN,
+    SensorDeviceClass,
+    SensorEntity,
+    SensorStateClass,
+)
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import UnitOfEnergy
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
+
+import datetime
+
+from .const import DOMAIN, VERSION
+from .data_client import StateGridDataClient
+from .coordinator import StateGridCoordinator
+
+CURRENCY_YUAN = "元"
+
+ENTITY_ID_SENSOR_FORMAT = SENSOR_DOMAIN + ".state_grid_"
+
+SENSOR_TYPES = [
+    {
+        "key":"balance",
+        "name": "账户余额",
+        "native_unit_of_measurement": CURRENCY_YUAN,
+        "device_class": SensorDeviceClass.MONETARY,
+        "state_class": SensorStateClass.TOTAL
+    },
+    {
+        "key": "year_ele_num",
+        "name": "年度累计用电",
+        "native_unit_of_measurement": UnitOfEnergy.KILO_WATT_HOUR,
+        "device_class": SensorDeviceClass.ENERGY,
+        "state_class": SensorStateClass.TOTAL
+    },
+    {
+        "key": "year_ele_cost",
+        "name": "年度累计电费",
+        "native_unit_of_measurement": CURRENCY_YUAN,
+        "device_class": SensorDeviceClass.MONETARY,
+        "state_class": SensorStateClass.TOTAL
+    },
+    {
+        "key": "last_month_ele_num",
+        "name": "上个月用电",
+        "native_unit_of_measurement": UnitOfEnergy.KILO_WATT_HOUR,
+        "device_class": SensorDeviceClass.ENERGY,
+        "state_class": SensorStateClass.TOTAL
+    },
+    {
+        "key": "last_month_ele_cost",
+        "name": "上个月电费",
+        "native_unit_of_measurement": CURRENCY_YUAN,
+        "device_class": SensorDeviceClass.MONETARY,
+        "state_class": SensorStateClass.TOTAL
+    },
+    {
+        "key": "month_ele_num",
+        "name": "当月累计用电",
+        "native_unit_of_measurement": UnitOfEnergy.KILO_WATT_HOUR,
+        "device_class": SensorDeviceClass.ENERGY,
+        "state_class": SensorStateClass.TOTAL
+    },
+    {
+        "key": "month_p_ele_num",
+        "name": "当月累计峰用电",
+        "native_unit_of_measurement": UnitOfEnergy.KILO_WATT_HOUR,
+        "device_class": SensorDeviceClass.ENERGY,
+        "state_class": SensorStateClass.TOTAL
+    },
+    {
+        "key": "month_v_ele_num",
+        "name": "当月累计谷用电",
+        "native_unit_of_measurement": UnitOfEnergy.KILO_WATT_HOUR,
+        "device_class": SensorDeviceClass.ENERGY,
+        "state_class": SensorStateClass.TOTAL
+    },
+    {
+        "key": "month_n_ele_num",
+        "name": "当月累计平用电",
+        "native_unit_of_measurement": UnitOfEnergy.KILO_WATT_HOUR,
+        "device_class": SensorDeviceClass.ENERGY,
+        "state_class": SensorStateClass.TOTAL
+    },
+    {
+        "key": "month_t_ele_num",
+        "name": "当月累计尖用电",
+        "native_unit_of_measurement": UnitOfEnergy.KILO_WATT_HOUR,
+        "device_class": SensorDeviceClass.ENERGY,
+        "state_class": SensorStateClass.TOTAL
+    },
+    {
+        "key": "daily_ele_num",
+        "name": "日总用电",
+        "native_unit_of_measurement": UnitOfEnergy.KILO_WATT_HOUR,
+        "device_class": SensorDeviceClass.ENERGY,
+        "state_class": SensorStateClass.TOTAL
+    },
+    {
+        "key": "daily_p_ele_num",
+        "name": "日峰用电",
+        "native_unit_of_measurement": UnitOfEnergy.KILO_WATT_HOUR,
+        "device_class": SensorDeviceClass.ENERGY,
+        "state_class": SensorStateClass.TOTAL
+    },
+    {
+        "key": "daily_v_ele_num",
+        "name": "日谷用电",
+        "native_unit_of_measurement": UnitOfEnergy.KILO_WATT_HOUR,
+        "device_class": SensorDeviceClass.ENERGY,
+        "state_class": SensorStateClass.TOTAL
+    },
+    {
+        "key": "daily_n_ele_num",
+        "name": "日平用电",
+        "native_unit_of_measurement": UnitOfEnergy.KILO_WATT_HOUR,
+        "device_class": SensorDeviceClass.ENERGY,
+        "state_class": SensorStateClass.TOTAL
+    },
+    {
+        "key": "daily_t_ele_num",
+        "name": "日尖用电",
+        "native_unit_of_measurement": UnitOfEnergy.KILO_WATT_HOUR,
+        "device_class": SensorDeviceClass.ENERGY,
+        "state_class": SensorStateClass.TOTAL
+    },
+    {
+        "key": "daily_lasted_date",
+        "name": "最新日用电日期"
+    },
+    {
+        "key": "refresh_time",
+        "name": "最近刷新时间"
+    }
+]
+
+SENSOR_TYPES_FOR_LADDER = [
+    {
+        "key": "ladder_level",
+        "name": "当前阶梯"
+    },
+]
+
+async def async_setup_entry(
+    hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
+) -> None:
+    coordinator = StateGridCoordinator(hass)
+    await coordinator.async_config_entry_first_refresh()
+    data_client: StateGridDataClient = hass.data[DOMAIN]
+    door_account_list = await data_client.get_door_account_list()
+    for door_account in door_account_list:
+        async_add_entities(
+            [StateGridSensor(door_account, sensor_type, entry.entry_id, coordinator) for sensor_type in SENSOR_TYPES]
+        )
+        # if door_account["ladder_flag"] == 1:
+        #     async_add_entities(
+        #         StateGridSensor(door_account, sensor_type, entry.entry_id)
+        #         for sensor_type in SENSOR_TYPES_FOR_LADDER
+        #     )
+
+class StateGridSensor(CoordinatorEntity[StateGridCoordinator], SensorEntity):
+
+    _attr_has_entity_name = True
+
+    def __init__(
+        self, door_account, sensor_type, entry_id: str, coordinator: StateGridCoordinator,
+    ) -> None:
+        super().__init__(coordinator)
+        self.door_account = door_account
+        self.sensor_type = sensor_type
+        self.entity_id = SENSOR_DOMAIN + ".state_grid" + "_" + door_account["consNo_dst"] + "_" + sensor_type["key"]
+        self._attr_name = sensor_type["name"]
+        self._attr_unique_id = entry_id + "-" + door_account["consNo_dst"] + "-" + sensor_type["key"]
+
+        if "device_class" in sensor_type:
+            self._attr_device_class = sensor_type["device_class"]
+
+        if "state_class" in sensor_type:
+            self._attr_state_class = sensor_type["state_class"]
+
+        if "native_unit_of_measurement" in sensor_type:
+            self._attr_native_unit_of_measurement = sensor_type["native_unit_of_measurement"]
+
+        self._attr_device_info = {
+            "name": door_account["elecAddr_dst"],
+            "identifiers": {(DOMAIN, door_account["consNo_dst"])},
+            "sw_version": VERSION,
+            "manufacturer": "HassBox",
+            "model": "户号：" + door_account["consName_dst"] + " - " + door_account["consNo_dst"]
+        }
+
+    @property
+    def native_value(self):
+        data = self.coordinator.data[self.door_account["consNo_dst"]]
+        return data[self.sensor_type["key"]]
+