christmas.py

from pwnagotchi.ui.components import LabeledValue
from pwnagotchi.ui.view import BLACK
import pwnagotchi.ui.fonts as fonts
import pwnagotchi.plugins as plugins
import pwnagotchi
import logging
import datetime
import math
import yaml


class Christmas(plugins.Plugin):
    __author__ = 'https://github.com/LoganMD'
    __version__ = '1.2.0'
    __license__ = 'GPL3'
    __description__ = 'Christmas Countdown timer for pwnagotchi'

    def on_loaded(self):
        logging.info("Christmas Plugin loaded.")

    def on_ui_setup(self, ui):
        memenable = False
        with open('/etc/pwnagotchi/config.yml') as f:
            data = yaml.load(f, Loader=yaml.FullLoader)

            if 'memtemp' in data["main"]["plugins"]:
                if 'enabled' in data["main"]["plugins"]["memtemp"]:
                    if data["main"]["plugins"]["memtemp"]["enabled"]:
                        memenable = True
                        logging.info("Christmas Plugin: memtemp is enabled")
        if ui.is_waveshare_v2():
            pos = (130, 80) if memenable else (200, 80)
            ui.add_element('christmas', LabeledValue(color=BLACK, label='', value='christmas\n',
                                                     position=pos,
                                                     label_font=fonts.Small, text_font=fonts.Small))

    def on_ui_update(self, ui):
        now = datetime.datetime.now()
        christmas = datetime.datetime(now.year, 12, 25)
        if now > christmas:
            christmas = christmas.replace(year=now.year + 1)

        difference = (christmas - now)

        days = difference.days
        hours = difference.seconds // 3600
        minutes = (difference.seconds % 3600) // 60

        if now.month == 12 and now.day == 25:
            ui.set('christmas', "merry\nchristmas!")
        elif days == 0:
            ui.set('christmas', "christmas\n%dH %dM" % (hours, minutes))
        else:
            ui.set('christmas', "christmas\n%dD %dH" % (days, hours))