mirror of
https://github.com/rr-/pq-cli.git
synced 2026-11-12 22:40:36 -07:00
Dash
0a4358e2b8
Python 3.10 is the minimum, so annotations use `X | None`, `list[X]` and friends, enforced by ruff's pyupgrade rules. The remaining typing names are imported directly, with abstract types coming from collections.abc. first() is now generic over the item and default.
83 lines
2.9 KiB
Python
83 lines
2.9 KiB
Python
import curses
|
|
from typing import Any
|
|
|
|
from pqcli.ui.curses.colors import COLOR_HIGHLIGHT, has_colors
|
|
|
|
from .scrollable import Scrollable
|
|
|
|
|
|
class DataTable(Scrollable):
|
|
def __init__(
|
|
self,
|
|
parent: Any,
|
|
h: int,
|
|
w: int,
|
|
y: int,
|
|
x: int,
|
|
align_right: bool = False,
|
|
) -> None:
|
|
super().__init__(parent, h, w, y, x)
|
|
|
|
self._align_right = align_right
|
|
self._items: list[tuple[str, str]]
|
|
self._selected: int | None = None
|
|
|
|
def add(self, text: str, value: str) -> None:
|
|
self._items.append((text, value))
|
|
self.scroll_to_item(-1)
|
|
|
|
def set(self, text: str, value: str) -> None:
|
|
idx = self.get_idx(text)
|
|
if idx is None:
|
|
self.add(text, value)
|
|
else:
|
|
self._items[idx] = (self._items[idx][0], value)
|
|
self.scroll_to_item(idx)
|
|
|
|
def delete(self, text: str) -> None:
|
|
idx = self.get_idx(text)
|
|
if idx is not None:
|
|
del self._items[idx]
|
|
self.scroll_to_item(idx)
|
|
|
|
def select(self, text: str | None) -> None:
|
|
self._selected = None if text is None else self.get_idx(text)
|
|
|
|
def get_idx(self, text: str) -> int | None:
|
|
for idx, row in enumerate(self._items):
|
|
if row[0].strip() == text.strip():
|
|
return idx
|
|
return None
|
|
|
|
def _render_impl(self, h: int, w: int) -> None:
|
|
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))
|
|
# Determine the number of characters to write
|
|
nwrite = min(len(row[1]), w - col2_x)
|
|
# Ensure we have enough columns, and we are writing more
|
|
# than zero
|
|
if col2_x < w and nwrite > 0:
|
|
self._pad.addnstr(y, col2_x, row[1], nwrite)
|
|
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])
|