uhr/uhr.py

#!/usr/bin/python3

import datetime
import os
import shlex
import subprocess
import threading
import time

import gpiozero
import toml
import neopixel
import board

SKRIPTPFAD = os.path.abspath(os.path.dirname(__file__))


def load_config(pfad=SKRIPTPFAD):
    configfile = os.path.join(pfad, "uhr_cfg.toml")
    with open(configfile) as conffile:
        config = toml.loads(conffile.read())
    return config


CONFIG = load_config()


# LED strip configuration
LED_COUNT = 60      # Number of LED pixels.
LED_PIN = board.D18      # GPIO pin connected to the pixels (must support PWM!).
LED_PIXEL_ORDER = neopixel.GRB   # Strip type and colour ordering

# Zahl stellt den Wert in Minuten dar, wie lange kein Gerät erreichbar sein darf dass die Uhr "abgeschalten" wird
ABSCHALTWERT = 4

# LED Mapping
std0 = [0, 1, 59]
std1 = [4, 5, 6]
std2 = [9, 10, 11]
std3 = [14, 15, 16]
std4 = [19, 20, 21]
std5 = [24, 25, 26]
std6 = [29, 30, 31]
std7 = [34, 35, 36]
std8 = [39, 40, 41]
std9 = [44, 45, 46]
std10 = [49, 50, 51]
std11 = [54, 55, 56]
stdliste = [std0, std1, std2, std3, std4, std5, std6, std7, std8, std9, std10, std11]


class Uhr:
    def __init__(self, pixels, mode):
        self.helligkeit = pixels.brightness
        self.pixels = pixels
        self.helligkeit_geaendert = None
        self.mode = mode
        self._rgb_leer = None
        self._rgb_sekunde = None
        self._rgb_minute = None
        self._rgb_stunde = None

    def set_helligkeit(self, helligkeit):
        self.helligkeit = helligkeit
        self.pixels.brightness = self.helligkeit
        self.helligkeit_geaendert = datetime.datetime.now()

    def helligkeit_erhoehen(self):
        self.helligkeit += 0.05
        if self.helligkeit > 1:
            self.helligkeit = 0
        self.set_helligkeit(self.helligkeit)

    def helligkeit_verringern(self):
        self.helligkeit -= 0.05
        if self.helligkeit < 0:
            self.helligkeit = 1
        self.set_helligkeit(self.helligkeit)

    def rgb_farben_lesen(self):
        self._rgb_leer = CONFIG["mode"][self.mode]["leer"]
        self._rgb_sekunde = CONFIG["mode"][self.mode]["sekunde"]
        self._rgb_minute = CONFIG["mode"][self.mode]["minute"]
        self._rgb_stunde = CONFIG["mode"][self.mode]["stunde"]

        rgbconf = {"rgb_leer": self._rgb_leer, "rgb_s": self._rgb_sekunde, "rgb_min": self._rgb_minute,
                   "rgb_std": self._rgb_stunde}
        print("In der Klasse:")
        print(rgbconf)
        return rgbconf


# GPIO
I_MODE_TASTER = gpiozero.Button(3)
I_PLUS_TASTER = gpiozero.Button(4)
I_MINUS_TASTER = gpiozero.Button(17)


def config_schreiben(pfad=SKRIPTPFAD):
    configfile = os.path.join(pfad, "uhr_cfg.toml")
    with open(configfile, "w") as conffile:
        conffile.write(toml.dumps(CONFIG))


def alle_led(r, g, b, pixels):
    pixels.fill((r, g, b))
    pixels.show()


def led_calc(zeit, stdliste, rgbdict, led_gesetzt, pixels):
    if zeit.hour > 11:
        index = zeit.hour - 12
    else:
        index = zeit.hour
    hour = stdliste[index]
    minute = zeit.minute
    second = zeit.second

    # Schnittfarben berechnen falls "Zeiger" übereinander liegen
    # Stunde mit Minute vergleichen

    if bool(set(hour) & {minute}):
        for counter in range(0, 3):
            rgbdict["rgb_min"][counter] = int((rgbdict["rgb_std"][counter] + rgbdict["rgb_min"][counter]) / 2)
    # Stunde mit Sekunde vergleichen
    if set(hour) & {second}:
        for counter in range(0, 3):
            rgbdict["rgb_s"][counter] = int((rgbdict["rgb_std"][counter] + rgbdict["rgb_s"][counter]) / 2)
    # Minute mit Sekunde vergleichen
    if {minute} & {second}:
        for counter in range(0, 3):
            rgbdict["rgb_s"][counter] = int((rgbdict["rgb_min"][counter] + rgbdict["rgb_s"][counter]) / 2)
    led_gesetzt = led_setzen(hour, minute, second, rgbdict, led_gesetzt, pixels)
    return zeit.second, led_gesetzt


def led_setzen(hour, minute, second, rgbdict, led_gesetzt, pixels):
    if not isinstance(led_gesetzt, list):
        led_gesetzt = [led_gesetzt]
    # Stunden setzen
    for stunde in hour:
        pixels[stunde] = rgbdict["rgb_std"][0], rgbdict["rgb_std"][1], rgbdict["rgb_std"][2]

    # Minute setzen
    pixels[minute] = rgbdict["rgb_min"][0], rgbdict["rgb_min"][1], rgbdict["rgb_min"][2]

    # Sekunde setzen
    pixels[second] = rgbdict["rgb_s"][0], rgbdict["rgb_s"][1], rgbdict["rgb_s"][2]

    # Neue Leere LED berechnen
    led_gesetzt_neu = set(hour) | {minute} | {second}
    leerliste = list(set(led_gesetzt) - led_gesetzt_neu)
    for leer in leerliste:
        pixels[leer] = rgbdict["rgb_leer"][0], rgbdict["rgb_leer"][1], rgbdict["rgb_leer"][2]

    pixels.show()
    return list(led_gesetzt_neu)


