-
Notifications
You must be signed in to change notification settings - Fork 0
/
slxopenevse.py
323 lines (276 loc) · 10.4 KB
/
slxopenevse.py
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
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
""" slxmodule for connecting with OpenEVSE"""
from homeassistant.helpers.event import async_track_state_change_event
from homeassistant.helpers import entity_registry, device_registry
from homeassistant.helpers.entity_registry import EntityRegistry
from homeassistant.util import slugify
import logging
from typing import Any
from enum import Enum
from homeassistant.core import (
HomeAssistant,
Event,
Callable,
State,
callback,
)
from .slxevse import SLXEvse
from .const import (
CHARGER_MODES,
CHR_MODE_UNKNOWN,
CHR_MODE_STOPPED,
CHR_MODE_PVCHARGE,
CHR_MODE_NORMAL,
)
# entities we subscribe to
WatchedEntities = {
# type of value : entity_id
"sessionenergy": "sensor.{devicename}_usage_this_session",
"plug": "binary_sensor.{devicename}_vehicle_connected",
}
# entities we are taking value from
GetEntities = {
"maxcurrent": "select.{devicename}_max_current",
"divertactive": "binary_sensor.{devicename}_divert_active",
}
# entities we are setting value
SetEntities = {
"manualoverride": "switch.{devicename}_manual_override",
"divertmode": "select.{devicename}_divert_mode",
}
_LOGGER = logging.getLogger(__name__)
class SLXOpenEvse(SLXEvse):
"""Class for OpenEVSE connection"""
def __init__(
self,
hass: HomeAssistant,
):
_LOGGER.debug("SLXOpenEVSE")
super().__init__(hass)
def connect(
self,
cb_sessionenergy: Callable[[Event], Any],
cb_plug: Callable[[Event], Any],
device_id: str | None = None,
) -> bool:
self.openevse_id = device_id
self.openevse_name = SLXOpenEvse.get_openevse_devicename(self.hass, device_id)
if self.openevse_name is None:
_LOGGER.error("OpenEVSE device with device_id=%s does not exist", device_id)
self.openevse_slugified_name = SLXOpenEvse._slugify_device_name(
self.openevse_name
)
if (
SLXOpenEvse.check_all_entities(
self.hass, self.openevse_id, self.openevse_name
)
is False
):
_LOGGER.error(
"OpenEVSE device wasn't found or there were not all required entities"
)
return False
else:
self._subscribe_entity(
SLXOpenEvse.__traslate_entity_name(
WatchedEntities["sessionenergy"], self.openevse_slugified_name
),
cb_sessionenergy,
)
self._subscribe_entity(
SLXOpenEvse.__traslate_entity_name(
WatchedEntities["plug"], self.openevse_slugified_name
),
cb_plug,
)
_LOGGER.info("SLXOpenEVSE correctly initialized")
@staticmethod
def __traslate_entity_name(template_name: str, openevse_slugified_name: str) -> str:
if openevse_slugified_name is not None:
result = template_name.format(devicename=openevse_slugified_name)
else:
result = template_name.format(devicename="")
_LOGGER.warning(
"Missing OpenEVSE device name when trying to translate entity %s",
template_name,
)
return result
def _get_value_translated(self, name: str) -> Any:
translated_name: str | None = None
if name in GetEntities:
translated_name = SLXOpenEvse.__traslate_entity_name(
GetEntities[name], self.openevse_slugified_name
)
if name in SetEntities:
translated_name = SLXOpenEvse.__traslate_entity_name(
SetEntities[name], self.openevse_slugified_name
)
if name in WatchedEntities:
translated_name = SLXOpenEvse.__traslate_entity_name(
WatchedEntities[name], self.openevse_slugified_name
)
if translated_name is None:
return None
entity_state = self.hass.states.get(translated_name)
if entity_state is None:
return None
return entity_state.state
def _set_value_translated(self, name: str, value: Any) -> bool:
if name in SetEntities:
translated_name = SLXOpenEvse.__traslate_entity_name(
SetEntities[name], self.openevse_slugified_name
)
self.hass.async_add_executor_job(
self.hass.states.set,
translated_name,
value,
{},
)
return True
else:
return False
def _select_option(self, name: str, value: str) -> bool:
if name in SetEntities:
self.hass.async_add_executor_job(
self.hass.services.call,
"select",
"select_option",
{
"entity_id": SLXOpenEvse.__traslate_entity_name(
SetEntities[name], self.openevse_slugified_name
),
"option": value,
},
)
_LOGGER.debug("_select_option %s to %s", name, value)
return True
else:
_LOGGER.warning("_select_option is invalid: %s", name)
return False
def _activate_override(self, charge: bool) -> bool:
value_to_set: str = ""
if charge:
value_to_set = "active"
else:
value_to_set = "disabled"
_LOGGER.debug(
"_activate_override - value: %s, device_id = %s",
value_to_set,
self.openevse_id,
)
self.hass.async_add_executor_job(
self.hass.services.call,
"openevse",
"set_override",
{"state": value_to_set, "device_id": [self.openevse_id]},
)
def _clear_override(self) -> bool:
_LOGGER.debug("_clear_override, device_id = %s", self.openevse_id)
self.hass.async_add_executor_job(
self.hass.services.call,
"openevse",
"clear_override",
{"device_id": [self.openevse_id]},
)
def set_charger_mode(self, mode: str) -> None:
if not mode in CHARGER_MODES:
_LOGGER.warning("Invalid charing mode %s", mode)
return
_LOGGER.info("Set charger mode %s", mode)
if mode == self.charge_mode:
return
self.charge_mode = mode
if self.charge_mode == CHR_MODE_STOPPED:
if self._get_value_translated("divertmode") != "fast":
self._select_option("divertmode", "fast")
self._activate_override(False)
return
if self.charge_mode == CHR_MODE_PVCHARGE:
if self._get_value_translated("divertmode") != "eco":
self._select_option("divertmode", "eco")
self._clear_override()
return
if self.charge_mode == CHR_MODE_NORMAL:
if self._get_value_translated("divertmode") != "fast":
self._select_option("divertmode", "fast")
self._activate_override(True)
return
def get_session_energy(self) -> float | None:
value_str = self._get_value_translated("sessionenergy")
if value_str is None:
_LOGGER.warning("Get_session_energy - value_str is None")
return None
try:
value = float(value_str)
return value
except ValueError:
_LOGGER.warning("Get_session_energy - invalid state= %s", value_str)
return None
return None
@staticmethod
def _slugify_device_name(device_name: str) -> str:
return slugify(device_name.lower())
@staticmethod
def find_openevse_devices(hass: HomeAssistant) -> dict[str, str]:
"""Finds list of compatible OpenEVSE devices
:returns: Dictionary of [deviceID, DeviceName]
"""
devices_found: dict[str, str] = {}
deviceregistry = device_registry.async_get(hass)
for device_id in deviceregistry.devices:
device = deviceregistry.async_get(device_id)
if device.manufacturer == "OpenEVSE":
if device.model == "openevse_wifi_v1":
devices_found[device_id] = device.name
return devices_found
@staticmethod
def _find_device_entities(
hass: HomeAssistant, device_id: str
) -> dict[str, EntityRegistry] | None:
"""
Lists all entities belonging to a given device. It's universal and requires only device_id
:returns: None if no device was found. Otherwise returns list of entities.
"""
entity_list: dict[str, EntityRegistry] = {}
entityregistry = entity_registry.async_get(hass)
for entity_id in entityregistry.entities:
entity = entityregistry.async_get(entity_id)
if entity.device_id == device_id:
entity_list[entity.entity_id] = entity
return entity_list
@staticmethod
def get_openevse_devicename(hass: HomeAssistant, device_id: str) -> str | None:
found_devices = SLXOpenEvse.find_openevse_devices(hass)
if device_id in found_devices:
return found_devices[device_id]
else:
return None
@staticmethod
def check_all_entities(
hass: HomeAssistant, device_id: str, device_name: str | None = None
) -> bool:
"""Checks if we have OpenEVSE device with all required entities"""
if device_name is None:
device_name = SLXOpenEvse.get_openevse_devicename(hass, device_id)
if device_name is None:
return False
slugified_device_name = SLXOpenEvse._slugify_device_name(device_name)
entity_list = SLXOpenEvse._find_device_entities(hass, device_id)
if not entity_list:
return False
all_good = True
to_check_list = [WatchedEntities, SetEntities, GetEntities]
list_of_entity_names = []
for checking_list in to_check_list:
for entity_template in checking_list.values():
list_of_entity_names.append(
SLXOpenEvse.__traslate_entity_name(
entity_template, slugified_device_name
)
)
for entity_name in list_of_entity_names:
if entity_name in entity_list:
_LOGGER.info("Found %s", entity_name)
else:
_LOGGER.warning("Missing entity %s", entity_name)
all_good = False
return all_good