ui/curses: highlight changed and added items

This commit is contained in:
rr-
2019-01-14 16:10:26 +01:00
parent 5f34a85c80
commit 555f386b24
9 changed files with 50 additions and 3 deletions
+14
View File
@@ -2,6 +2,8 @@ import curses
import enum
import typing as T
from pqcli.ui.curses.colors import COLOR_HIGHLIGHT, has_colors
from .scrollable import Scrollable
@@ -19,6 +21,7 @@ class DataTable(Scrollable):
self._align_right = align_right
self._items: T.List[T.Tuple[str, str]]
self._selected: T.Optional[int] = None
def add(self, text: str, value: str) -> None:
self._items.append((text, value))
@@ -38,6 +41,9 @@ class DataTable(Scrollable):
del self._items[idx]
self.scroll_to_item(idx)
def select(self, text: T.Optional[str]) -> None:
self._selected = None if text is None else self.get_idx(text)
def get_idx(self, text: str) -> T.Optional[int]:
for idx, row in enumerate(self._items):
if row[0].strip() == text.strip():
@@ -48,20 +54,28 @@ class DataTable(Scrollable):
assert self._pad
if self._align_right:
for y, row in enumerate(self._items):
if y == self._selected and has_colors():
self._pad.attron(curses.color_pair(COLOR_HIGHLIGHT))
self._pad.move(y, 0)
self._pad.addnstr(row[0], min(len(row[0]), w))
col2_x = max(0, w - len(row[1]))
if col2_x < w:
self._pad.move(y, col2_x)
self._pad.addstr(row[1])
if y == self._selected and has_colors():
self._pad.attroff(curses.color_pair(COLOR_HIGHLIGHT))
else:
col2_x = self._get_col_width(0)
for y, row in enumerate(self._items):
if y == self._selected and has_colors():
self._pad.attron(curses.color_pair(COLOR_HIGHLIGHT))
self._pad.addnstr(y, 0, row[0], min(len(row[0]), w))
if col2_x < w:
self._pad.addnstr(
y, col2_x, row[1], min(len(row[1]), w - col2_x)
)
if y == self._selected and has_colors():
self._pad.attroff(curses.color_pair(COLOR_HIGHLIGHT))
def _get_col_width(self, x: int) -> int:
return max([len(row[0]) for row in self._items] + [0])