showbikes.py

#!/usr/bin/env python3

import curses
from datetime import datetime
from curses import wrapper
from widgets import Table, rectangle, Label

class bikeswin:
    def __init__(self, xpos, ypos, width, height):
        self.win = curses.newwin(height, width, ypos, xpos)
        self.height = height
        self.width = width
        self.xpos = xpos
        self.ypos = ypos
        
        self.label = Label(self.win, 0, 0, self.width, "Bikes not connected").draw()
        self.table = Table(self.win,1,2,width-2,height-3,[{
                "text": lambda col,row,bike,data: ("#%s" % bike["bike_number"]) if "bike_number" in bike else "",
                "attributes":[curses.color_pair(0),curses.color_pair(3)],
                "width": 8
            },{
                "text": lambda col,row,bike,data: str(bike["address"] or "") if "address" in bike else "",
                "attributes":[curses.color_pair(0),curses.color_pair(3)],
                "width":width-2-8-5
            },{
                "text": lambda col,row,bike,data: ("%dm" % bike["distance"]) if "distance" in bike else "",
                "attributes":[curses.color_pair(0),curses.color_pair(3)],
                "width":5
            }],
            {
                "line_delay": 0.025
            })
        rectangle(self.win,0,1,self.width,self.height-1)

    def update(self, data):
        curses.init_pair(3, curses.COLOR_WHITE, curses.COLOR_BLUE)
        
        try:
            self.label.update_text("Bikes %s" % data['time']).draw()
            
            bikes = data["bikes"]
            bikes.sort(key=lambda bike: bike["distance"])
            self.table.apply_data(bikes)
        except Exception as msg:
            self.win.addstr(2,2,"Something went wrong! " + str(msg), curses.color_pair(1))

    def show(self):
        self.win.refresh()