picotui_labels.py

class WLabel(WLabel):
    def set_text(self, text):
        self.t = text
        self.w = len(text)


class WColorLabel(WLabel):
    def __init__(self, text, fg=None, bg=-1):
        self.c = (fg, bg) if fg else None
        super().__init__(text)

    def set_text(self, text, fg=None, bg=-1):
        self.t = text
        self.w = len(text)
        self.c = (fg, bg) if fg is not None else None

    def redraw(self):
        self.goto(self.x, self.y)
        if self.c:
            self.attr_color(*self.c)
        self.wr(self.t)
        if self.c:
            self.attr_reset()