def shutdown():
    cmd = "sudo shutdown now"
    cmd = shlex.split(cmd)
    subprocess.call(cmd)


# Threads
def check_anwesenheit(uhr, pixels):
    """Funktion, welche als eigener Thread laeuft, um selbststaendig in einem gewissenen Intervall
    alle Geraete in der Toml Liste zu pingen
    arg: Objekt des neopixel LED Ringes
    toml File: status "anwesend", ist kein Geraet von "anwesend" oder "dimmen" erreichbar, LED Helligkeit auf 0
    sobald eine Adresse von status "dimmen" erreichbar ist, wird die Helligkeit verringert"""

    def ping_ping(ip):
        """pingt die IP 2x an
        return (0 | !0) 0 wenn erreichbar"""
        befehl = "ping -c2 -W1 {}".format(ip)
        cmd = shlex.split(befehl)
        return subprocess.call(cmd)

    def ping_bt(bt):
        """pingt die IP 2x an
        return (0 | !0) 0 wenn erreichbar"""
        befehl = "sudo /usr/bin/l2ping -c1 -t1 {}".format(bt)
        cmd = shlex.split(befehl)
        return subprocess.call(cmd)

    def ping_arping(ip, interface):
        """pingt die IP 3x an
               return (0 | !0) 0 wenn erreichbar
        """
        befehl = "/usr/bin/arping -q -c 3 -w 10 -b -f -I {interface} {ip}".format(interface=interface, ip=ip)
        cmd = shlex.split(befehl)
        return subprocess.call(cmd)

    wlanliste = CONFIG["ping"]
    interface = CONFIG["interface"]
    status_anwesend_liste = [] 
    delta = datetime.timedelta(seconds=301)
    last = datetime.datetime.now()
    status = {}
    while True:
        helligkeit = uhr.helligkeit
        now = datetime.datetime.now()
        # Status der Geräte ermitteln
        if delta.seconds > 300:
            for key_status in wlanliste.keys():
                for key_funkart in wlanliste[key_status].keys():
                    for ip in wlanliste[key_status][key_funkart]:
                        if key_funkart == "ping":
                            status_return = ping_ping(ip)
                        elif key_funkart == "bt":
                            status_return = ping_bt(ip)
                        elif key_funkart == "arping":
                            status_return = ping_arping(ip, interface)
                        else:
                            status_return = False
                        if not status_return:
                            status[key_status] = True
                            break
                        else:
                            status[key_status] = False
            if status["anwesend"]:
                status_anwesend_liste = []  # Geraet von anwesend erreichbar
                helligkeit = uhr.helligkeit
            elif not status["anwesend"] and not status["dimmen"]:  # Wenn kein Geraet erreichbar ist
                if len(status_anwesend_liste) < ABSCHALTWERT + 1:
                    status_anwesend_liste.append(status["anwesend"])
                print("Nichts los daheim die ", len(status_anwesend_liste), ".")
                if len(status_anwesend_liste) > ABSCHALTWERT: 
                    helligkeit = 0  
            if status["dimmen"]:
                helligkeit = 0.03
            if now.hour < 5 and not status["dimmen"]:
                helligkeit = 0
            pixels.brightness = helligkeit
            last = datetime.datetime.now()
        delta = now - last
        time.sleep(30)


def main():
    lastsecond = None
    led_gesetzt = 0

    pixels = neopixel.NeoPixel(LED_PIN, LED_COUNT, brightness=CONFIG["led_helligkeit"],
                               auto_write=False, pixel_order=LED_PIXEL_ORDER)
    uhr = Uhr(pixels, CONFIG["mode_nummer"])
    thread_check_wlan = threading.Thread(target=check_anwesenheit, args=(uhr, pixels))
    thread_check_wlan.start()

    rgbconf = uhr.rgb_farben_lesen()
    print(rgbconf)
    alle_led(rgbconf["rgb_leer"][0], rgbconf["rgb_leer"][1], rgbconf["rgb_leer"][2], pixels)

    I_PLUS_TASTER.when_pressed = uhr.helligkeit_erhoehen
    I_MINUS_TASTER.when_pressed = uhr.helligkeit_verringern

    try:
        while True:
            rgbconf = uhr.rgb_farben_lesen()
            print(rgbconf)
            now = datetime.datetime.now()
            if lastsecond != now.second:
                lastsecond, led_gesetzt = led_calc(now, stdliste, rgbconf, led_gesetzt, pixels)
            if uhr.helligkeit_geaendert is not None:
                if (datetime.datetime.now() - uhr.helligkeit_geaendert) > datetime.timedelta(seconds=30):
                    CONFIG["led_helligkeit"] = uhr.helligkeit
                    config_schreiben()
                    uhr.helligkeit_geaendert = None
            time.sleep(0.2)
    finally:
        pixels.brightness = 0
        alle_led(0, 0, 0, pixels)


if __name__ == "__main__":
    main()