mirror of
https://github.com/rr-/pq-cli.git
synced 2026-09-29 06:44:00 -06:00
52 lines
1.5 KiB
Python
52 lines
1.5 KiB
Python
import curses
|
|
import enum
|
|
import typing as T
|
|
|
|
from pqcli.ui.curses.colors import COLOR_HIGHLIGHT, has_colors
|
|
|
|
from .scrollable import Scrollable
|
|
|
|
|
|
class ListBox(Scrollable):
|
|
_items: T.List[str]
|
|
|
|
def __init__(self, parent: T.Any, h: int, w: int, y: int, x: int) -> None:
|
|
super().__init__(parent, h, w, y, x)
|
|
self._selected: T.Optional[int] = None
|
|
|
|
def add(self, text: str) -> None:
|
|
self._items.append(text)
|
|
self.scroll_to_item(-1)
|
|
|
|
def set(self, idx: int, value: str) -> None:
|
|
if idx < 0:
|
|
idx %= len(self._items)
|
|
self._items[idx] = value
|
|
|
|
def get(self, idx: int) -> T.Optional[str]:
|
|
if not self._items:
|
|
return None
|
|
if idx < 0:
|
|
idx %= len(self._items)
|
|
try:
|
|
return self._items[idx]
|
|
except ValueError:
|
|
return None
|
|
|
|
def delete(self, idx: int, count: int = 1) -> None:
|
|
del self._items[idx : idx + count]
|
|
|
|
def select(self, idx: T.Optional[int]) -> None:
|
|
if idx < 0 and len(self._items):
|
|
idx %= len(self._items)
|
|
self._selected = idx
|
|
|
|
def _render_impl(self, h: int, w: int) -> None:
|
|
assert self._pad
|
|
for y, item in enumerate(self._items):
|
|
if y == self._selected and has_colors():
|
|
self._pad.attron(curses.color_pair(COLOR_HIGHLIGHT))
|
|
self._pad.addnstr(y, 0, item, min(len(item), w))
|
|
if y == self._selected and has_colors():
|
|
self._pad.attroff(curses.color_pair(COLOR_HIGHLIGHT))
|