pi_hat.py

#!/usr/bin/env python
#############################
#
#    copyright 2021 Cascoda Ltd.
#    Redistribution and use in source and binary forms, with or without modification,
#    are permitted provided that the following conditions are met:
#    1.  Redistributions of source code must retain the above copyright notice,
#        this list of conditions and the following disclaimer.
#    2.  Redistributions in binary form must reproduce the above copyright notice,
#        this list of conditions and the following disclaimer in the documentation
#        and/or other materials provided with the distribution.
#
#    THIS SOFTWARE IS PROVIDED "AS IS"
#    AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
#    THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE OR
#    WARRANTIES OF NON-INFRINGEMENT, ARE DISCLAIMED. IN NO EVENT SHALL THE
#    CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,
#    OR CONSEQUENTIAL DAMAGES
#    (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
#    LOSS OF USE, DATA, OR PROFITS;OR BUSINESS INTERRUPTION)
#    HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
#    OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
#    EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#
#############################


# pylint: disable=C0114
# pylint: disable=C0116
# pylint: disable=C0209

# pylint: disable=E0401
# pylint: disable=R0402
# pylint: disable=C0411

import dothat.backlight as backlight
import dothat.lcd as lcd
import dothat.touch as touch

import fcntl
import socket
import struct
from time import sleep


try:
    import knx
except:
    print("Could not find KNX Module")
    pass

IDLE_BL = (128, 128, 128)
DARK_BL = (0, 0, 0)
FULL_BL = (0, 255, 0)
BLINK_TIME_S = 0.03

def on_release(buttons, bounce=-1):
    """
    Decorator. Use with @on_release(UP)
    Args:
      buttons (list): List, or single instance of cap touch button constant
      bounce (int): Maintained for compatibility with Dot3k joystick, unused
    """
    buttons = buttons if isinstance(buttons, list) else [buttons]
    
    def register(handler):
        for button in buttons:
            touch._cap1166.on(channel=button, event='release', handler=handler)
    
    return register

def get_addr(ifname):
    try:
        my_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
        return socket.inet_ntoa(fcntl.ioctl(
            my_socket.fileno(),
            0x8915,  # SIOCGIFADDR
            struct.pack('256s', ifname[:15].encode('utf-8'))
        )[20:24])
    except IOError:
        return 'Not Found!'

def init():
    lcd.clear()
    # turn off the blinding bright LEDS to the left of the screen
    backlight.graph_off()

    eth0 = get_addr('eth0')
    host = socket.gethostname()

    lcd.set_cursor_position(0,0)
    lcd.write('{}'.format(host))

    lcd.set_cursor_position(0,1)
    if eth0 != 'Not Found!':
        lcd.write(eth0)
    else:
        lcd.write('eth0 {}'.format(eth0))

def init_client():
    backlight.rgb(*IDLE_BL)
    lcd.clear()
    # turn off the blinding bright LEDS to the left of the screen
    backlight.graph_off()

    eth0 = get_addr('eth0')
    host = socket.gethostname()

    lcd.set_cursor_position(0,0)
    lcd.write('{}'.format(host))

    lcd.set_cursor_position(0,1)
    if eth0 != 'Not Found!':
        lcd.write(eth0)
    else:
        lcd.write('eth0 {}'.format(eth0))

def print_in_python():
    print("Python binding was succesful!")

def set_backlight(value):
    print("Set backlight to {}".format(value))
    if value is True:
        backlight.rgb(255, 255, 255)
    else:
        backlight.rgb(0, 0, 0)

def set_led(led, value):
    print("Set led {} to {}".format(led, value))
    backlight.graph_set_led_state(led, value)

@touch.on(touch.LEFT)
def handle_left(_ch, _evt):
    backlight.left_rgb(*FULL_BL)
    knx.handle_left()
    sleep(BLINK_TIME_S)
    backlight.rgb(*IDLE_BL)

@touch.on(touch.BUTTON)
def handle_mid(_ch, _evt):
    backlight.mid_rgb(*FULL_BL)
    knx.handle_mid()
    sleep(BLINK_TIME_S)
    backlight.rgb(*IDLE_BL)

@touch.on(touch.RIGHT)
def handle_right(_ch, _evt):
    backlight.right_rgb(*FULL_BL)
    knx.handle_right()
    sleep(BLINK_TIME_S)
    backlight.rgb(*IDLE_BL)

@touch.on(touch.UP)
def handle_up(_ch, _evt):
    backlight.right_rgb(*FULL_BL)
    knx.handle_up()
    sleep(BLINK_TIME_S)
    backlight.rgb(*IDLE_BL)

@touch.on(touch.DOWN)
def handle_down(_ch, _evt):
    backlight.right_rgb(*FULL_BL)
    knx.handle_down()
    sleep(BLINK_TIME_S)
    backlight.rgb(*IDLE_BL)
    
@touch.on(touch.CANCEL)
def handle_cancel(_ch, _evt):
    backlight.right_rgb(*FULL_BL)
    knx.handle_cancel()
    sleep(BLINK_TIME_S)
    backlight.rgb(*IDLE_BL)