import appdaemon.plugins.hass.hassapi as hass

#
# Water leak detector
# close water valve and send a report
#
# Args:
#
# sensor = water leak sensor group
# valve = water valve - app will close it, when leak is detected
# notify = notify name to send notification
#
# None
#
# Release Notes
#
# Version 1.0:
#   Initial Version

class WaterLeak(hass.Hass):
  ringtone = 2
  gw_mac = ''
  def initialize(self):
    if 'sensor' not in self.args or 'valve' not in self.args or 'notify' not in self.args:
      self.error("Please provide sensor, valve and notify in config!")
      return
    self.listen_state(self.water_leak, self.args['sensor'])
    
  def water_leak(self, entity, attribute, old, new, kwargs):
    if new == "on":
      self.send_alarm()

  def send_alarm(self):
    if 'constraint' in self.args and not self.constrain_input_boolean(self.args['constraint']):
      return

    #close water valve
    self.turn_off(self.args['valve'])
    ## to ensure that water is closed. It is important!!!
    self.call_service("mqtt/publish", topic = 'home/bath/bath/valve/set', payload = 'close')
    self.call_service("xiaomi_aqara/play_ringtone", gw_mac=self.gw_mac, ringtone_id=self.ringtone, ringtone_vol=100)

    message = "ВНИМАНИЕ! В ваше отстутствие сработал датчик протечки воды!!! Выключаю подачу воды."
    self.notify(message, name = self.args['notify'])

    message = "Состояние датчиков протечки:\n"
    attributes = self.get_state(self.args['sensor'], attribute = "all")
    if 'attributes' in attributes:
      attributes = attributes['attributes']
    if 'entity_id' in attributes:
      for entity_id in attributes["entity_id"]:
        message += '{} ({}) = {}\n'.format(self.friendly_name(entity_id), entity_id, self.get_state(entity_id))
      self.notify(message, name = self.args['notify'])