showtraffic.py

#!/usr/bin/env python3

import curses
from widgets import Table, rectangle, Label

class trafficwin:
    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, "Traffic not connected").draw()
        self.table = Table(
                self.win,
                1,
                2,
                width-2,
                height-3,
                [
                    { # LINE
                        "width": 10,
                        "text": lambda col,row,dep,data: "   "+dep["line"],
                        "attributes": [curses.color_pair(0), curses.color_pair(3)]
                    },
                    { # DIRECTION
                        "width": width-8-9-6-5,
                        "text": lambda col,row,dep,data: dep["direction"],
                        "attributes": [curses.color_pair(0), curses.color_pair(3)]
                    },
                    { # PLATFORM
                        "width": 6,
                        "text": lambda col,row,dep,data: dep["platform"],
                        "attributes": [curses.color_pair(0), curses.color_pair(3)]
                    },
                    { # TIME
                        "width": 6,
                        "text": lambda col,row,dep,data: dep["departure"] if "departure" in dep else "",
                        "attributes": [curses.color_pair(0),curses.color_pair(3)]
                    },
                    { # DELAY
                        "width": 4,
                        "text": lambda col,row,dep,data: dep["delay"],
                        "attributes": [curses.color_pair(0), curses.color_pair(3)] 
                    },
                ],
                {
                    "line_delay": 0.025
                }
            )
        rectangle(self.win,0,1,self.width,self.height-1)

    def update(self, dep):
        curses.init_pair(3, curses.COLOR_WHITE, curses.COLOR_BLUE)
        curses.init_pair(4, curses.COLOR_RED, curses.COLOR_BLUE)
        curses.init_pair(5, curses.COLOR_GREEN, curses.COLOR_BLUE)

        try:
            #delays = sum(map(lambda x: x["delay"] if "delay" in x and x["delay"] > 0 else 0, dep["departures"]))
            #" | Total Delay: %d Min." % (delays) if delays > 0 else ""
            delays = 0
            
            self.label.update_text("Departures %s" % dep['srvtime']).draw()
            self.table.apply_data(dep["departures"])
                
        except Exception as msg:
            self.win.addstr(2,2,"Something went wrong! " + str(msg), curses.color_pair(1))

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