smart_brain/function/ground_floor_west.py

262 lines
15 KiB
Python
Raw Normal View History

2022-12-20 14:05:32 +01:00
#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
import devices
import logging
from function.rooms import room_shelly, room_shelly_tradfri_light, room_shelly_silvercrest_light
from function.helpers import changed_value_indicator
import task
2022-12-20 14:05:32 +01:00
try:
from config import APP_NAME as ROOT_LOGGER_NAME
except ImportError:
ROOT_LOGGER_NAME = 'root'
logger = logging.getLogger(ROOT_LOGGER_NAME).getChild(__name__)
class ground_floor_west_floor(room_shelly_silvercrest_light):
# https://shelly1l-84CCA8AD1148
def __init__(self, mqtt_client):
super().__init__(mqtt_client, "shellies/floor_eg_w", "gui/gfw/floor/main_light/switch", "zigbee_eg_w/light/floor_eg_w/a", "gui/gfw/floor/main_light/br_ct")
2022-12-20 14:05:32 +01:00
#
# Callback initialisation
#
self.main_light_tradfri_2 = devices.tradfri_light(mqtt_client, "zigbee_eg_w/light/floor_eg_w/b")
self.main_light_tradfri.add_callback(devices.tradfri_light.KEY_BRIGHTNESS, None, self.main_light_tradfri_2.set_brightness_mcb)
self.main_light_tradfri.add_callback(devices.tradfri_light.KEY_COLOR_TEMP, None, self.main_light_tradfri_2.set_color_temp_mcb)
2022-12-20 14:05:32 +01:00
def send_init_message_main_light(self):
super().send_init_message_main_light()
2022-12-20 14:05:32 +01:00
self.main_light_tradfri_2.mqtt_client.send(self.main_light_tradfri_2.topic + "/get", '{"state": ""}')
class ground_floor_west_marion(room_shelly):
# https://shelly1l-E8DB84A1E067
def __init__(self, mqtt_client):
super().__init__(mqtt_client, "shellies/marion", "gui/gfw/marion/main_light/switch")
2022-12-20 14:05:32 +01:00
class ground_floor_west_dirk(room_shelly_tradfri_light):
STATE_ACTIVE_DEVICE_MAIN_LIGHT = 0
STATE_ACTIVE_DEVICE_DESK_LIGHT = 1
STATE_ACTIVE_DEVICE_AMPLIFIER = 2
STATE_ACTIVE_DEVICE_MAX_VALUE = STATE_ACTIVE_DEVICE_AMPLIFIER
#
LED_ACTIVE_DEVICE_MAIN_LIGHT = devices.nodered_gui_leds.KEY_LED_0
LED_ACTIVE_DEVICE_DESK_LIGHT = devices.nodered_gui_leds.KEY_LED_1
LED_ACTIVE_DEVICE_AMPLIFIER = devices.nodered_gui_leds.KEY_LED_2
#
KEY_POWERPLUG_AMPLIFIER = devices.my_powerplug.KEY_OUTPUT_0
KEY_POWERPLUG_CD_PLAYER = devices.my_powerplug.KEY_OUTPUT_2
KEY_POWERPLUG_DESK_LIGHT = devices.my_powerplug.KEY_OUTPUT_1
KEY_POWERPLUG_PC_DOCK = devices.my_powerplug.KEY_OUTPUT_3
#
AUDIO_SOURCE_PC = 0
AUDIO_SOURCE_CD = 1
AUDIO_SOURCE_RASPI = 2
2022-12-20 14:05:32 +01:00
# https://shelly1l-3C6105E44F27
def __init__(self, mqtt_client):
super().__init__(mqtt_client, "shellies/dirk", "gui/gfw/dirk/main_light/switch", "zigbee_eg_w/light/dirk", "gui/gfw/dirk/main_light/br_ct")
2022-12-22 12:09:53 +01:00
#
self.powerplug_common = devices.my_powerplug(mqtt_client, "powerplug/dirk")
self.desk_light_tradfri = devices.tradfri_light(mqtt_client, "zigbee_eg_w/light/dirk_desk")
self.button_tradfri = devices.tradfri_button(mqtt_client, "zigbee_eg_w/input_device/dirk")
#
self.gui_switch_desk_light = devices.nodered_gui_switch(mqtt_client, "gui/gfw/dirk/desk_light/switch")
self.gui_br_cr_desk_light = devices.nodered_gui_brightness_color_temp(mqtt_client, "gui/gfw/dirk/desk_light/br_ct")
2022-12-22 12:09:53 +01:00
#
self.gui_switch_amplifier = devices.nodered_gui_switch(mqtt_client, "gui/gfw/dirk/amplifier/switch")
self.gui_switch_cd_player = devices.nodered_gui_switch(mqtt_client, "gui/gfw/dirk/cd_player/switch")
self.gui_switch_pc_dock = devices.nodered_gui_switch(mqtt_client, "gui/gfw/dirk/pc_dock/switch")
2022-12-22 12:09:53 +01:00
#
self.remote_amplifier = devices.remote(mqtt_client, "hifi/remote/RAS5")
self.gui_led_active_device = devices.nodered_gui_leds(mqtt_client, "gui/gfw/dirk/active_device_state/led")
#
self.spotify_state = devices.audio_status(mqtt_client, "hifi/spotify")
self.mpd_state = devices.audio_status(mqtt_client, "hifi/mpd")
#
self.delayed_task = task.delayed(1.0, self.send_audio_source)
#
self.cvi = changed_value_indicator()
#
# Callback initialisation
#
# main light
self.button_tradfri.add_callback(devices.tradfri_button.KEY_ACTION, devices.tradfri_button.ACTION_TOGGLE,
self.main_light_shelly.toggle_output_0_mcb)
# desk light
# switch
self.button_tradfri.add_callback(devices.tradfri_button.KEY_ACTION, devices.tradfri_button.ACTION_RIGHT,
self.powerplug_common.toggle_output_1_mcb)
self.gui_switch_desk_light.add_callback(devices.nodered_gui_switch.KEY_STATE, None, self.powerplug_common.set_output_1_mcb)
self.powerplug_common.add_callback(self.KEY_POWERPLUG_DESK_LIGHT, None, self.gui_switch_desk_light.set_state_mcb)
# brightness and color temp
self.gui_br_cr_desk_light.add_callback(devices.nodered_gui_brightness_color_temp.KEY_BRIGHTNESS,
None, self.desk_light_tradfri.set_brightness_mcb)
self.gui_br_cr_desk_light.add_callback(devices.nodered_gui_brightness_color_temp.KEY_COLOR_TEMP,
None, self.desk_light_tradfri.set_color_temp_mcb)
self.powerplug_common.add_callback(self.KEY_POWERPLUG_DESK_LIGHT, None, self.gui_br_cr_desk_light.set_enable_mcb)
self.desk_light_tradfri.add_callback(devices.tradfri_light.KEY_BRIGHTNESS, None, self.gui_br_cr_desk_light.set_brightness_mcb)
self.desk_light_tradfri.add_callback(devices.tradfri_light.KEY_COLOR_TEMP, None, self.gui_br_cr_desk_light.set_color_temp_mcb)
# amplifier
self.button_tradfri.add_callback(devices.tradfri_button.KEY_ACTION, devices.tradfri_button.ACTION_LEFT_LONG,
self.powerplug_common.toggle_output_0_mcb)
self.gui_switch_amplifier.add_callback(devices.nodered_gui_switch.KEY_STATE, None, self.powerplug_common.set_output_0_mcb)
self.powerplug_common.add_callback(self.KEY_POWERPLUG_AMPLIFIER, None, self.gui_switch_amplifier.set_state_mcb)
# amplifier auto on
self.powerplug_common.add_callback(self.KEY_POWERPLUG_CD_PLAYER, None, self.cd_amplifier_synchronisation)
self.spotify_state.add_callback(devices.status.KEY_STATE, None, self.raspi_amplifier_synchronisation)
self.mpd_state.add_callback(devices.status.KEY_STATE, None, self.raspi_amplifier_synchronisation)
# audio source
self.powerplug_common.add_callback(self.KEY_POWERPLUG_AMPLIFIER, None, self.audio_source_selector)
self.powerplug_common.add_callback(self.KEY_POWERPLUG_CD_PLAYER, None, self.audio_source_selector)
self.spotify_state.add_callback(devices.status.KEY_STATE, None, self.audio_source_selector)
self.mpd_state.add_callback(devices.status.KEY_STATE, None, self.audio_source_selector)
self.audio_source = self.AUDIO_SOURCE_PC
# cd player
self.button_tradfri.add_callback(devices.tradfri_button.KEY_ACTION, devices.tradfri_button.ACTION_RIGHT_LONG,
self.powerplug_common.toggle_output_2_mcb)
self.gui_switch_cd_player.add_callback(devices.nodered_gui_switch.KEY_STATE, None, self.powerplug_common.set_output_2_mcb)
self.powerplug_common.add_callback(self.KEY_POWERPLUG_CD_PLAYER, None, self.gui_switch_cd_player.set_state_mcb)
# pc dock
self.button_tradfri.add_callback(devices.tradfri_button.KEY_ACTION, devices.tradfri_button.ACTION_LEFT,
self.powerplug_common.toggle_output_3_mcb)
self.gui_switch_pc_dock.add_callback(devices.nodered_gui_switch.KEY_STATE, None, self.powerplug_common.set_output_3_mcb)
self.powerplug_common.add_callback(self.KEY_POWERPLUG_PC_DOCK, None, self.gui_switch_pc_dock.set_state_mcb)
# brightness
self.button_tradfri.add_callback(devices.tradfri_button.KEY_ACTION, None, self.brightness_action)
self.main_light_shelly.add_callback(devices.shelly.KEY_OUTPUT_0, None, self.device_chooser_action)
self.powerplug_common.add_callback(None, None, self.device_chooser_action)
self.button_tradfri.add_callback(devices.tradfri_button.KEY_ACTION, devices.tradfri_button.ACTION_BRIGHTNESS_UP, self.choose_next_device)
self.button_tradfri.add_callback(devices.tradfri_button.KEY_ACTION, devices.tradfri_button.ACTION_BRIGHTNESS_DOWN, self.choose_prev_device)
self.active_device_state = None
self.update_active_device_led()
def all_off(self, device=None, key=None, data=None):
super().all_off(device, key, data)
2022-12-22 12:09:53 +01:00
self.powerplug_common.set_output_all(False)
def cd_amplifier_synchronisation(self, device, key, data):
2022-12-23 11:45:00 +01:00
if self.cvi.changed_here(device.topic, key, data) and device.previous_value(key) is not None:
logger.info("Syncing \"%s\" amplifier with cd player: %s", type(self).__name__, data)
self.powerplug_common.set_output(self.KEY_POWERPLUG_AMPLIFIER, data)
def raspi_amplifier_synchronisation(self, device, key, data):
2022-12-23 11:45:00 +01:00
if self.cvi.changed_here(device.topic, key, data) and device.previous_value(key) is not None:
logger.info("Syncing \"%s\" amplifier with raspi player: %s", type(self).__name__, data)
self.powerplug_common.set_output(self.KEY_POWERPLUG_AMPLIFIER, data)
2022-12-22 12:09:53 +01:00
def device_chooser_action(self, device, key, data):
if device == self.main_light_shelly:
if self.cvi.changed_here(device.topic, key, data):
if data is True:
self.active_device_state = self.STATE_ACTIVE_DEVICE_MAIN_LIGHT
self.update_active_device_led()
else:
self.choose_next_device()
elif device == self.powerplug_common and key == self.KEY_POWERPLUG_DESK_LIGHT:
if self.cvi.changed_here(device.topic, key, data):
if data is True:
self.active_device_state = self.STATE_ACTIVE_DEVICE_DESK_LIGHT
self.update_active_device_led()
else:
self.choose_next_device()
elif device == self.powerplug_common and key == self.KEY_POWERPLUG_AMPLIFIER:
if self.cvi.changed_here(device.topic, key, data):
if data is True:
self.active_device_state = self.STATE_ACTIVE_DEVICE_AMPLIFIER
self.update_active_device_led()
else:
self.choose_next_device()
def get_activity_state(self, state):
if state == self.STATE_ACTIVE_DEVICE_MAIN_LIGHT:
return self.main_light_shelly.output_0
elif state == self.STATE_ACTIVE_DEVICE_DESK_LIGHT:
2022-12-23 11:45:00 +01:00
return self.powerplug_common.get(self.KEY_POWERPLUG_DESK_LIGHT)
elif state == self.STATE_ACTIVE_DEVICE_AMPLIFIER:
2022-12-23 11:45:00 +01:00
return self.powerplug_common.get(self.KEY_POWERPLUG_AMPLIFIER)
def update_active_device_led(self):
self.gui_led_active_device.set_led(self.LED_ACTIVE_DEVICE_AMPLIFIER, self.active_device_state == self.STATE_ACTIVE_DEVICE_AMPLIFIER)
self.gui_led_active_device.set_led(self.LED_ACTIVE_DEVICE_MAIN_LIGHT, self.active_device_state == self.STATE_ACTIVE_DEVICE_MAIN_LIGHT)
self.gui_led_active_device.set_led(self.LED_ACTIVE_DEVICE_DESK_LIGHT, self.active_device_state == self.STATE_ACTIVE_DEVICE_DESK_LIGHT)
2022-12-23 14:00:49 +01:00
def choose_prev_device(self, device=None, key=None, data=None):
if self.active_device_state is not None:
start_value = self.active_device_state
for i in range(0, self.STATE_ACTIVE_DEVICE_MAX_VALUE + 1):
target_state = (start_value + i + 1) % (self.STATE_ACTIVE_DEVICE_MAX_VALUE + 1)
if self.get_activity_state(target_state):
self.active_device_state = target_state
self.update_active_device_led()
return
self.active_device_state = None
self.update_active_device_led()
2022-12-23 14:00:49 +01:00
def choose_next_device(self, device=None, key=None, data=None):
if self.active_device_state is not None:
start_value = self.active_device_state
for i in range(0, self.STATE_ACTIVE_DEVICE_MAX_VALUE + 1):
target_state = (start_value - i - 1) % (self.STATE_ACTIVE_DEVICE_MAX_VALUE + 1)
if self.get_activity_state(target_state):
self.active_device_state = target_state
self.update_active_device_led()
return
self.active_device_state = None
self.update_active_device_led()
def brightness_action(self, device, key, data):
if self.active_device_state is not None:
brightness_func = {
self.STATE_ACTIVE_DEVICE_MAIN_LIGHT: self.main_light_tradfri,
self.STATE_ACTIVE_DEVICE_DESK_LIGHT: self.desk_light_tradfri,
self.STATE_ACTIVE_DEVICE_AMPLIFIER: self.remote_amplifier
}
target = brightness_func[self.active_device_state]
if target is None:
logger.error("Not yet implemented")
return
if data == devices.tradfri_button.ACTION_BRIGHTNESS_UP_LONG:
logger.info("Increasing \"%s\" - %s", type(self).__name__, target.topic)
target.default_inc()
elif data == devices.tradfri_button.ACTION_BRIGHTNESS_DOWN_LONG:
logger.info("Decreasing \"%s\" - %s", type(self).__name__, target.topic)
target.default_dec()
elif data in [devices.tradfri_button.ACTION_BRIGHTNESS_UP_RELEASE, devices.tradfri_button.ACTION_BRIGHTNESS_DOWN_RELEASE]:
target.default_stop()
def audio_source_selector(self, device, key, data):
if device == self.powerplug_common and key == self.KEY_POWERPLUG_CD_PLAYER:
if self.cvi.changed_here(device.topic, key, data) and data is True:
# switch on of cd player
self.audio_source = self.AUDIO_SOURCE_CD
elif device in [self.spotify_state, self.mpd_state]:
if self.cvi.changed_here(device.topic, key, data) and data is True:
# switch on raspi-source
self.audio_source = self.AUDIO_SOURCE_RASPI
elif device == self.powerplug_common and key == self.KEY_POWERPLUG_AMPLIFIER:
if self.cvi.changed_here(device.topic, key, data) and data is True:
# switch on of amplifier -> select source and reset stored source value
self.delayed_task.run()
def send_audio_source(self):
if self.audio_source == self.AUDIO_SOURCE_PC:
logger.info("Sending IR command to change audio source to pc")
self.remote_amplifier.set_line3()
elif self.audio_source == self.AUDIO_SOURCE_CD:
logger.info("Sending IR command to change audio source to cd")
self.remote_amplifier.set_cd()
elif self.audio_source == self.AUDIO_SOURCE_RASPI:
logger.info("Sending IR command to change audio source to raspi")
self.remote_amplifier.set_line1()
self.audio_source = self.AUDIO_SOURCE_PC