home-assistant-trv-opener/trv-script.py at master · charlesomer/home-assistant-trv-opener · GitHub

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
from ast import literal_eval

# Identifiers for TRV temperatures.
state.persist('pyscript.script_boosted_entities')

current_temperature_identifier = "current_temperature"
target_temperature_identifier  = "temperature"

service_name_to_set_temperatures = "set_temperature"

# The temperature to set TRVs to to force them open.
# I've found that 21 degrees is a good balance between forcing them open
# and rooms getting too hot but this may need adjusting based on your
# preference.
temperature_used_to_open_trv = 20

attribute_trv_preset_mode = "preset_mode"
attribute_trv_when_manually_opened_value = "boost"

attribute_demand_check = "percentage_demand"

service_to_use_to_reset_trvs       = "set_preset_mode"
service_to_use_to_reset_trvs_value = "Cancel Boost"

# pyscript.script_boosted_entities = []

@service
def trv_monitor():
  # Get array of rooms with trvs.
  trv_rooms = state.names("climate")

  # We need to find out if a user (not this script) has manually increased the temperature.

  isHeatingRequired = False
  for room in trv_rooms:
    # Get attributes for the room.
    attributes = state.getattr(room)

    current_temperature = attributes[current_temperature_identifier]
    target_temperature  = attributes[target_temperature_identifier]

    if (
      attributes[attribute_demand_check] > 0 and
      (room not in literal_eval(pyscript.script_boosted_entities)) and
      attributes['hvac_action'] != 'idle'
    ):
      # Check if this has been set by the script or not.
      isHeatingRequired = True
      # There is a TRV which should be heating, we can now exit this loop
      # and loop again and this time set TRVs which aren't supposed to be
      # heating to a higher temperature to force them open.
      break

  if isHeatingRequired == True:
    log.info("Heating is required.")

    tmp_script_boosted_entities = literal_eval(pyscript.script_boosted_entities)
    for room in literal_eval(pyscript.script_boosted_entities):
      attributes = state.getattr(room)
      if attributes['preset_mode'] != 'boost':
         tmp_script_boosted_entities.remove(room)
    pyscript.script_boosted_entities = tmp_script_boosted_entities

    script_boosted_entities = []
    for room in trv_rooms:
      attributes = state.getattr(room)

      current_temperature = attributes[current_temperature_identifier]
      target_temperature  = attributes[target_temperature_identifier]

      # This time, if the current temperature is higher than the target
      # temperature (when the TRV will be off), we can increase the target
      # temperature to force the TRV open.
      if (
        attributes[attribute_demand_check] == 0 and
        (room not in literal_eval(pyscript.script_boosted_entities))
      ):
        service.call("climate", service_name_to_set_temperatures,
         entity_id=room,
         temperature=temperature_used_to_open_trv
        )
        script_boosted_entities.append(room)
        log.info("Setting TRV for: " + room)
    pyscript.script_boosted_entities = literal_eval(pyscript.script_boosted_entities) + script_boosted_entities
  else:
    log.info("Heating is not required.")
    # Make sure to reset any TRVs that were previously set to a high
    # back to their defaults.
    for room in literal_eval(pyscript.script_boosted_entities):
      attributes = state.getattr(room)
      service.call("climate", service_to_use_to_reset_trvs,
        entity_id=room,
        preset_mode=service_to_use_to_reset_trvs_value
      )
      log.info("Resetting TRV for: " + room)
    pyscript.script_boosted_entities = []