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
|
"""Home Assistant helper methods."""
from aiopvpc.const import (
ALL_SENSORS,
KEY_ADJUSTMENT,
KEY_INDEXED,
KEY_INJECTION,
KEY_MAG,
KEY_OMIE,
KEY_PVPC,
TARIFFS,
)
_ha_uniqueid_to_sensor_key = {
TARIFFS[0]: KEY_PVPC,
TARIFFS[1]: KEY_PVPC,
f"{TARIFFS[0]}_{KEY_INJECTION}": KEY_INJECTION,
f"{TARIFFS[1]}_{KEY_INJECTION}": KEY_INJECTION,
f"{TARIFFS[0]}_INYECTION": KEY_INJECTION,
f"{TARIFFS[1]}_INYECTION": KEY_INJECTION,
f"{TARIFFS[0]}_{KEY_MAG}": KEY_MAG,
f"{TARIFFS[1]}_{KEY_MAG}": KEY_MAG,
f"{TARIFFS[0]}_{KEY_OMIE}": KEY_OMIE,
f"{TARIFFS[1]}_{KEY_OMIE}": KEY_OMIE,
f"{TARIFFS[0]}_{KEY_ADJUSTMENT}": KEY_ADJUSTMENT,
f"{TARIFFS[1]}_{KEY_ADJUSTMENT}": KEY_ADJUSTMENT,
f"{TARIFFS[0]}_{KEY_INDEXED}": KEY_INDEXED,
f"{TARIFFS[1]}_{KEY_INDEXED}": KEY_INDEXED,
}
def get_enabled_sensor_keys(
using_private_api: bool, disabled_sensor_ids: list[str]
) -> set[str]:
"""(HA) Get enabled API indicators."""
sensor_keys = set(ALL_SENSORS) if using_private_api else {KEY_PVPC}
for unique_id in disabled_sensor_ids:
disabled_ind = _ha_uniqueid_to_sensor_key[unique_id]
if disabled_ind in sensor_keys:
sensor_keys.remove(disabled_ind)
return sensor_keys
def make_sensor_unique_id(config_entry_id: str, sensor_key: str) -> str:
"""(HA) Generate unique_id for each sensor kind and config entry."""
assert sensor_key in ALL_SENSORS or sensor_key == KEY_INDEXED
if sensor_key == KEY_PVPC:
# for old compatibility
return config_entry_id
return f"{config_entry_id}_{sensor_key}"
|
