55 lines
2.0 KiB
Python
55 lines
2.0 KiB
Python
|
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))
|