nspanel-lovelace-ui/nspanel-lovelace-ui/rootfs/usr/bin/mqtt-manager/panel.py

from dateutil import tz
import datetime
import threading
import logging
import libs.panel_cmd
from scheduler import Scheduler
import scheduler.trigger as trigger
import time
import babel.dates
from ha_cards import Screensaver, EntitiesCard, card_factory
import ha_control


class LovelaceUIPanel:

    def __init__(self, mqtt_client_from_manager, name_panel, settings_panel):
        self.mqtt_client = mqtt_client_from_manager
        self.name = name_panel
        self.settings = settings_panel
        self.sendTopic = self.settings["panelSendTopic"]

        self.current_card = None
        self.privious_cards = []
        self.cards = {}
        self.hidden_cards = {}
        self.navigate_keys = {}
        self.entity_iids = {}

        # generate cards for input settings
        for c in self.settings.get("cards"):
            iid, card = card_factory(self.settings["locale"], c, self)
            self.cards[iid] = card
            # collect nav keys of cards
            if card.navigate_key:
                self.navigate_keys[card.navigate_key] = iid
            # collect iids of entities
            for e in card.get_iid_entities():
                self.entity_iids[e[0]] = e[1]

        # setup prev and next iids
        top_level_cards = list(self.cards.values())
        card_iids = [card.iid for card in top_level_cards]
        prev_iids = card_iids[-1:] + card_iids[:-1]
        next_iids = card_iids[1:] + card_iids[: 1]
        if len(card_iids) > 1:
            for prev_iids, card, next_iids in zip(prev_iids, top_level_cards, next_iids):
                (card.iid_prev, card.iid_next) = (prev_iids, next_iids)

        # generate cards for input settings
        for c in self.settings.get("hiddenCards", []):
            iid, card = card_factory(self.settings["locale"], c, self)
            self.hidden_cards[iid] = card
            # collect nav keys of cards
            if card.navigate_key:
                self.navigate_keys[card.navigate_key] = iid
            # collect iids of entities
            for e in card.get_iid_entities():
                self.entity_iids[e[0]] = e[1]

        self.schedule = Scheduler()
        self.schedule.minutely(datetime.time(second=0), self.update_time)
        self.schedule.hourly(datetime.time(
            minute=0, second=0), self.update_time)
        schedule_thread = threading.Thread(target=self.schedule_thread_target)
        schedule_thread.daemon = True
        schedule_thread.start()

    def schedule_thread_target(self):
        while True:
            self.schedule.exec_jobs()
            time.sleep(1)

    def update_time(self):
        use_timezone = tz.gettz("Europe/Berlin")
        time_string = datetime.datetime.now(
            use_timezone).strftime(self.settings["timeFormat"])
        libs.panel_cmd.send_time(self.sendTopic, time_string)

    def update_date(self):
        dateformat = self.settings["dateFormat"]
        date_string = babel.dates.format_date(
            datetime.datetime.now(), dateformat, locale=self.settings["locale"])
        libs.panel_cmd.send_date(self.sendTopic, date_string)

    def searchCard(self, iid):
        print(f"searchcard {iid} {self.navigate_keys}")
        if iid in self.navigate_keys:
            iid = self.navigate_keys[iid]
        if iid in self.cards:
            return self.cards[iid]
        if iid in self.hidden_cards:
            return self.hidden_cards[iid]

    def ha_event_callback(self, entity_id):
        logging.debug(f"{entity_id} updated/state changed")
        # TODO: Check if entity is on current card
        libs.panel_cmd.entityUpd(self.sendTopic, self.current_card.render())

    def customrecv_event_callback(self, msg):
        logging.debug("Recv Message from NsPanel: %s", msg)
        msg = msg.split(",")
        # run action based on received command
        if msg[0] == "event":
            if msg[1] == "startup":
                # TODO: Handle Update Messages
                self.update_date()
                self.update_time()

                libs.panel_cmd.page_type(self.sendTopic, "screensaver")
                self.current_card = Screensaver(
                    self.settings["locale"], self.settings["screensaver"], self)
                libs.panel_cmd.entityUpd(
                    self.sendTopic, self.current_card.render())

            if msg[1] == "sleepReached":
                self.privious_cards.append(self.current_card)
                libs.panel_cmd.page_type(self.sendTopic, "screensaver")
                self.current_card = Screensaver(
                    self.settings["locale"], self.settings["screensaver"], self)
                libs.panel_cmd.entityUpd(
                    self.sendTopic, self.current_card.render())
            if msg[1] == "buttonPress2":
                entity_id = msg[2]
                btype = msg[3]
                value = msg[4] if len(msg) > 4 else None

                if btype == "bExit":
                    # TODO: Impelement default card
                    # in case privious_cards is empty add a default card
                    if len(self.privious_cards) == 0:
                        self.privious_cards.append(
                            list(self.cards.values())[0])
                    self.current_card = self.privious_cards.pop()
                    libs.panel_cmd.page_type(
                        self.sendTopic, self.current_card.type)
                    libs.panel_cmd.entityUpd(
                        self.sendTopic, self.current_card.render()
                    )

                # replace iid with real entity id
                if entity_id.startswith("iid."):
                    iid = entity_id.split(".")[1]
                    if iid in self.entity_iids:
                        entity_id = self.entity_iids[iid]

                if btype == "button":
                    match entity_id.split(".")[0]:
                        case 'navigate':
                            iid = entity_id.split(".")[1]
                            self.privious_cards.append(self.current_card)
                            self.current_card = self.searchCard(iid)
                            libs.panel_cmd.page_type(
                                self.sendTopic, self.current_card.type)
                            libs.panel_cmd.entityUpd(
                                self.sendTopic, self.current_card.render())
                        case _:
                            ha_control.button_press(entity_id, value)
                if btype == "OnOff":
                    ha_control.on_off(entity_id, value)
                if btype == "number-set":
                    ha_control.number_set(entity_id, value)

            if msg[1] == "pageOpenDetail":
                print("pageOpenDetail")