hassconfig/pyscript/state_goto.py

from homeassistant.const import EVENT_STATE_CHANGED

def getCurrentState():
    if( light.main_button_1 == 'on' and     # All ON
        light.main_button_2 == 'on' and
        light.main_button_3 == 'on' and
        light.main_button_4 == 'on' ):
        return 1
    elif( light.main_button_1 == 'off' and  # Window
        light.main_button_2 == 'off' and
        light.main_button_3 == 'off' and
        light.main_button_4 == 'on' ):
        return 2
    elif( light.main_button_1 == 'off' and  # Morning
        light.main_button_2 == 'on' and
        light.main_button_3 == 'off' and
        light.main_button_4 == 'on' ):
        return 3
    elif( light.main_button_1 == 'off' and  # All OFF
        light.main_button_2 == 'off' and
        light.main_button_3 == 'off' and
        light.main_button_4 == 'off' ):
        return 4

@service
def state_goto_all_on():
    task.unique('state_goto_py')
    cs = getCurrentState()
    log.info(f"********* ALL ON *********** State: " + str(cs) )
    light.turn_on(entity_id='light.tradfri_bulb')   # Stora flyglampan
    light.turn_on(entity_id='light.hall_inner')
    light.turn_on(entity_id='light.hall_door')
    #switch.turn_on(entity_id='switch.matsal')
    switch.turn_on(entity_id='switch.koksfonster')
    for x in range(2):
        switch.turn_on(entity_id='switch.group_button_all')
        task.sleep(0.5)
        #light.turn_on(entity_id='light.liv_room_table_lamp',brightness='1') # Temp disable when high energy prices
        #task.sleep(0.5)
        light.turn_on(entity_id='light.liv_room_corner_lamp',brightness='1')
        task.sleep(0.5)

    kitchen_worklights_button_trigger()
    #pyscript.handle_bedroom_light()   # Temp disable when high energy prices


@service
def state_goto_window():
    task.unique('state_goto_py')
    cs = getCurrentState()
    log.info(f"********* WINDOW *********** State: " + str(cs) )
    light.turn_off(entity_id='light.tradfri_bulb')
    #switch.turn_off(entity_id='switch.matsal')
    switch.turn_on(entity_id='switch.koksfonster')
    light.turn_off(entity_id='light.hall_inner')
    light.turn_off(entity_id='light.hall_door')

    
    if cs == 4 or cs == 2:  # From Off or already Window
        for x in range(2):
            light.turn_on(entity_id=' light.main_button_4')
            task.sleep(0.5)
    elif cs == 3:           # From morning
        for x in range(2):
            light.turn_off(entity_id=' light.main_button_2')
            task.sleep(0.5)
            light.turn_off(entity_id='light.liv_room_corner_lamp')
            task.sleep(0.5)
    else:                   # From ALL ON
        for x in range(2):
            switch.turn_off(entity_id='switch.group_button_all')
            task.sleep(0.5)
            light.turn_off(entity_id='light.liv_room_table_lamp')
            task.sleep(0.5)
            light.turn_off(entity_id='light.liv_room_corner_lamp')
            task.sleep(0.5)
        for x in range(2):
            light.turn_on(entity_id=' light.main_button_4')
            task.sleep(0.5)
    
    kitchen_worklights_button_trigger()
    #pyscript.handle_bedroom_light()

@service
def state_goto_morning():
    task.unique('state_goto_py')
    cs = getCurrentState()
    log.info(f"********* MORNING *********** State: " + str(cs) )
    light.turn_off(entity_id='light.tradfri_bulb')
    #switch.turn_off(entity_id='switch.matsal')
    switch.turn_on(entity_id='switch.koksfonster')
    light.turn_off(entity_id='light.hall_inner')
    light.turn_on(entity_id='light.hall_door')

    if cs == 4 or cs == 3:  # Off or already Morning
        for x in range(2):
            light.turn_on(entity_id=' light.main_button_4')
            task.sleep(0.5)
            light.turn_on(entity_id=' light.main_button_2')
            task.sleep(0.5)
            light.turn_on(entity_id='light.liv_room_corner_lamp',brightness='1')
            task.sleep(0.5)
    elif cs == 2:           # From Window
        for x in range(2):
            light.turn_on(entity_id=' light.main_button_2')
            task.sleep(0.5)
            light.turn_on(entity_id='light.liv_room_corner_lamp',brightness='1')
            task.sleep(0.5)
    else:                   # From ALL ON
        for x in range(2):
            switch.turn_off(entity_id='switch.group_button_all')
            task.sleep(0.5)
        for x in range(2):
            light.turn_on(entity_id=' light.main_button_2')
            task.sleep(0.5)
            light.turn_on(entity_id=' light.main_button_4')
            task.sleep(0.5)
            light.turn_on(entity_id='light.liv_room_corner_lamp',brightness='1')
            task.sleep(0.5)
    
    kitchen_worklights_button_trigger()
    #pyscript.handle_bedroom_light()

