1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
|
"""SkyKettle."""
import logging
from homeassistant.components.sensor import SensorEntity, SensorStateClass, SensorDeviceClass
from homeassistant.helpers.entity import EntityCategory
from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.const import *
from .skykettle import SkyKettle
from .const import *
_LOGGER = logging.getLogger(__name__)
SENSOR_TYPE_WATER_FRESHNESS = "water_freshness"
SENSOR_TYPE_SUCCESS_RATE = "success_rate"
SENSOR_TYPE_ENERGY = "energy"
SENSOR_TYPE_ONTIME = "ontime"
SENSOR_TYPE_HEATER_ON_COUNT = "heater_on_count"
SENSOR_TYPE_USER_ON_COUNT = "user_on_count"
async def async_setup_entry(hass, entry, async_add_entities):
"""Set up the SkyKettle entry."""
model_code = hass.data[DOMAIN][entry.entry_id][DATA_CONNECTION].model_code
async_add_entities([
SkySensor(hass, entry, SENSOR_TYPE_SUCCESS_RATE),
])
if model_code in [SkyKettle.MODELS_4]: # RK-G2xxS, RK-M13xS, RK-M21xS, RK-M223S but not sure
async_add_entities([
SkySensor(hass, entry, SENSOR_TYPE_ENERGY),
SkySensor(hass, entry, SENSOR_TYPE_ONTIME),
SkySensor(hass, entry, SENSOR_TYPE_HEATER_ON_COUNT),
SkySensor(hass, entry, SENSOR_TYPE_USER_ON_COUNT),
SkySensor(hass, entry, SENSOR_TYPE_WATER_FRESHNESS),
])
class SkySensor(SensorEntity):
"""Representation of a SkyKettle sensor device."""
def __init__(self, hass, entry, sensor_type):
"""Initialize the sensor device."""
self.hass = hass
self.entry = entry
self.sensor_type = sensor_type
async def async_added_to_hass(self):
self.update()
self.async_on_remove(async_dispatcher_connect(self.hass, DISPATCHER_UPDATE, self.update))
def update(self):
self.schedule_update_ha_state()
@property
def kettle(self):
return self.hass.data[DOMAIN][self.entry.entry_id][DATA_CONNECTION]
@property
def unique_id(self):
return f"{self.entry.entry_id}_{self.sensor_type}"
@property
def device_info(self):
return self.hass.data[DOMAIN][DATA_DEVICE_INFO]()
@property
def should_poll(self):
return False
@property
def assumed_state(self):
return False
@property
def last_reset(self):
return None
@property
def name(self):
"""Name of the entity."""
if self.sensor_type == SENSOR_TYPE_ENERGY:
return (FRIENDLY_NAME + " " + self.entry.data.get(CONF_FRIENDLY_NAME, "")).strip() + " total energy consumed"
if self.sensor_type == SENSOR_TYPE_ONTIME:
return (FRIENDLY_NAME + " " + self.entry.data.get(CONF_FRIENDLY_NAME, "")).strip() + " total work time"
if self.sensor_type == SENSOR_TYPE_HEATER_ON_COUNT:
return (FRIENDLY_NAME + " " + self.entry.data.get(CONF_FRIENDLY_NAME, "")).strip() + " heater on count"
if self.sensor_type == SENSOR_TYPE_USER_ON_COUNT:
return (FRIENDLY_NAME + " " + self.entry.data.get(CONF_FRIENDLY_NAME, "")).strip() + " user on count"
if self.sensor_type == SENSOR_TYPE_WATER_FRESHNESS:
return (FRIENDLY_NAME + " " + self.entry.data.get(CONF_FRIENDLY_NAME, "")).strip() + " water freshness"
if self.sensor_type == SENSOR_TYPE_SUCCESS_RATE:
return (FRIENDLY_NAME + " " + self.entry.data.get(CONF_FRIENDLY_NAME, "")).strip() + " success rate"
@property
def icon(self):
if self.sensor_type == SENSOR_TYPE_ONTIME:
return "mdi:timelapse"
if self.sensor_type == SENSOR_TYPE_HEATER_ON_COUNT:
return "mdi:dip-switch"
if self.sensor_type == SENSOR_TYPE_USER_ON_COUNT:
return "mdi:dip-switch"
if self.sensor_type == SENSOR_TYPE_WATER_FRESHNESS:
return "mdi:water-sync"
if self.sensor_type == SENSOR_TYPE_SUCCESS_RATE:
return "mdi:bluetooth-connect"
return None
@property
def available(self):
if self.sensor_type == SENSOR_TYPE_ENERGY:
return self.kettle.available and self.kettle.energy_wh != None
if self.sensor_type == SENSOR_TYPE_ONTIME:
return self.kettle.available and self.kettle.ontime != None
if self.sensor_type == SENSOR_TYPE_HEATER_ON_COUNT:
return self.kettle.available and self.kettle.heater_on_count != None
if self.sensor_type == SENSOR_TYPE_USER_ON_COUNT:
return self.kettle.available and self.kettle.user_on_count != None
if self.sensor_type == SENSOR_TYPE_WATER_FRESHNESS:
return self.kettle.available and self.kettle.water_freshness_hours != None
if self.sensor_type == SENSOR_TYPE_SUCCESS_RATE:
return True # Always readable
@property
def entity_category(self):
if self.sensor_type == SENSOR_TYPE_SUCCESS_RATE:
return EntityCategory.DIAGNOSTIC
return None
@property
def device_class(self):
if self.sensor_type == SENSOR_TYPE_ENERGY:
return SensorDeviceClass.ENERGY
if self.sensor_type == SENSOR_TYPE_ONTIME:
return SensorDeviceClass.DURATION
if self.sensor_type == SENSOR_TYPE_WATER_FRESHNESS:
return SensorDeviceClass.DURATION
return None # Unusual class
@property
def state_class(self):
if self.sensor_type == SENSOR_TYPE_ENERGY:
return SensorStateClass.TOTAL_INCREASING
if self.sensor_type == SENSOR_TYPE_ONTIME:
return SensorStateClass.TOTAL_INCREASING
if self.sensor_type == SENSOR_TYPE_HEATER_ON_COUNT:
return SensorStateClass.TOTAL_INCREASING
if self.sensor_type == SENSOR_TYPE_USER_ON_COUNT:
return SensorStateClass.TOTAL_INCREASING
if self.sensor_type == SENSOR_TYPE_WATER_FRESHNESS:
return SensorStateClass.MEASUREMENT
if self.sensor_type == SENSOR_TYPE_SUCCESS_RATE:
return SensorStateClass.MEASUREMENT
@property
def native_unit_of_measurement(self):
if self.sensor_type == SENSOR_TYPE_ENERGY:
return ENERGY_WATT_HOUR
if self.sensor_type == SENSOR_TYPE_ONTIME:
return TIME_SECONDS
if self.sensor_type == SENSOR_TYPE_WATER_FRESHNESS:
return TIME_HOURS
if self.sensor_type == SENSOR_TYPE_SUCCESS_RATE:
return PERCENTAGE
return None
@property
def native_value(self):
if self.sensor_type == SENSOR_TYPE_ENERGY:
return self.kettle.energy_wh
if self.sensor_type == SENSOR_TYPE_ONTIME:
return self.kettle.ontime.total_seconds()
if self.sensor_type == SENSOR_TYPE_HEATER_ON_COUNT:
return self.kettle.heater_on_count
if self.sensor_type == SENSOR_TYPE_USER_ON_COUNT:
return self.kettle.user_on_count
if self.sensor_type == SENSOR_TYPE_WATER_FRESHNESS:
return self.kettle.water_freshness_hours
if self.sensor_type == SENSOR_TYPE_SUCCESS_RATE:
return self.kettle.success_rate
|
