showheartbeat.py

import curses
from widgets import Table, rectangle

class heartbeatwin:
    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.table = Table(
            self.win,
            1,
            2,
            self.width-3,
            self.height-3,
            [
                {
                    "width": self.width-3,
                    "padding_left": 1,
                    "text": lambda col, row, item, data: item[0],
                    "attributes": lambda col, row, item, data: curses.color_pair(2) if item[1] else curses.color_pair(1)
                }
            ])
        
        self.win.addstr(0, 0, 'Infrastructure:')
        rectangle(self.win,0,1,self.width,self.height-1)
        self.heartbeats = {}

    def update(self, topic, payload):
        if len(payload) == 0:
            self.heartbeats.pop(topic[len('heartbeat/'):])

        else:
            self.heartbeats[topic[len('heartbeat/'):]] = payload[0]
        
        self.table.apply_data( list(sorted(self.heartbeats.items())) )

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