mirror of
https://github.com/rr-/pq-cli.git
synced 2026-09-29 06:44:00 -06:00
57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
import functools
|
|
import typing as T
|
|
|
|
from pqcli.roster import Roster
|
|
|
|
from ..event_handler import EventHandler
|
|
from ..util import Choice
|
|
from ..widgets import Menu
|
|
from .base_view import BaseView
|
|
|
|
|
|
class ChooseCharacterView(BaseView):
|
|
def __init__(self, screen: T.Any, roster: Roster, title: str) -> None:
|
|
super().__init__(screen)
|
|
|
|
self.title = f"{title}:"
|
|
self.on_cancel = EventHandler()
|
|
self.on_confirm = EventHandler()
|
|
|
|
self.choices: T.List[Choice] = []
|
|
for y, player in enumerate(roster.players, 1):
|
|
key: T.Optional[str] = str(y % 10) if y <= 10 else None
|
|
self.choices.append(
|
|
Choice(
|
|
keys=[ord(key)] if key is not None else [],
|
|
desc=f"[{key or '-'}] {player.name}",
|
|
callback=functools.partial(self.on_confirm, player),
|
|
)
|
|
)
|
|
self.choices.append(
|
|
Choice(
|
|
keys=list(map(ord, "qQ\N{ESC}")),
|
|
desc="[Q] Cancel",
|
|
callback=self.on_cancel,
|
|
)
|
|
)
|
|
|
|
def start(self) -> None:
|
|
scr_height, scr_width = self.screen.getmaxyx()
|
|
self.screen.erase()
|
|
self.screen.refresh()
|
|
|
|
self.main_menu = Menu(
|
|
header=self.title,
|
|
choices=self.choices,
|
|
scr_height=scr_height,
|
|
scr_width=scr_width,
|
|
)
|
|
self.main_menu.render()
|
|
|
|
def stop(self) -> None:
|
|
self.main_menu.stop()
|
|
|
|
def keypress(self, key: int) -> None:
|
|
self.main_menu.keypress(key)
|
|
self.main_menu.render()
|