countdownwidget.py

#!/usr/bin/env python3

from rgbmatrix import RGBMatrix, RGBMatrixOptions
from widget import Widget
from PIL import Image, ImageDraw, ImageFont
import time
import re


class CountdownWidget(Widget):
    def __init__(self,x=0,y=0,color=None,size=14,width=64,height=64,font=None,bigat=-1):
        if color is None:
            self.dynamiccolor = True
        else:
            self.dynamiccolor = False
        super(CountdownWidget,self).__init__(x,y,color,size,width,height)
        if font:
            self.font = ImageFont.load("fonts/"+font)
        else:
            self.font = ImageFont.load("fonts/"+self.size2font(size))
        
        self.bigat = bigat
        self.bigfont = ImageFont.truetype(font="RobotoCondensed-Light.ttf",size=height)


        self.starttime = 0
        self.endtime = 0
        self.running = False
        self.started = False
        self.ended = False
        self.cancelled = False
        self.lasttime= 0

    def start(self,minutes=0,seconds=0):
        if (minutes > 0 or seconds > 0):
            self.starttime = int(time.time())
            self.endtime   = self.starttime + (60*minutes) + seconds
            self.running   = True
            self.started   = True

    def cancel(self):
        if (self.running):
            self.running = False
            self.started = False
            self.ended = False
            self.cancelled = True

    def mqttstart(self,topic=None,value=None):
        v = value.decode()
        m1 = re.match("(\d+)\s+(\d+)",v)
        m2 = re.match("^\d+$",v)
        at = re.match("(\d+):(\d+)",v)
        if (v == "cancel"):
            self.cancel()
        elif m1:
            minutes = int(m1.group(1))
            seconds = int(m1.group(2))
            self.start(minutes,seconds)
        elif m2:
            seconds = int(m2.group(0))
            self.start(seconds = seconds)
        elif at:
            lthen = list(time.localtime())
            lthen[3]=int(at.group(1))
            lthen[4]=int(at.group(2)) # minutes
            lthen[5]=0
            then = time.mktime(tuple(lthen))
            self.start(seconds=(then-int(time.time())))


    def big(self):
        now = int(time.time())
        seconds_left = int(self.endtime - now)
        if seconds_left < 0:
            self.running = False
            self.ended = True
            self.changed = True
        elif (self.running and self.lasttime != now):
            self.image = Image.new("RGBA",(self.width,self.height))
            draw = ImageDraw.Draw(self.image)
            if self.dynamiccolor:
                if seconds_left < 60:
                    self.color = (255,0,0)
                elif seconds_left < 5*60:
                    self.color = (255,255,0)
                else:
                    self.color = (64,255,64)
            draw.text((0,0),"{:^3}".format(seconds_left),font=self.bigfont,fill=self.color)
            self.lasttime = now
            self.changed = True
        else:
            self.changed = False

    def update(self):
        now = int(time.time())
        seconds_left = self.endtime - now
        if seconds_left <= self.bigat:
            self.big()
        elif seconds_left < 0:
            self.running = False
            self.ended = True
            self.changed = True
        elif (self.running and self.lasttime != now):
            current_time = time.gmtime(seconds_left)
            time_string = time.strftime("%H:%M:%S",current_time)
            self.image = Image.new("RGBA",(self.width,self.height))
            draw = ImageDraw.Draw(self.image)
            if self.dynamiccolor:
                if seconds_left < 60:
                    self.color = (255,0,0)
                elif seconds_left < 5*60:
                    self.color = (255,255,0)
                else:
                    self.color = (64,255,64)
            draw.text((self.x,self.y),time_string,font=self.font,fill=self.color)
            self.changed = True
            self.lasttime = now
        else:
            self.changed = False



if __name__ == "__main__":
    options = RGBMatrixOptions()
    options.rows = 64
    options.cols = 64

    matrix = RGBMatrix(options = options)

    currenttime = CountdownWidget()
    currenttime.mqttstart("countdown",b"16:00")

    while True:
        currenttime.update()
        if (currenttime.changed):
            im = Image.new("RGBA",(64,64))
            im.alpha_composite(currenttime.image)
            matrix.SetImage(im.convert("RGB"))
        time.sleep(0.5)