@service
def state_goto_all_off():
    task.unique('state_goto_py')
    cs = getCurrentState()
    log.info(f"********* ALL OFF *********** State: " + str(cs) )
    light.turn_off(entity_id='light.tradfri_bulb')
    light.turn_off(entity_id='light.hall_inner')
    light.turn_off(entity_id='light.hall_door')

    switch.turn_off(entity_id='switch.koksfonster')
    #switch.turn_off(entity_id='switch.matsal')  # Temp disable when high energy prices
    #switch.turn_off(entity_id='switch.tradfri_outlet')  # Sovrum OFF  # Temp disable when high energy prices
    for x in range(2):
        switch.turn_off(entity_id='switch.group_button_all')
        task.sleep(0.5)
        #light.turn_off(entity_id='light.liv_room_table_lamp')  # Temp disable when high energy prices
        #task.sleep(0.5)
        light.turn_off(entity_id='light.liv_room_corner_lamp')
        task.sleep(0.5)
    input_boolean.movie_mode = 'off'
    kitchen_worklights_button_trigger()

# Reacts to when the Kitchen Worklights button is changed
# This could be from HA UI or from web-interface
@state_trigger("input_boolean.kitchen_worklights")
def kitchen_worklights_button_trigger():
    log.info(f"Kitchen worklight changed. State: {input_boolean.kitchen_worklights}")
    if( input_boolean.kitchen_worklights=='on' or light.hall_door == 'on' ):
        log.info("Kitchen LED: Sec switch ON")
        if( switch.kok_led_sec_switch == 'on' ):
            log.info("K LED Sec Switch already On, go directly to PWW-Control")
            handle_kitchen_worklights()
        else:
            switch.turn_on(entity_id='switch.kok_led_sec_switch')
    else:
        log.info("Kitchen LED: Sec switch OFF")
        switch.turn_off(entity_id='switch.kok_led_sec_switch')

# Triggers when the IKEA Switch changes state. Goes On or Off
@event_trigger(EVENT_STATE_CHANGED, "entity_id=='switch.kok_led_sec_switch'")
def kitchen_worklights_security_switch_trigger(entity_id, new_state, old_state):
    log.info(f"kitchen_worklights_security_switch_trigger() {switch.kok_led_sec_switch}")
    if( switch.kok_led_sec_switch=='on' ):
        log.info("Kitchen LED: Control PWM Wait...")
        task.sleep(0.5)
        log.info("Kitchen LED: Go....")
    handle_kitchen_worklights()

# Controls the PWM of the kitchen workbench LED-Strips
# Does not control the IKEA Switch that turns everything On/Off
def handle_kitchen_worklights():
    log.info(f"handle_kitchen_worklights WL:{input_boolean.kitchen_worklights} HALL:{light.hall_door}")
    if( input_boolean.kitchen_worklights == 'on' ):
        log.info("K LED Full mode...")
        light.turn_on(entity_id='light.kitchen_worktop_4k',brightness='63')
        light.turn_on(entity_id='light.kitchen_worktop_27k',brightness='127')
    elif( light.hall_door == 'on' ):
        log.info("K LED Soft mode...")
        light.turn_off(entity_id='light.kitchen_worktop_4k')
        light.turn_on(entity_id='light.kitchen_worktop_27k',brightness='12')
    else:
        log.info("K LED Off mode...")
        light.turn_off(entity_id='light.kitchen_worktop_4k')
        light.turn_off(entity_id='light.kitchen_worktop_27k')




@state_trigger("input_boolean.movie_mode")
def movie_mode_button_trigger():
    cs = getCurrentState()
    log.info(f"********* MOVIE MODE *********** To:" + input_boolean.movie_mode + " State: " + str(cs) )
    if input_boolean.movie_mode == 'off' and switch.group_button_all == 'on':
        log.info(f"Movie mode, turn lights ON")
        for x in range(2):
            light.turn_on(entity_id='light.liv_room_table_lamp',brightness='1')
            task.sleep(0.5)
            light.turn_on(entity_id='light.liv_room_corner_lamp',brightness='1')
            task.sleep(0.5)
 
    else:
        log.info(f"Movie mode, turn lights OFF")
        for x in range(2):
            light.turn_off(entity_id='light.liv_room_table_lamp')
            task.sleep(0.5)
            light.turn_off(entity_id='light.liv_room_corner_lamp')
            task.sleep(0.5)


@event_trigger(EVENT_STATE_CHANGED, "entity_id=='input_button.av'")
def on_ui_button_all_off_clicken(entity_id, new_state, old_state):
    pyscript.state_goto_all_off()

@event_trigger(EVENT_STATE_CHANGED, "entity_id=='input_button.allt_pa'")
def on_ui_button_all_on_clicken(entity_id, new_state, old_state):
    pyscript.state_goto_all_on()