mirror of
https://github.com/rr-/pq-cli.git
synced 2026-09-29 14:54:00 -06:00
Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
746ca551f3 | ||
|
|
5d47a0f44c | ||
|
|
555f386b24 | ||
|
|
5f34a85c80 | ||
|
|
97552a711a | ||
|
|
561f28bd95 | ||
|
|
0bd4583c73 | ||
|
|
adc3c7914d | ||
|
|
d2c06b08bb | ||
|
|
4dc674816b | ||
|
|
8c6b1a2504 | ||
|
|
8d385523ad | ||
|
|
05c9bcd0e6 | ||
|
|
2c3da6484b | ||
|
|
d4f9326759 | ||
|
|
e85a8d5826 | ||
|
|
2da46a44a4 | ||
|
|
0ecfc2c1fd | ||
|
|
ffe5734433 | ||
|
|
ae99f1f1a4 | ||
|
|
a47d6233be | ||
|
|
1980049c6f | ||
|
|
9fc078ad94 | ||
|
|
7a835c615e | ||
|
|
9412a03da7 | ||
|
|
9012363437 | ||
|
|
021d7eb602 | ||
|
|
3d8034b285 | ||
|
|
bd3f196f84 | ||
|
|
9a90b7b7b4 | ||
|
|
e3ffe51979 | ||
|
|
0133802570 | ||
|
|
b20c4b4b34 | ||
|
|
5bdfc2ed25 | ||
|
|
8b38eeecac | ||
|
|
ed9e513271 | ||
|
|
97d958d2e7 | ||
|
|
9db61f7191 | ||
|
|
4230c57e73 | ||
|
|
871144bf0c | ||
|
|
3fdc2459ca | ||
|
|
e008fc6048 | ||
|
|
ca7ab192c5 | ||
|
|
81f6dbede2 | ||
|
|
478075347c | ||
|
|
595538cdca | ||
|
|
e27f3470a4 |
@@ -10,16 +10,23 @@ Relive the great adventure… this time in the terminal realm!
|
|||||||
|
|
||||||
## Features
|
## Features
|
||||||
|
|
||||||
- Saves (with backups) to `$XDG_CONFIG_HOME/pqcli/save.dat`
|
|
||||||
- Clean terminal interface
|
|
||||||
- Smooth progress bars using UTF-8 block characters
|
|
||||||
- Scrollable list boxes
|
|
||||||
- Faithful port of the game logic
|
- Faithful port of the game logic
|
||||||
|
- Saves (with backups) to `$XDG_CONFIG_HOME/pqcli/save.dat`
|
||||||
|
- Terminal interface that comes in 2 flavors:
|
||||||
|
- Rich and colorful (`--curses`, default)
|
||||||
|
- Minimal, suitable for raw grind (`--basic`)
|
||||||
- Ideal to run on your server
|
- Ideal to run on your server
|
||||||
|
|
||||||
## How it looks like
|
## How it looks like
|
||||||
|
|
||||||

|
Curses interface:
|
||||||
|
|
||||||
|

|
||||||
|

|
||||||
|
|
||||||
|
Basic interface:
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
## How to install
|
## How to install
|
||||||
|
|
||||||
|
|||||||
+28
-10
@@ -4,22 +4,39 @@ from pathlib import Path
|
|||||||
import xdg
|
import xdg
|
||||||
|
|
||||||
from pqcli.roster import Roster
|
from pqcli.roster import Roster
|
||||||
from pqcli.ui import Ui
|
from pqcli.ui.basic import BasicUserInterface
|
||||||
|
from pqcli.ui.curses import CursesUserInterface
|
||||||
|
|
||||||
SAVE_PATH = Path(xdg.XDG_CONFIG_HOME) / "pqcli" / "save.dat"
|
SAVE_PATH = Path(xdg.XDG_CONFIG_HOME) / "pqcli" / "save.dat"
|
||||||
|
|
||||||
|
|
||||||
def parse_args() -> argparse.Namespace:
|
def parse_args() -> argparse.Namespace:
|
||||||
parser = argparse.ArgumentParser(prog="pqcli")
|
parser = argparse.ArgumentParser(prog="pqcli")
|
||||||
parser.add_argument(
|
|
||||||
"--no-config",
|
group = parser.add_mutually_exclusive_group()
|
||||||
dest="use_config",
|
group.set_defaults(ui=CursesUserInterface)
|
||||||
action="store_false",
|
group.add_argument(
|
||||||
help=(
|
"--basic",
|
||||||
"Don't load or save configuration files from $XDG_CONFIG_HOME "
|
dest="ui",
|
||||||
"(player data will still be saved)"
|
action="store_const",
|
||||||
),
|
const=BasicUserInterface,
|
||||||
|
help="Use basic user interface (very crude, but uses least CPU)",
|
||||||
)
|
)
|
||||||
|
group.add_argument(
|
||||||
|
"--curses",
|
||||||
|
dest="ui",
|
||||||
|
action="store_const",
|
||||||
|
const=CursesUserInterface,
|
||||||
|
help="Use curses user interface (fast, but no colors output)",
|
||||||
|
)
|
||||||
|
|
||||||
|
parser.add_argument(
|
||||||
|
"--no-colors",
|
||||||
|
dest="colors",
|
||||||
|
action="store_false",
|
||||||
|
help="Disable color highlighting in curses interface",
|
||||||
|
)
|
||||||
|
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--no-save",
|
"--no-save",
|
||||||
dest="use_saves",
|
dest="use_saves",
|
||||||
@@ -35,7 +52,8 @@ def main() -> None:
|
|||||||
roster = Roster.load(SAVE_PATH)
|
roster = Roster.load(SAVE_PATH)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
Ui(roster, args).run()
|
ui = args.ui(roster, args)
|
||||||
|
ui.run()
|
||||||
finally:
|
finally:
|
||||||
if args.use_saves:
|
if args.use_saves:
|
||||||
roster.save()
|
roster.save()
|
||||||
|
|||||||
@@ -1,8 +1,30 @@
|
|||||||
|
import datetime
|
||||||
import typing as T
|
import typing as T
|
||||||
|
|
||||||
from pqcli import random
|
from pqcli import random
|
||||||
|
|
||||||
|
|
||||||
|
def format_float(num: float) -> str:
|
||||||
|
ret = f"{num:.01f}"
|
||||||
|
if ret.endswith("0"):
|
||||||
|
ret = ret[:-2]
|
||||||
|
return ret
|
||||||
|
|
||||||
|
|
||||||
|
def format_timespan(timespan: datetime.timedelta) -> str:
|
||||||
|
num = timespan.total_seconds()
|
||||||
|
if num < 60.0:
|
||||||
|
return f"~{int(num)}s"
|
||||||
|
num /= 60
|
||||||
|
if num < 60.0:
|
||||||
|
return f"~{int(num)}m"
|
||||||
|
num /= 60
|
||||||
|
if num < 24.0:
|
||||||
|
return f"~{format_float(num)}h"
|
||||||
|
num /= 24
|
||||||
|
return f"~{format_float(num)}d"
|
||||||
|
|
||||||
|
|
||||||
def generate_name() -> str:
|
def generate_name() -> str:
|
||||||
parts = [
|
parts = [
|
||||||
"br|cr|dr|fr|gr|j|kr|l|m|n|pr||||r|sh|tr|v|wh|x|y|z".split("|"),
|
"br|cr|dr|fr|gr|j|kr|l|m|n|pr||||r|sh|tr|v|wh|x|y|z".split("|"),
|
||||||
@@ -122,3 +144,8 @@ def special(m: int, subject: str) -> str:
|
|||||||
return prefix(
|
return prefix(
|
||||||
["Battle-", "cursed ", "Were-", "undead ", "demon "], m, subject, ""
|
["Battle-", "cursed ", "Were-", "undead ", "demon "], m, subject, ""
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def terminate_message(player_name: str) -> str:
|
||||||
|
adjective = random.choice(["faithful", "noble", "loyal", "brave"])
|
||||||
|
return f"Terminate {adjective} {player_name}?"
|
||||||
|
|||||||
+37
-18
@@ -1,11 +1,11 @@
|
|||||||
import datetime
|
import datetime
|
||||||
import itertools
|
import itertools
|
||||||
import logging
|
import logging
|
||||||
|
import math
|
||||||
import typing as T
|
import typing as T
|
||||||
|
from collections import defaultdict
|
||||||
from dataclasses import dataclass
|
from dataclasses import dataclass
|
||||||
|
|
||||||
import urwid
|
|
||||||
|
|
||||||
from pqcli import random
|
from pqcli import random
|
||||||
from pqcli.config import *
|
from pqcli.config import *
|
||||||
from pqcli.lingo import *
|
from pqcli.lingo import *
|
||||||
@@ -13,23 +13,29 @@ from pqcli.lingo import *
|
|||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
_SIGNALS: T.Dict[
|
||||||
|
T.Tuple["SignalMixin", str], T.List[T.Callable[..., T.Any]]
|
||||||
|
] = defaultdict(list)
|
||||||
|
|
||||||
|
|
||||||
class SignalMixin:
|
class SignalMixin:
|
||||||
def __init_subclass__(cls: T.Any, **kwargs: T.Any) -> None:
|
|
||||||
urwid.signals.register_signal(cls, cls.signals)
|
|
||||||
super().__init_subclass__(**kwargs)
|
|
||||||
|
|
||||||
def emit(self, signal_name: str, *user_data: T.Any) -> None:
|
def emit(self, signal_name: str, *user_data: T.Any) -> None:
|
||||||
urwid.signals.emit_signal(self, signal_name, *user_data)
|
for callback in _SIGNALS[self, signal_name]:
|
||||||
|
callback(*user_data)
|
||||||
|
|
||||||
def connect(self, signal_name: str, callback: T.Callable) -> None:
|
def connect(
|
||||||
urwid.signals.connect_signal(self, signal_name, callback)
|
self, signal_name: str, callback: T.Callable[..., T.Any]
|
||||||
|
) -> None:
|
||||||
|
_SIGNALS[self, signal_name].append(callback)
|
||||||
|
|
||||||
def __getstate__(self) -> T.Any:
|
def disconnect(
|
||||||
return {
|
self, signal_name: str, callback: T.Callable[..., T.Any]
|
||||||
key: value
|
) -> None:
|
||||||
for key, value in self.__dict__.items()
|
try:
|
||||||
if "signal" not in key
|
idx = _SIGNALS[self, signal_name].index(callback)
|
||||||
}
|
except ValueError:
|
||||||
|
return
|
||||||
|
del _SIGNALS[self, signal_name][idx]
|
||||||
|
|
||||||
|
|
||||||
def level_up_time(level: int) -> int:
|
def level_up_time(level: int) -> int:
|
||||||
@@ -46,6 +52,9 @@ class Bar(SignalMixin):
|
|||||||
self.emit("change")
|
self.emit("change")
|
||||||
|
|
||||||
def reset(self, new_max: int, position: float = 0.0) -> None:
|
def reset(self, new_max: int, position: float = 0.0) -> None:
|
||||||
|
if not math.isclose(new_max, self._max) or not math.isclose(
|
||||||
|
position, self._position
|
||||||
|
):
|
||||||
self._max = new_max
|
self._max = new_max
|
||||||
self._position = position
|
self._position = position
|
||||||
self.emit("change")
|
self.emit("change")
|
||||||
@@ -66,7 +75,10 @@ class Bar(SignalMixin):
|
|||||||
return self._position >= self.max_
|
return self._position >= self.max_
|
||||||
|
|
||||||
def reposition(self, new_pos: float) -> None:
|
def reposition(self, new_pos: float) -> None:
|
||||||
self._position = float(min(new_pos, self._max))
|
old_pos = self._position
|
||||||
|
new_pos = float(min(new_pos, self._max))
|
||||||
|
self._position = new_pos
|
||||||
|
if not math.isclose(new_pos, old_pos):
|
||||||
self.emit("change")
|
self.emit("change")
|
||||||
|
|
||||||
|
|
||||||
@@ -170,7 +182,7 @@ class Inventory(SignalMixin):
|
|||||||
|
|
||||||
def pop(self, index: int) -> None:
|
def pop(self, index: int) -> None:
|
||||||
item = self._items[index]
|
item = self._items[index]
|
||||||
logger.info("Lost %s (qty=%d)", item.name, item.quantity)
|
logger.info("Lost %s", indefinite(item.name, item.quantity))
|
||||||
self._items.pop(index)
|
self._items.pop(index)
|
||||||
self.sync_encumbrance()
|
self.sync_encumbrance()
|
||||||
self.emit("item_del", item)
|
self.emit("item_del", item)
|
||||||
@@ -237,6 +249,12 @@ class SpellBook(SignalMixin):
|
|||||||
def __iter__(self) -> T.Iterator[Spell]:
|
def __iter__(self) -> T.Iterator[Spell]:
|
||||||
return iter(self._spells)
|
return iter(self._spells)
|
||||||
|
|
||||||
|
def __bool__(self) -> bool:
|
||||||
|
return bool(self._spells)
|
||||||
|
|
||||||
|
def __len__(self) -> int:
|
||||||
|
return len(self._spells)
|
||||||
|
|
||||||
def add(self, spell_name: str, level: int) -> None:
|
def add(self, spell_name: str, level: int) -> None:
|
||||||
for spell in self._spells:
|
for spell in self._spells:
|
||||||
if spell.name == spell_name:
|
if spell.name == spell_name:
|
||||||
@@ -322,7 +340,7 @@ class Player(SignalMixin):
|
|||||||
self.task: T.Optional[BaseTask] = None
|
self.task: T.Optional[BaseTask] = None
|
||||||
self.queue: T.List[BaseTask] = []
|
self.queue: T.List[BaseTask] = []
|
||||||
|
|
||||||
def __setstate__(self, obj) -> None:
|
def __setstate__(self, obj: T.Any) -> None:
|
||||||
self.__dict__.update(obj)
|
self.__dict__.update(obj)
|
||||||
self.elapsed = obj.get("elapsed", 0.0)
|
self.elapsed = obj.get("elapsed", 0.0)
|
||||||
|
|
||||||
@@ -899,6 +917,7 @@ class StatsBuilder:
|
|||||||
return stats
|
return stats
|
||||||
|
|
||||||
def unroll(self) -> Stats:
|
def unroll(self) -> Stats:
|
||||||
|
if len(self.history) > 1:
|
||||||
self.history.pop()
|
self.history.pop()
|
||||||
return self.history[-1]
|
return self.history[-1]
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,7 @@
|
|||||||
|
import logging
|
||||||
import pickle
|
import pickle
|
||||||
import typing as T
|
import typing as T
|
||||||
|
from datetime import datetime
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
from pqcli.mechanic import Player
|
from pqcli.mechanic import Player
|
||||||
@@ -9,6 +11,7 @@ class Roster:
|
|||||||
def __init__(self, path: Path, players: T.List[Player]) -> None:
|
def __init__(self, path: Path, players: T.List[Player]) -> None:
|
||||||
self.path = path
|
self.path = path
|
||||||
self.players = players
|
self.players = players
|
||||||
|
self._last_save = datetime.now()
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def load(path: T.Union[str, Path]) -> "Roster":
|
def load(path: T.Union[str, Path]) -> "Roster":
|
||||||
@@ -20,6 +23,7 @@ class Roster:
|
|||||||
return Roster(real_path, players=[])
|
return Roster(real_path, players=[])
|
||||||
|
|
||||||
def save(self) -> None:
|
def save(self) -> None:
|
||||||
|
self._last_save = datetime.now()
|
||||||
old_path = self.path.with_name(self.path.name + ".old")
|
old_path = self.path.with_name(self.path.name + ".old")
|
||||||
tmp_path = self.path.with_name(self.path.name + ".new")
|
tmp_path = self.path.with_name(self.path.name + ".new")
|
||||||
tmp_path.parent.mkdir(parents=True, exist_ok=True)
|
tmp_path.parent.mkdir(parents=True, exist_ok=True)
|
||||||
@@ -27,3 +31,8 @@ class Roster:
|
|||||||
if self.path.exists():
|
if self.path.exists():
|
||||||
self.path.rename(old_path)
|
self.path.rename(old_path)
|
||||||
tmp_path.rename(self.path)
|
tmp_path.rename(self.path)
|
||||||
|
|
||||||
|
def save_periodically(self) -> None:
|
||||||
|
if (datetime.now() - self._last_save).total_seconds() >= 300:
|
||||||
|
logging.info("Saving...")
|
||||||
|
self.save()
|
||||||
|
|||||||
-115
@@ -1,115 +0,0 @@
|
|||||||
import argparse
|
|
||||||
import datetime
|
|
||||||
import logging
|
|
||||||
import sys
|
|
||||||
from pathlib import Path
|
|
||||||
from signal import SIG_DFL, SIGPIPE, signal
|
|
||||||
|
|
||||||
from pqcli import random
|
|
||||||
from pqcli.config import CLASSES, RACES
|
|
||||||
from pqcli.mechanic import (
|
|
||||||
Player,
|
|
||||||
Simulation,
|
|
||||||
StatsBuilder,
|
|
||||||
act_name,
|
|
||||||
create_player,
|
|
||||||
generate_name,
|
|
||||||
)
|
|
||||||
from pqcli.roster import Roster
|
|
||||||
|
|
||||||
signal(SIGPIPE, SIG_DFL)
|
|
||||||
logging.basicConfig(
|
|
||||||
stream=sys.stdout, level=logging.DEBUG, format="%(message)s"
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
# make things predictable
|
|
||||||
SEED_A = 1_664_525
|
|
||||||
SEED_B = 2141
|
|
||||||
SEED_C = 1_013_904_223
|
|
||||||
|
|
||||||
|
|
||||||
def custom_random(num: int) -> int:
|
|
||||||
global SEED_A, SEED_B, SEED_C
|
|
||||||
SEED_B = (SEED_A * SEED_B + SEED_C) % 0x100000000
|
|
||||||
return SEED_B % num
|
|
||||||
|
|
||||||
|
|
||||||
random.below = custom_random
|
|
||||||
|
|
||||||
|
|
||||||
def character_sheet(player: Player) -> None:
|
|
||||||
print(f"{player.name} the {player.race.name}")
|
|
||||||
print(f"Level {player.level} {player.class_.name}")
|
|
||||||
print(f"Plot stage: {act_name(player.quest_book.act)}")
|
|
||||||
print(f"Quest: {player.quest_book.current_quest}")
|
|
||||||
print()
|
|
||||||
print("Stats:")
|
|
||||||
for stat, value in player.stats:
|
|
||||||
print(f" {stat.value}: {value}")
|
|
||||||
print()
|
|
||||||
print(f"Equipment (best: {player.equipment.best}):")
|
|
||||||
for equipment, item_name in player.equipment:
|
|
||||||
print(f" {equipment.value}: {item_name}")
|
|
||||||
print()
|
|
||||||
print("Spell Book:")
|
|
||||||
for spell in player.spell_book:
|
|
||||||
print(f" {spell.name}: {spell.level}")
|
|
||||||
print()
|
|
||||||
print("Inventory:")
|
|
||||||
print(f" Gold: {player.inventory.gold}")
|
|
||||||
for item in player.inventory:
|
|
||||||
print(f" {item.name}: {item.quantity}")
|
|
||||||
|
|
||||||
|
|
||||||
def parse_args() -> argparse.Namespace:
|
|
||||||
parser = argparse.ArgumentParser()
|
|
||||||
parser.add_argument("-r", "--roster", type=Path)
|
|
||||||
parser.add_argument("-l", "--max-level", type=int, default=5)
|
|
||||||
return parser.parse_args()
|
|
||||||
|
|
||||||
|
|
||||||
def random_player() -> Player:
|
|
||||||
return create_player(
|
|
||||||
stats=StatsBuilder().roll(),
|
|
||||||
name=generate_name(),
|
|
||||||
race=random.choice(RACES),
|
|
||||||
class_=random.choice(CLASSES),
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def main() -> None:
|
|
||||||
args = parse_args()
|
|
||||||
|
|
||||||
if args.roster:
|
|
||||||
roster = Roster.load(args.roster)
|
|
||||||
if not roster.players:
|
|
||||||
roster.players.append(random_player())
|
|
||||||
roster.save()
|
|
||||||
player = roster.players[0]
|
|
||||||
else:
|
|
||||||
roster = None
|
|
||||||
player = random_player()
|
|
||||||
|
|
||||||
start = datetime.datetime.now()
|
|
||||||
elapsed = 0
|
|
||||||
last_level = 0
|
|
||||||
simulation = Simulation(player)
|
|
||||||
try:
|
|
||||||
while True:
|
|
||||||
if player.level != last_level:
|
|
||||||
last_level = player.level
|
|
||||||
character_sheet(player)
|
|
||||||
if player.level >= args.max_level:
|
|
||||||
break
|
|
||||||
simulation.tick()
|
|
||||||
elapsed += 100
|
|
||||||
finally:
|
|
||||||
print("real time: ", (datetime.datetime.now() - start))
|
|
||||||
print("virtual time: ", datetime.timedelta(milliseconds=elapsed))
|
|
||||||
if roster:
|
|
||||||
roster.save()
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
main()
|
|
||||||
@@ -1,167 +0,0 @@
|
|||||||
import argparse
|
|
||||||
import json
|
|
||||||
import typing as T
|
|
||||||
from pathlib import Path
|
|
||||||
|
|
||||||
import urwid
|
|
||||||
import xdg
|
|
||||||
|
|
||||||
from pqcli import random
|
|
||||||
from pqcli.mechanic import Player
|
|
||||||
from pqcli.roster import Roster
|
|
||||||
from pqcli.ui.confirm_dialog import ConfirmDialog
|
|
||||||
from pqcli.ui.game_view import GameView
|
|
||||||
from pqcli.ui.new_game_view import NewGameView
|
|
||||||
from pqcli.ui.roster_view import RosterView
|
|
||||||
|
|
||||||
|
|
||||||
TICK_FREQ = 0.1
|
|
||||||
SAVE_FREQ = 60.0
|
|
||||||
PALETTE: T.List[T.Tuple[str, ...]] = [
|
|
||||||
("button", "", ""),
|
|
||||||
("button-focus", "light red", "black"),
|
|
||||||
("linebox", "", ""),
|
|
||||||
("linebox-focus", "light red", "black"),
|
|
||||||
("linebox-content", "", ""),
|
|
||||||
("linebox-content-focus", "", ""),
|
|
||||||
("progressbar-normal", "", "", "standout"),
|
|
||||||
("progressbar-done", "black", "light red"),
|
|
||||||
("progressbar-smooth", "light red", ""),
|
|
||||||
("scrollbar", "light gray", "black"),
|
|
||||||
("scrollbar-focus", "light red", "black"),
|
|
||||||
("scrollable", "", ""),
|
|
||||||
("scrollable-focus", "", ""),
|
|
||||||
]
|
|
||||||
PALETTE_PATH = Path(xdg.XDG_CONFIG_HOME) / "pqcli" / "palette.json"
|
|
||||||
|
|
||||||
|
|
||||||
def load_palette() -> bool:
|
|
||||||
global PALETTE
|
|
||||||
if PALETTE_PATH.exists():
|
|
||||||
PALETTE = json.loads(PALETTE_PATH.read_text())
|
|
||||||
return True
|
|
||||||
return False
|
|
||||||
|
|
||||||
|
|
||||||
def save_palette() -> None:
|
|
||||||
global PALETTE
|
|
||||||
PALETTE_PATH.parent.mkdir(exist_ok=True, parents=True)
|
|
||||||
PALETTE_PATH.write_text(json.dumps(PALETTE, indent=4))
|
|
||||||
|
|
||||||
|
|
||||||
class ConfirmExitDialog(ConfirmDialog):
|
|
||||||
def __init__(self, old_view: urwid.Widget) -> None:
|
|
||||||
super().__init__("Really quit?", old_view)
|
|
||||||
|
|
||||||
|
|
||||||
def bind_commands() -> None:
|
|
||||||
for key, direction in {
|
|
||||||
"k": "up",
|
|
||||||
"j": "down",
|
|
||||||
"h": "left",
|
|
||||||
"l": "right",
|
|
||||||
}.items():
|
|
||||||
for key_variant in {key, key.upper()}:
|
|
||||||
urwid.command_map[key_variant] = f"cursor {direction}"
|
|
||||||
|
|
||||||
|
|
||||||
class Ui:
|
|
||||||
def __init__(self, roster: Roster, args: argparse.Namespace) -> None:
|
|
||||||
bind_commands()
|
|
||||||
|
|
||||||
if args.use_config:
|
|
||||||
if not load_palette():
|
|
||||||
save_palette() # create a file for the user to edit
|
|
||||||
|
|
||||||
self.roster = roster
|
|
||||||
self.args = args
|
|
||||||
self.loop = urwid.MainLoop(
|
|
||||||
None, PALETTE, unhandled_input=self.unhandled_input
|
|
||||||
)
|
|
||||||
self.old_views: T.List[urwid.Widget] = []
|
|
||||||
|
|
||||||
self.switch_to_roster_view()
|
|
||||||
|
|
||||||
def run(self) -> None:
|
|
||||||
self.loop.set_alarm_in(TICK_FREQ, self.on_tick)
|
|
||||||
self.loop.set_alarm_in(SAVE_FREQ, self.on_save)
|
|
||||||
self.loop.run()
|
|
||||||
|
|
||||||
def unhandled_input(self, key: str) -> bool:
|
|
||||||
callback = getattr(self.loop.widget, "unhandled_input", None)
|
|
||||||
if callback and callback(key):
|
|
||||||
return True
|
|
||||||
|
|
||||||
if key == "ctrl q":
|
|
||||||
self.switch_to_exit_view()
|
|
||||||
return True
|
|
||||||
|
|
||||||
return False
|
|
||||||
|
|
||||||
def switch_to_roster_view(self) -> None:
|
|
||||||
self.loop.widget = RosterView(self.roster)
|
|
||||||
self._connect("load_game", self.switch_to_game_view)
|
|
||||||
self._connect("new_game", self.switch_to_new_game_view)
|
|
||||||
self._connect("delete_game", self.switch_to_delete_player_view)
|
|
||||||
self._connect("exit", self.switch_to_exit_view)
|
|
||||||
|
|
||||||
def switch_to_exit_view(self) -> None:
|
|
||||||
if isinstance(self.loop.widget, ConfirmExitDialog):
|
|
||||||
return
|
|
||||||
self.old_views.append(self.loop.widget)
|
|
||||||
self.loop.widget = ConfirmExitDialog(self.old_views[-1])
|
|
||||||
self._connect("confirm", self.exit)
|
|
||||||
self._connect("cancel", self.cancel_dialog)
|
|
||||||
|
|
||||||
def switch_to_new_game_view(self) -> None:
|
|
||||||
self.loop.widget = NewGameView()
|
|
||||||
self._connect("confirm", self.create_player)
|
|
||||||
self._connect("cancel", self.switch_to_roster_view)
|
|
||||||
|
|
||||||
def create_player(self, player: Player) -> None:
|
|
||||||
self.roster.players.append(player)
|
|
||||||
self.switch_to_roster_view()
|
|
||||||
|
|
||||||
def delete_player(self, player_idx: int) -> None:
|
|
||||||
del self.roster.players[player_idx]
|
|
||||||
self.switch_to_roster_view()
|
|
||||||
|
|
||||||
def switch_to_game_view(self, player_idx: int) -> None:
|
|
||||||
player = self.roster.players[player_idx]
|
|
||||||
self.loop.widget = GameView(self.loop, player, self.args)
|
|
||||||
self._connect("cancel", self.switch_to_roster_view)
|
|
||||||
|
|
||||||
def switch_to_delete_player_view(self, player_idx: int) -> None:
|
|
||||||
adjective = random.choice(["faithful", "noble", "loyal", "brave"])
|
|
||||||
player_name = self.roster.players[player_idx].name
|
|
||||||
|
|
||||||
self.old_views.append(self.loop.widget)
|
|
||||||
self.loop.widget = ConfirmDialog(
|
|
||||||
f"Terminate {adjective} {player_name}?", self.old_views[-1]
|
|
||||||
)
|
|
||||||
self._connect("confirm", lambda: self.delete_player(player_idx))
|
|
||||||
self._connect("cancel", self.cancel_dialog)
|
|
||||||
|
|
||||||
def exit(self) -> None:
|
|
||||||
raise urwid.ExitMainLoop()
|
|
||||||
|
|
||||||
def cancel_dialog(self) -> None:
|
|
||||||
assert len(self.old_views)
|
|
||||||
self.loop.widget = self.old_views.pop()
|
|
||||||
|
|
||||||
def _connect(self, signal_name: str, callback: T.Callable) -> None:
|
|
||||||
urwid.signals.connect_signal(
|
|
||||||
self.loop.widget,
|
|
||||||
signal_name,
|
|
||||||
lambda _sender, *args: callback(*args),
|
|
||||||
)
|
|
||||||
|
|
||||||
def on_tick(self, _loop: urwid.MainLoop, _user_data: T.Any) -> None:
|
|
||||||
if hasattr(self.loop.widget, "tick"):
|
|
||||||
self.loop.widget.tick()
|
|
||||||
self.loop.set_alarm_in(TICK_FREQ, self.on_tick)
|
|
||||||
|
|
||||||
def on_save(self, _loop: urwid.MainLoop, _user_data: T.Any) -> None:
|
|
||||||
if self.args.use_saves:
|
|
||||||
self.roster.save()
|
|
||||||
self.loop.set_alarm_in(SAVE_FREQ, self.on_save)
|
|
||||||
|
|||||||
@@ -0,0 +1,14 @@
|
|||||||
|
import abc
|
||||||
|
import argparse
|
||||||
|
|
||||||
|
from pqcli.roster import Roster
|
||||||
|
|
||||||
|
|
||||||
|
class BaseUserInterface(abc.ABC):
|
||||||
|
def __init__(self, roster: Roster, args: argparse.Namespace) -> None:
|
||||||
|
self.roster = roster
|
||||||
|
self.args = args
|
||||||
|
|
||||||
|
@abc.abstractmethod
|
||||||
|
def run(self) -> None:
|
||||||
|
raise NotImplementedError("not implemented")
|
||||||
@@ -0,0 +1,222 @@
|
|||||||
|
import argparse
|
||||||
|
import enum
|
||||||
|
import logging
|
||||||
|
import signal
|
||||||
|
import sys
|
||||||
|
import time
|
||||||
|
import typing as T
|
||||||
|
from datetime import datetime, timedelta
|
||||||
|
|
||||||
|
from pqcli import lingo
|
||||||
|
from pqcli.config import CLASSES, PRIME_STATS, RACES
|
||||||
|
from pqcli.mechanic import Player, Simulation, StatsBuilder, create_player
|
||||||
|
from pqcli.roster import Roster
|
||||||
|
from pqcli.ui.base import BaseUserInterface
|
||||||
|
|
||||||
|
LOGO = """
|
||||||
|
____ ___ _
|
||||||
|
| _ \ _ __ ___ __ _ _ __ ___ ___ ___ / _ \ _ _ ___ ___| |_
|
||||||
|
| |_) | '__/ _ \ / _` | '__/ _ \/ __/ __| | | | | | |/ _ \/ __| __|
|
||||||
|
| __/| | | (_) | (_| | | | __/\__ \__ \ |_| | |_| | __/\__ \ |_
|
||||||
|
|_| |_| \___/ \__, |_| \___||___/___/\__\_\\\\__,_|\___||___/\__|
|
||||||
|
|___/
|
||||||
|
"""
|
||||||
|
|
||||||
|
|
||||||
|
class MainMenu(enum.IntEnum):
|
||||||
|
create = 1
|
||||||
|
play = 2
|
||||||
|
info = 3
|
||||||
|
delete = 4
|
||||||
|
quit = 5
|
||||||
|
|
||||||
|
|
||||||
|
class BasicUserInterface(BaseUserInterface):
|
||||||
|
def __init__(self, roster: Roster, args: argparse.Namespace) -> None:
|
||||||
|
super().__init__(roster, args)
|
||||||
|
|
||||||
|
def signal_handler(sig: T.Any, frame: T.Any) -> None:
|
||||||
|
logging.info("Quitting")
|
||||||
|
if self.args.use_saves:
|
||||||
|
self.roster.save()
|
||||||
|
sys.exit(0)
|
||||||
|
|
||||||
|
signal.signal(signal.SIGINT, signal_handler)
|
||||||
|
|
||||||
|
def run(self) -> None:
|
||||||
|
logging.basicConfig(
|
||||||
|
stream=sys.stdout,
|
||||||
|
level=logging.INFO,
|
||||||
|
format="[%(asctime)s] %(message)s",
|
||||||
|
)
|
||||||
|
|
||||||
|
self.logo()
|
||||||
|
self.main_menu()
|
||||||
|
|
||||||
|
def logo(self) -> None:
|
||||||
|
print(LOGO)
|
||||||
|
|
||||||
|
def main_menu(self) -> None:
|
||||||
|
while True:
|
||||||
|
choice = self.menu(
|
||||||
|
[
|
||||||
|
(MainMenu.create, "Create a new character"),
|
||||||
|
(MainMenu.play, "Play as character"),
|
||||||
|
(MainMenu.info, "Query character info"),
|
||||||
|
(MainMenu.delete, "Delete a character"),
|
||||||
|
(MainMenu.quit, "Quit"),
|
||||||
|
]
|
||||||
|
)
|
||||||
|
|
||||||
|
if choice == MainMenu.create:
|
||||||
|
player = self.create_player()
|
||||||
|
|
||||||
|
elif choice == MainMenu.play:
|
||||||
|
player = self.choose_player()
|
||||||
|
if not player:
|
||||||
|
continue
|
||||||
|
self.play(player)
|
||||||
|
|
||||||
|
elif choice == MainMenu.info:
|
||||||
|
player = self.choose_player()
|
||||||
|
if not player:
|
||||||
|
continue
|
||||||
|
self.print_player_info(player)
|
||||||
|
|
||||||
|
elif choice == MainMenu.delete:
|
||||||
|
player = self.choose_player()
|
||||||
|
if not player:
|
||||||
|
continue
|
||||||
|
self.delete_player(player)
|
||||||
|
|
||||||
|
elif choice == MainMenu.quit:
|
||||||
|
self.quit()
|
||||||
|
break
|
||||||
|
|
||||||
|
def create_player(self) -> T.Optional[Player]:
|
||||||
|
name = input("Name your new character: ")
|
||||||
|
if not name:
|
||||||
|
print("Cancelled.")
|
||||||
|
return None
|
||||||
|
|
||||||
|
race = self.menu([(race, race.name) for race in RACES])
|
||||||
|
class_ = self.menu([(class_, class_.name) for class_ in CLASSES])
|
||||||
|
|
||||||
|
stats_builder = StatsBuilder()
|
||||||
|
while True:
|
||||||
|
stats = stats_builder.roll()
|
||||||
|
for stat in PRIME_STATS:
|
||||||
|
print(f"{stat.value}: {stats[stat]}")
|
||||||
|
print(f"Total: {sum(stats[stat] for stat in PRIME_STATS)}")
|
||||||
|
if self.confirm("Is this okay?"):
|
||||||
|
break
|
||||||
|
|
||||||
|
player = create_player(
|
||||||
|
name=name, race=race, class_=class_, stats=stats
|
||||||
|
)
|
||||||
|
self.roster.players.append(player)
|
||||||
|
if self.confirm("Do you want to play as your new character?"):
|
||||||
|
self.play(player)
|
||||||
|
return player
|
||||||
|
|
||||||
|
def play(self, player: Player) -> None:
|
||||||
|
print(f"Playing as {player.name}")
|
||||||
|
simulation = Simulation(player)
|
||||||
|
last_save = datetime.now()
|
||||||
|
last_tick = datetime.now()
|
||||||
|
last_level = 0
|
||||||
|
while True:
|
||||||
|
elapsed = datetime.now() - last_tick
|
||||||
|
simulation.tick(elapsed.total_seconds() * 1000)
|
||||||
|
last_tick = datetime.now()
|
||||||
|
wait = (timedelta(milliseconds=100) - elapsed).total_seconds()
|
||||||
|
wait = max(0, wait)
|
||||||
|
time.sleep(wait)
|
||||||
|
if player.level != last_level:
|
||||||
|
last_level = player.level
|
||||||
|
self.print_player_info(player)
|
||||||
|
if self.args.use_saves:
|
||||||
|
self.roster.save_periodically()
|
||||||
|
|
||||||
|
def print_player_info(self, player: Player) -> None:
|
||||||
|
print("--- Character Sheet ---")
|
||||||
|
print(f"Name: {player.name}")
|
||||||
|
print(f"Race: {player.race.name}")
|
||||||
|
print(f"Class: {player.class_.name}")
|
||||||
|
print(f"Level: {player.level}")
|
||||||
|
print()
|
||||||
|
for stat, value in player.stats:
|
||||||
|
print(f"{stat.value}: {value}")
|
||||||
|
print()
|
||||||
|
print("--- Spell Book ---")
|
||||||
|
if not player.spell_book:
|
||||||
|
print("No spells memorized yet.")
|
||||||
|
else:
|
||||||
|
for spell in player.spell_book:
|
||||||
|
print(f"{spell.name} {lingo.to_roman(spell.level)}")
|
||||||
|
print()
|
||||||
|
print("--- Equipment ---")
|
||||||
|
for equipment_type, name in player.equipment:
|
||||||
|
print(f"{equipment_type.value}: {name}")
|
||||||
|
print()
|
||||||
|
print("--- Inventory ---")
|
||||||
|
print(f"Gold: {player.inventory.gold}")
|
||||||
|
for item in player.inventory:
|
||||||
|
print(f"{item.name}: {item.quantity}")
|
||||||
|
print()
|
||||||
|
print("--- Plot ---")
|
||||||
|
print(f"Current act: {lingo.to_roman(player.quest_book.act)}")
|
||||||
|
print(f"Current quest: {player.quest_book.current_quest or '?'}")
|
||||||
|
print(
|
||||||
|
f"Current task: {player.task.description if player.task else '?'}"
|
||||||
|
)
|
||||||
|
|
||||||
|
def delete_player(self, player: Player) -> None:
|
||||||
|
if self.confirm(lingo.terminate_message(player.name)):
|
||||||
|
del self.roster.players[self.roster.players.index(player)]
|
||||||
|
|
||||||
|
def quit(self) -> None:
|
||||||
|
if self.args.use_saves:
|
||||||
|
self.roster.save()
|
||||||
|
|
||||||
|
def choose_player(self) -> T.Optional[Player]:
|
||||||
|
if not self.roster.players:
|
||||||
|
print("No characters to choose from!")
|
||||||
|
return None
|
||||||
|
|
||||||
|
return self.menu(
|
||||||
|
[(player, player.name) for player in self.roster.players],
|
||||||
|
title="Choose your character",
|
||||||
|
)
|
||||||
|
|
||||||
|
def confirm(self, message: str) -> bool:
|
||||||
|
while True:
|
||||||
|
choice = input(f"{message} [y/n] ")
|
||||||
|
if choice.lower() in {"y", "yes", "1"}:
|
||||||
|
return True
|
||||||
|
if choice.lower() in {"n", "no", "0"}:
|
||||||
|
return False
|
||||||
|
|
||||||
|
def menu(
|
||||||
|
self,
|
||||||
|
options: T.Sequence[T.Tuple[T.Any, str]],
|
||||||
|
title: T.Optional[str] = None,
|
||||||
|
) -> T.Any:
|
||||||
|
print()
|
||||||
|
if title:
|
||||||
|
print(f"{title}:")
|
||||||
|
for i, item in enumerate(options, 1):
|
||||||
|
_, description = item
|
||||||
|
print(f"{i}) {description}")
|
||||||
|
print()
|
||||||
|
|
||||||
|
while True:
|
||||||
|
try:
|
||||||
|
num = int(input("Your choice: "))
|
||||||
|
except ValueError:
|
||||||
|
print("Not a number")
|
||||||
|
continue
|
||||||
|
if num not in range(1, len(options) + 1):
|
||||||
|
print(f"Expected a number betwen 1..{len(options)}")
|
||||||
|
continue
|
||||||
|
return options[num - 1][0]
|
||||||
@@ -1,47 +0,0 @@
|
|||||||
import typing as T
|
|
||||||
|
|
||||||
import urwid
|
|
||||||
|
|
||||||
from pqcli.ui.custom_button import CustomButton
|
|
||||||
from pqcli.ui.custom_line_box import CustomLineBox
|
|
||||||
from pqcli.ui.layout import NPile
|
|
||||||
|
|
||||||
|
|
||||||
class ConfirmDialog(urwid.Overlay):
|
|
||||||
signals = ["confirm", "cancel"]
|
|
||||||
|
|
||||||
def __init__(self, label: str, parent: urwid.Widget) -> None:
|
|
||||||
question = urwid.Text(("bold", label), "center")
|
|
||||||
yes_btn = CustomButton(
|
|
||||||
"Yes", hint="Y", on_press=lambda _user_data: self.confirm()
|
|
||||||
)
|
|
||||||
no_btn = CustomButton(
|
|
||||||
"No", hint="N", on_press=lambda _user_data: self.cancel()
|
|
||||||
)
|
|
||||||
|
|
||||||
line_box = CustomLineBox(
|
|
||||||
urwid.Filler(NPile([question, no_btn, yes_btn], outermost=True))
|
|
||||||
)
|
|
||||||
|
|
||||||
super().__init__(
|
|
||||||
line_box, parent, "center", len(label) + 6, "middle", 5
|
|
||||||
)
|
|
||||||
|
|
||||||
def keypress(self, size: T.Any, key: str) -> T.Optional[str]:
|
|
||||||
if key in {"y", "Y"}:
|
|
||||||
self.confirm()
|
|
||||||
return None
|
|
||||||
if key in {"n", "N"}:
|
|
||||||
self.cancel()
|
|
||||||
return None
|
|
||||||
return T.cast(T.Optional[str], super().keypress(size, key))
|
|
||||||
|
|
||||||
def confirm(self) -> None:
|
|
||||||
self._emit("confirm")
|
|
||||||
|
|
||||||
def cancel(self) -> None:
|
|
||||||
self._emit("cancel")
|
|
||||||
|
|
||||||
def tick(self) -> None:
|
|
||||||
if hasattr(self.bottom_w, "tick"):
|
|
||||||
self.bottom_w.tick()
|
|
||||||
@@ -0,0 +1,196 @@
|
|||||||
|
import argparse
|
||||||
|
import curses
|
||||||
|
import os
|
||||||
|
import signal
|
||||||
|
import time
|
||||||
|
import typing as T
|
||||||
|
|
||||||
|
from pqcli import lingo
|
||||||
|
from pqcli.config import Class, Race
|
||||||
|
from pqcli.mechanic import Player, Stats, create_player
|
||||||
|
from pqcli.roster import Roster
|
||||||
|
from pqcli.ui.base import BaseUserInterface
|
||||||
|
from pqcli.ui.curses.colors import (
|
||||||
|
COLOR_FOCUSED,
|
||||||
|
COLOR_HIGHLIGHT,
|
||||||
|
COLOR_LOGO,
|
||||||
|
COLOR_LOGO_ALT,
|
||||||
|
COLOR_PROGRESSBAR,
|
||||||
|
COLOR_SCROLLBAR_THUMB,
|
||||||
|
COLOR_SCROLLBAR_TRACK,
|
||||||
|
has_colors,
|
||||||
|
set_colors,
|
||||||
|
)
|
||||||
|
from pqcli.ui.curses.views import (
|
||||||
|
BaseView,
|
||||||
|
ChooseCharacterClassView,
|
||||||
|
ChooseCharacterNameView,
|
||||||
|
ChooseCharacterRaceView,
|
||||||
|
ChooseCharacterStatsView,
|
||||||
|
ChooseCharacterView,
|
||||||
|
ConfirmView,
|
||||||
|
GameView,
|
||||||
|
RosterView,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class StopMainLoop(Exception):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class CursesUserInterface(BaseUserInterface):
|
||||||
|
def __init__(self, roster: Roster, args: argparse.Namespace) -> None:
|
||||||
|
super().__init__(roster, args)
|
||||||
|
os.environ.setdefault("ESCDELAY", "25")
|
||||||
|
self._screen = curses.initscr()
|
||||||
|
self._screen.nodelay(True)
|
||||||
|
self._screen.keypad(True)
|
||||||
|
curses.noecho()
|
||||||
|
curses.curs_set(0)
|
||||||
|
|
||||||
|
set_colors(args.colors)
|
||||||
|
if has_colors():
|
||||||
|
curses.start_color()
|
||||||
|
curses.use_default_colors()
|
||||||
|
curses.init_pair(COLOR_LOGO, -1, -1)
|
||||||
|
curses.init_pair(COLOR_LOGO_ALT, -1, 9)
|
||||||
|
curses.init_pair(COLOR_FOCUSED, -1, 8)
|
||||||
|
curses.init_pair(COLOR_SCROLLBAR_THUMB, -1, 7)
|
||||||
|
curses.init_pair(COLOR_SCROLLBAR_TRACK, -1, 0)
|
||||||
|
curses.init_pair(COLOR_PROGRESSBAR, 0, 9)
|
||||||
|
curses.init_pair(COLOR_HIGHLIGHT, 3, -1)
|
||||||
|
|
||||||
|
def signal_handler(sig: T.Any, frame: T.Any) -> None:
|
||||||
|
if self.args.use_saves:
|
||||||
|
self.roster.save()
|
||||||
|
exit(0)
|
||||||
|
|
||||||
|
signal.signal(signal.SIGINT, signal_handler)
|
||||||
|
|
||||||
|
self._roster_view = RosterView(self._screen)
|
||||||
|
self._roster_view.on_create += self._switch_to_create_char_name_view
|
||||||
|
self._roster_view.on_delete += self._switch_to_delete_char_view
|
||||||
|
self._roster_view.on_play += self._switch_to_play_view
|
||||||
|
self._roster_view.on_quit += self._quit
|
||||||
|
self._view: BaseView = self._roster_view
|
||||||
|
self._view.start()
|
||||||
|
|
||||||
|
def run(self) -> None:
|
||||||
|
try:
|
||||||
|
self._switch_to_roster_view()
|
||||||
|
|
||||||
|
while True:
|
||||||
|
key = T.cast(int, self._screen.getch())
|
||||||
|
|
||||||
|
if key == curses.ERR:
|
||||||
|
time.sleep(0.1)
|
||||||
|
self._view.idle()
|
||||||
|
continue
|
||||||
|
|
||||||
|
if key == curses.KEY_RESIZE:
|
||||||
|
self._view.stop()
|
||||||
|
self._view.start()
|
||||||
|
else:
|
||||||
|
try:
|
||||||
|
self._view.keypress(key)
|
||||||
|
except StopMainLoop:
|
||||||
|
break
|
||||||
|
|
||||||
|
finally:
|
||||||
|
curses.endwin()
|
||||||
|
curses.curs_set(1)
|
||||||
|
|
||||||
|
def _switch_to_roster_view(self) -> None:
|
||||||
|
self._switch_view(self._roster_view)
|
||||||
|
|
||||||
|
def _switch_to_create_char_name_view(
|
||||||
|
self, player_name: T.Optional[str] = None
|
||||||
|
) -> None:
|
||||||
|
view = ChooseCharacterNameView(self._screen, player_name)
|
||||||
|
view.on_cancel += self._switch_to_roster_view
|
||||||
|
view.on_confirm += self._switch_to_create_char_race_view
|
||||||
|
self._switch_view(view)
|
||||||
|
|
||||||
|
def _switch_to_create_char_race_view(
|
||||||
|
self, player_name: str, race: T.Optional[Race] = None
|
||||||
|
) -> None:
|
||||||
|
view = ChooseCharacterRaceView(self._screen, race)
|
||||||
|
view.on_cancel += lambda: self._switch_to_create_char_name_view(
|
||||||
|
player_name
|
||||||
|
)
|
||||||
|
view.on_confirm += lambda race: self._switch_to_create_char_class_view(
|
||||||
|
player_name, race
|
||||||
|
)
|
||||||
|
self._switch_view(view)
|
||||||
|
|
||||||
|
def _switch_to_create_char_class_view(
|
||||||
|
self, player_name: str, race: Race, class_: T.Optional[Class] = None
|
||||||
|
) -> None:
|
||||||
|
view = ChooseCharacterClassView(self._screen, class_)
|
||||||
|
view.on_cancel += lambda: self._switch_to_create_char_race_view(
|
||||||
|
player_name, race
|
||||||
|
)
|
||||||
|
view.on_confirm += lambda class_: self._switch_to_create_char_stats_view(
|
||||||
|
player_name, race, class_
|
||||||
|
)
|
||||||
|
self._switch_view(view)
|
||||||
|
|
||||||
|
def _switch_to_create_char_stats_view(
|
||||||
|
self, player_name: str, race: Race, class_: Class
|
||||||
|
) -> None:
|
||||||
|
view = ChooseCharacterStatsView(self._screen)
|
||||||
|
view.on_cancel += lambda: self._switch_to_create_char_class_view(
|
||||||
|
player_name, race, class_
|
||||||
|
)
|
||||||
|
view.on_confirm += lambda stats: self._create_character(
|
||||||
|
player_name, race, class_, stats
|
||||||
|
)
|
||||||
|
self._switch_view(view)
|
||||||
|
|
||||||
|
def _switch_to_delete_char_view(self) -> None:
|
||||||
|
view = ChooseCharacterView(
|
||||||
|
self._screen, self.roster, "Choose character to delete"
|
||||||
|
)
|
||||||
|
view.on_cancel += self._switch_to_roster_view
|
||||||
|
view.on_confirm += self._switch_to_confirm_delete_char_view
|
||||||
|
self._switch_view(view)
|
||||||
|
|
||||||
|
def _switch_to_confirm_delete_char_view(self, player: Player) -> None:
|
||||||
|
view = ConfirmView(self._screen, lingo.terminate_message(player.name))
|
||||||
|
view.on_cancel += self._switch_to_delete_char_view
|
||||||
|
view.on_confirm += lambda: self._delete_character(player)
|
||||||
|
self._switch_view(view)
|
||||||
|
|
||||||
|
def _switch_to_play_view(self) -> None:
|
||||||
|
view = ChooseCharacterView(
|
||||||
|
self._screen, self.roster, "Choose character to play"
|
||||||
|
)
|
||||||
|
view.on_cancel += self._switch_to_roster_view
|
||||||
|
view.on_confirm += self._switch_to_game_view
|
||||||
|
self._switch_view(view)
|
||||||
|
|
||||||
|
def _switch_to_game_view(self, player: Player) -> None:
|
||||||
|
view = GameView(self._screen, self.roster, player, self.args)
|
||||||
|
view.on_exit += self._switch_to_roster_view
|
||||||
|
self._switch_view(view)
|
||||||
|
|
||||||
|
def _switch_view(self, view: BaseView) -> None:
|
||||||
|
self._view.stop()
|
||||||
|
self._view = view
|
||||||
|
self._view.start()
|
||||||
|
|
||||||
|
def _create_character(
|
||||||
|
self, name: str, race: Race, class_: Class, stats: Stats
|
||||||
|
) -> None:
|
||||||
|
player = create_player(name, race, class_, stats)
|
||||||
|
self.roster.players.append(player)
|
||||||
|
if self.args.use_saves:
|
||||||
|
self.roster.save()
|
||||||
|
self._switch_to_game_view(player)
|
||||||
|
|
||||||
|
def _delete_character(self, player: Player) -> None:
|
||||||
|
self.roster.players.remove(player)
|
||||||
|
self._switch_to_roster_view()
|
||||||
|
|
||||||
|
def _quit(self) -> None:
|
||||||
|
raise StopMainLoop
|
||||||
@@ -0,0 +1,18 @@
|
|||||||
|
_HAS_COLORS = False
|
||||||
|
|
||||||
|
COLOR_LOGO = 1
|
||||||
|
COLOR_LOGO_ALT = 2
|
||||||
|
COLOR_FOCUSED = 3
|
||||||
|
COLOR_SCROLLBAR_TRACK = 4
|
||||||
|
COLOR_SCROLLBAR_THUMB = 5
|
||||||
|
COLOR_PROGRESSBAR = 6
|
||||||
|
COLOR_HIGHLIGHT = 7
|
||||||
|
|
||||||
|
|
||||||
|
def has_colors() -> bool:
|
||||||
|
return _HAS_COLORS
|
||||||
|
|
||||||
|
|
||||||
|
def set_colors(has_colors: bool) -> None:
|
||||||
|
global _HAS_COLORS
|
||||||
|
_HAS_COLORS = has_colors
|
||||||
@@ -0,0 +1,14 @@
|
|||||||
|
import typing as T
|
||||||
|
|
||||||
|
|
||||||
|
class EventHandler:
|
||||||
|
def __init__(self) -> None:
|
||||||
|
self.callbacks: T.List[T.Callable[..., T.Any]] = []
|
||||||
|
|
||||||
|
def __call__(self, *args: T.Any, **kwargs: T.Any) -> None:
|
||||||
|
for callback in self.callbacks:
|
||||||
|
callback(*args, **kwargs)
|
||||||
|
|
||||||
|
def __iadd__(self, callback: T.Callable[..., T.Any]) -> "EventHandler":
|
||||||
|
self.callbacks.append(callback)
|
||||||
|
return self
|
||||||
@@ -0,0 +1,22 @@
|
|||||||
|
import curses
|
||||||
|
import curses.ascii
|
||||||
|
import typing as T
|
||||||
|
from dataclasses import dataclass
|
||||||
|
|
||||||
|
KEYS_CYCLE = {curses.ascii.TAB}
|
||||||
|
KEYS_DOWN = set(map(ord, "jJ")) | {curses.KEY_DOWN}
|
||||||
|
KEYS_LEFT = set(map(ord, "hH")) | {curses.KEY_LEFT}
|
||||||
|
KEYS_RIGHT = set(map(ord, "lL")) | {curses.KEY_RIGHT}
|
||||||
|
KEYS_UP = set(map(ord, "kK")) | {curses.KEY_UP}
|
||||||
|
KEYS_CANCEL = set(map(ord, "qQ")) | {curses.ascii.ESC}
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class Choice:
|
||||||
|
keys: T.List[int]
|
||||||
|
desc: str
|
||||||
|
callback: T.Callable[..., T.Any]
|
||||||
|
|
||||||
|
|
||||||
|
def first(source: T.Iterable[T.Any], default: T.Any = None) -> T.Any:
|
||||||
|
return next(iter(source), default)
|
||||||
@@ -0,0 +1,11 @@
|
|||||||
|
from .base_view import BaseView
|
||||||
|
from .choose_character_view import ChooseCharacterView
|
||||||
|
from .confirm_view import ConfirmView
|
||||||
|
from .create_character_view import (
|
||||||
|
ChooseCharacterClassView,
|
||||||
|
ChooseCharacterNameView,
|
||||||
|
ChooseCharacterRaceView,
|
||||||
|
ChooseCharacterStatsView,
|
||||||
|
)
|
||||||
|
from .game_view import GameView
|
||||||
|
from .roster_view import RosterView
|
||||||
@@ -0,0 +1,18 @@
|
|||||||
|
import typing as T
|
||||||
|
|
||||||
|
|
||||||
|
class BaseView:
|
||||||
|
def __init__(self, screen: T.Any) -> None:
|
||||||
|
self.screen = screen
|
||||||
|
|
||||||
|
def start(self) -> None:
|
||||||
|
pass
|
||||||
|
|
||||||
|
def stop(self) -> None:
|
||||||
|
pass
|
||||||
|
|
||||||
|
def idle(self) -> None:
|
||||||
|
pass
|
||||||
|
|
||||||
|
def keypress(self, key: int) -> None:
|
||||||
|
pass
|
||||||
@@ -0,0 +1,55 @@
|
|||||||
|
import functools
|
||||||
|
import typing as T
|
||||||
|
|
||||||
|
from pqcli.lingo import act_name, to_roman
|
||||||
|
from pqcli.roster import Roster
|
||||||
|
from pqcli.ui.curses.event_handler import EventHandler
|
||||||
|
from pqcli.ui.curses.util import KEYS_CANCEL, Choice
|
||||||
|
from pqcli.ui.curses.views.menu_view import MenuView
|
||||||
|
from pqcli.ui.curses.widgets import Menu
|
||||||
|
|
||||||
|
|
||||||
|
class ChooseCharacterView(MenuView):
|
||||||
|
def __init__(self, screen: T.Any, roster: Roster, title: str) -> None:
|
||||||
|
super().__init__(screen, title)
|
||||||
|
|
||||||
|
for y, player in enumerate(roster.players, 1):
|
||||||
|
key: T.Optional[str] = str(y % 10) if y <= 10 else None
|
||||||
|
|
||||||
|
best_equip = player.equipment.best
|
||||||
|
|
||||||
|
best_spell = player.spell_book.best
|
||||||
|
best_spell_name = (
|
||||||
|
(best_spell.name + " " + to_roman(best_spell.level))
|
||||||
|
if best_spell
|
||||||
|
else "-"
|
||||||
|
)
|
||||||
|
|
||||||
|
best_stat_name = (
|
||||||
|
f"{player.stats.best_prime.value} "
|
||||||
|
f"{player.stats[player.stats.best_prime]}"
|
||||||
|
)
|
||||||
|
|
||||||
|
label = (
|
||||||
|
f"[{key or '-'}] "
|
||||||
|
f"{player.name} the {player.race.name} "
|
||||||
|
f"({act_name(player.quest_book.act)})\n"
|
||||||
|
f" Level {player.level} {player.class_.name}\n"
|
||||||
|
f" {best_equip} / {best_spell_name} / {best_stat_name}"
|
||||||
|
)
|
||||||
|
|
||||||
|
self._choices.append(
|
||||||
|
Choice(
|
||||||
|
keys=[ord(key)] if key is not None else [],
|
||||||
|
desc=label,
|
||||||
|
callback=functools.partial(self.on_confirm, player),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
self._choices.append(
|
||||||
|
Choice(
|
||||||
|
keys=list(KEYS_CANCEL),
|
||||||
|
desc="[Q] Cancel",
|
||||||
|
callback=self.on_cancel,
|
||||||
|
)
|
||||||
|
)
|
||||||
@@ -0,0 +1,50 @@
|
|||||||
|
import typing as T
|
||||||
|
|
||||||
|
from pqcli.ui.curses.event_handler import EventHandler
|
||||||
|
from pqcli.ui.curses.util import Choice
|
||||||
|
from pqcli.ui.curses.widgets import Menu
|
||||||
|
|
||||||
|
from .base_view import BaseView
|
||||||
|
|
||||||
|
|
||||||
|
class ConfirmView(BaseView):
|
||||||
|
def __init__(self, screen: T.Any, title: str) -> None:
|
||||||
|
super().__init__(screen)
|
||||||
|
|
||||||
|
self.title = f"{title}:"
|
||||||
|
self.on_cancel = EventHandler()
|
||||||
|
self.on_confirm = EventHandler()
|
||||||
|
|
||||||
|
self.choices: T.List[Choice] = [
|
||||||
|
Choice(
|
||||||
|
keys=list(map(ord, "yY")),
|
||||||
|
desc=f"[Y] Yes",
|
||||||
|
callback=self.on_confirm,
|
||||||
|
),
|
||||||
|
Choice(
|
||||||
|
keys=list(map(ord, "nNqQ\N{ESC}")),
|
||||||
|
desc="[N] No",
|
||||||
|
callback=self.on_cancel,
|
||||||
|
),
|
||||||
|
]
|
||||||
|
|
||||||
|
def start(self) -> None:
|
||||||
|
scr_height, scr_width = self.screen.getmaxyx()
|
||||||
|
self.screen.erase()
|
||||||
|
self.screen.noutrefresh()
|
||||||
|
|
||||||
|
self.main_menu = Menu(
|
||||||
|
header=self.title,
|
||||||
|
choices=self.choices,
|
||||||
|
active_choice=0,
|
||||||
|
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()
|
||||||
@@ -0,0 +1,4 @@
|
|||||||
|
from .choose_character_class_view import ChooseCharacterClassView
|
||||||
|
from .choose_character_name_view import ChooseCharacterNameView
|
||||||
|
from .choose_character_race_view import ChooseCharacterRaceView
|
||||||
|
from .choose_character_stats_view import ChooseCharacterStatsView
|
||||||
@@ -0,0 +1,35 @@
|
|||||||
|
import functools
|
||||||
|
import typing as T
|
||||||
|
|
||||||
|
from pqcli.config import CLASSES, Class
|
||||||
|
from pqcli.ui.curses.util import KEYS_CANCEL, Choice
|
||||||
|
from pqcli.ui.curses.views.menu_view import MenuView
|
||||||
|
|
||||||
|
|
||||||
|
class ChooseCharacterClassView(MenuView):
|
||||||
|
def __init__(
|
||||||
|
self, screen: T.Any, class_: T.Optional[Class] = None
|
||||||
|
) -> None:
|
||||||
|
super().__init__(
|
||||||
|
screen,
|
||||||
|
"Choose character class",
|
||||||
|
CLASSES.index(class_) if class_ is not None else 0,
|
||||||
|
)
|
||||||
|
|
||||||
|
for y, class_ in enumerate(CLASSES, 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 '-'}] {class_.name}",
|
||||||
|
callback=functools.partial(self.on_confirm, class_),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
self._choices.append(
|
||||||
|
Choice(
|
||||||
|
keys=list(KEYS_CANCEL),
|
||||||
|
desc="[Q] Cancel",
|
||||||
|
callback=self.on_cancel,
|
||||||
|
)
|
||||||
|
)
|
||||||
@@ -0,0 +1,120 @@
|
|||||||
|
import curses
|
||||||
|
import curses.ascii
|
||||||
|
import typing as T
|
||||||
|
|
||||||
|
from pqcli.mechanic import generate_name
|
||||||
|
from pqcli.ui.curses.event_handler import EventHandler
|
||||||
|
from pqcli.ui.curses.util import KEYS_CANCEL, KEYS_CYCLE, KEYS_DOWN, KEYS_UP
|
||||||
|
from pqcli.ui.curses.views.base_view import BaseView
|
||||||
|
from pqcli.ui.curses.widgets.focusable import focus_standout
|
||||||
|
|
||||||
|
|
||||||
|
class ChooseCharacterNameView(BaseView):
|
||||||
|
def __init__(
|
||||||
|
self, screen: T.Any, character_name: T.Optional[str] = None
|
||||||
|
) -> None:
|
||||||
|
super().__init__(screen)
|
||||||
|
|
||||||
|
self.on_cancel = EventHandler()
|
||||||
|
self.on_confirm = EventHandler()
|
||||||
|
|
||||||
|
self._active_widget = 0
|
||||||
|
self._win: T.Optional[T.Any] = None
|
||||||
|
self._text_win: T.Optional[T.Any] = None
|
||||||
|
self._text = character_name or generate_name()
|
||||||
|
|
||||||
|
def start(self) -> None:
|
||||||
|
self.screen.erase()
|
||||||
|
self.screen.noutrefresh()
|
||||||
|
|
||||||
|
scr_height, scr_width = self.screen.getmaxyx()
|
||||||
|
|
||||||
|
h = 8
|
||||||
|
w = 30
|
||||||
|
y = (scr_height - h - 2) // 2
|
||||||
|
x = (scr_width - w - 2) // 2
|
||||||
|
self._text_win = curses.newwin(3, w + 2, y + 2, x - 1)
|
||||||
|
self._win = curses.newwin(h + 2, w + 2, y, x)
|
||||||
|
|
||||||
|
self._render()
|
||||||
|
|
||||||
|
def stop(self) -> None:
|
||||||
|
del self._win
|
||||||
|
del self._text_win
|
||||||
|
self._win = None
|
||||||
|
self._text_win = None
|
||||||
|
|
||||||
|
def keypress(self, key: int) -> None:
|
||||||
|
if key == curses.KEY_F5:
|
||||||
|
self._text = generate_name()
|
||||||
|
|
||||||
|
elif key == curses.KEY_F10:
|
||||||
|
self._finish()
|
||||||
|
|
||||||
|
elif key in KEYS_CYCLE:
|
||||||
|
self._active_widget = (self._active_widget + 1) % 4
|
||||||
|
|
||||||
|
elif self._active_widget == 0:
|
||||||
|
if key == curses.ascii.ESC:
|
||||||
|
self.on_cancel()
|
||||||
|
elif curses.ascii.isprint(key):
|
||||||
|
self._text += chr(key)
|
||||||
|
elif key == curses.KEY_BACKSPACE or key == curses.ascii.DEL:
|
||||||
|
self._text = self._text[:-1]
|
||||||
|
elif key == curses.ascii.NL:
|
||||||
|
self._finish()
|
||||||
|
elif key == curses.ascii.ETB: # ^w
|
||||||
|
self._text = ""
|
||||||
|
elif key == curses.KEY_DOWN:
|
||||||
|
self._active_widget = 1
|
||||||
|
|
||||||
|
else:
|
||||||
|
if key == curses.ascii.NL:
|
||||||
|
if self._active_widget == 1:
|
||||||
|
self._text = generate_name()
|
||||||
|
elif self._active_widget == 2:
|
||||||
|
self._finish()
|
||||||
|
elif self._active_widget == 3:
|
||||||
|
self.on_cancel()
|
||||||
|
else:
|
||||||
|
assert False
|
||||||
|
|
||||||
|
elif key in KEYS_CANCEL:
|
||||||
|
self.on_cancel()
|
||||||
|
|
||||||
|
elif key in KEYS_DOWN and self._active_widget < 3:
|
||||||
|
self._active_widget += 1
|
||||||
|
|
||||||
|
elif key in KEYS_UP and self._active_widget > 0:
|
||||||
|
self._active_widget -= 1
|
||||||
|
|
||||||
|
self._render()
|
||||||
|
|
||||||
|
def _finish(self) -> None:
|
||||||
|
self._text = self._text.strip()
|
||||||
|
if self._text:
|
||||||
|
self.on_confirm(self._text)
|
||||||
|
else:
|
||||||
|
self.on_cancel()
|
||||||
|
|
||||||
|
def _render(self) -> None:
|
||||||
|
if not self._win or not self._text_win:
|
||||||
|
return
|
||||||
|
|
||||||
|
self._win.erase()
|
||||||
|
self._win.addstr(0, 0, "Choose character name:")
|
||||||
|
with focus_standout(self._active_widget == 1, self._win):
|
||||||
|
self._win.addstr(6, 0, "[F5 ] Generate random name")
|
||||||
|
with focus_standout(self._active_widget == 2, self._win):
|
||||||
|
self._win.addstr(7, 0, "[F10 ] Continue")
|
||||||
|
with focus_standout(self._active_widget == 3, self._win):
|
||||||
|
self._win.addstr(8, 0, "[Esc ] Cancel")
|
||||||
|
self._win.refresh()
|
||||||
|
|
||||||
|
self._text_win.erase()
|
||||||
|
with focus_standout(self._active_widget == 0, self._text_win):
|
||||||
|
self._text_win.box()
|
||||||
|
self._text_win.addnstr(
|
||||||
|
1, 1, self._text, self._text_win.getmaxyx()[1] - 2
|
||||||
|
)
|
||||||
|
self._text_win.refresh()
|
||||||
@@ -0,0 +1,33 @@
|
|||||||
|
import functools
|
||||||
|
import typing as T
|
||||||
|
|
||||||
|
from pqcli.config import RACES, Race
|
||||||
|
from pqcli.ui.curses.util import KEYS_CANCEL, Choice
|
||||||
|
from pqcli.ui.curses.views.menu_view import MenuView
|
||||||
|
|
||||||
|
|
||||||
|
class ChooseCharacterRaceView(MenuView):
|
||||||
|
def __init__(self, screen: T.Any, race: T.Optional[Race] = None) -> None:
|
||||||
|
super().__init__(
|
||||||
|
screen,
|
||||||
|
"Choose character race",
|
||||||
|
RACES.index(race) if race is not None else 0,
|
||||||
|
)
|
||||||
|
|
||||||
|
for y, race in enumerate(RACES, 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 '-'}] {race.name}",
|
||||||
|
callback=functools.partial(self.on_confirm, race),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
self._choices.append(
|
||||||
|
Choice(
|
||||||
|
keys=list(KEYS_CANCEL),
|
||||||
|
desc="[Q] Cancel",
|
||||||
|
callback=self.on_cancel,
|
||||||
|
)
|
||||||
|
)
|
||||||
@@ -0,0 +1,111 @@
|
|||||||
|
import curses.ascii
|
||||||
|
import typing as T
|
||||||
|
|
||||||
|
from pqcli.config import PRIME_STATS
|
||||||
|
from pqcli.mechanic import StatsBuilder
|
||||||
|
from pqcli.ui.curses.event_handler import EventHandler
|
||||||
|
from pqcli.ui.curses.util import KEYS_CANCEL, KEYS_CYCLE, KEYS_DOWN, KEYS_UP
|
||||||
|
from pqcli.ui.curses.views.base_view import BaseView
|
||||||
|
from pqcli.ui.curses.widgets.focusable import focus_standout
|
||||||
|
|
||||||
|
|
||||||
|
class ChooseCharacterStatsView(BaseView):
|
||||||
|
def __init__(self, screen: T.Any) -> None:
|
||||||
|
super().__init__(screen)
|
||||||
|
|
||||||
|
self.on_cancel = EventHandler()
|
||||||
|
self.on_confirm = EventHandler()
|
||||||
|
|
||||||
|
self._active_widget = 0
|
||||||
|
self._win: T.Optional[T.Any] = None
|
||||||
|
self._stats_win: T.Optional[T.Any] = None
|
||||||
|
|
||||||
|
self._stats_builder = StatsBuilder()
|
||||||
|
self._stats = self._stats_builder.roll()
|
||||||
|
|
||||||
|
def start(self) -> None:
|
||||||
|
scr_height, scr_width = self.screen.getmaxyx()
|
||||||
|
|
||||||
|
h2 = len(PRIME_STATS) + 2
|
||||||
|
h = 7 + h2
|
||||||
|
w = 30
|
||||||
|
y = (scr_height - h - 2) // 2
|
||||||
|
x = (scr_width - w - 2) // 2
|
||||||
|
|
||||||
|
self.screen.erase()
|
||||||
|
self.screen.noutrefresh()
|
||||||
|
|
||||||
|
self._win = curses.newwin(h, w, y, x)
|
||||||
|
self._stats_win = curses.newwin(h2, w, y + 2, x - 1)
|
||||||
|
self._render()
|
||||||
|
|
||||||
|
def stop(self) -> None:
|
||||||
|
del self._win
|
||||||
|
del self._stats_win
|
||||||
|
self._win = None
|
||||||
|
self._stats_win = None
|
||||||
|
|
||||||
|
def keypress(self, key: int) -> None:
|
||||||
|
if key in KEYS_CANCEL:
|
||||||
|
self.on_cancel()
|
||||||
|
|
||||||
|
elif key == curses.ascii.NL:
|
||||||
|
if self._active_widget == 0:
|
||||||
|
self._stats = self._stats_builder.roll()
|
||||||
|
elif self._active_widget == 1:
|
||||||
|
self._stats = self._stats_builder.unroll()
|
||||||
|
elif self._active_widget == 2:
|
||||||
|
self.on_confirm(self._stats)
|
||||||
|
elif self._active_widget == 3:
|
||||||
|
self.on_cancel()
|
||||||
|
else:
|
||||||
|
assert False
|
||||||
|
|
||||||
|
elif key == curses.KEY_F5:
|
||||||
|
self._stats = self._stats_builder.roll()
|
||||||
|
|
||||||
|
elif key == curses.KEY_F6:
|
||||||
|
self._stats = self._stats_builder.unroll()
|
||||||
|
|
||||||
|
elif key == curses.KEY_F10:
|
||||||
|
self.on_confirm(self._stats)
|
||||||
|
|
||||||
|
elif key in KEYS_CYCLE:
|
||||||
|
self._active_widget = (self._active_widget + 1) % 4
|
||||||
|
|
||||||
|
elif key in KEYS_DOWN and self._active_widget < 3:
|
||||||
|
self._active_widget += 1
|
||||||
|
|
||||||
|
elif key in KEYS_UP and self._active_widget > 0:
|
||||||
|
self._active_widget -= 1
|
||||||
|
|
||||||
|
self._render()
|
||||||
|
|
||||||
|
def _render(self) -> None:
|
||||||
|
if not self._win or not self._stats_win:
|
||||||
|
return
|
||||||
|
|
||||||
|
y, x = self._win.getbegyx()
|
||||||
|
stats_h = self._stats_win.getmaxyx()[0]
|
||||||
|
|
||||||
|
self._win.erase()
|
||||||
|
self._win.addstr(0, 0, "Roll character stats:")
|
||||||
|
with focus_standout(self._active_widget == 0, self._win):
|
||||||
|
self._win.addstr(3 + stats_h + 0, 0, "[F5 ] Roll")
|
||||||
|
with focus_standout(self._active_widget == 1, self._win):
|
||||||
|
self._win.addstr(3 + stats_h + 1, 0, "[F6 ] Unroll")
|
||||||
|
with focus_standout(self._active_widget == 2, self._win):
|
||||||
|
self._win.addstr(3 + stats_h + 2, 0, "[F10 ] Continue")
|
||||||
|
with focus_standout(self._active_widget == 3, self._win):
|
||||||
|
self._win.addstr(3 + stats_h + 3, 0, "[Esc ] Cancel")
|
||||||
|
self._win.noutrefresh()
|
||||||
|
|
||||||
|
self._stats_win.erase()
|
||||||
|
for y, stat in enumerate(PRIME_STATS):
|
||||||
|
self._stats_win.addstr(
|
||||||
|
y + 1, 1, f"{stat.value}: {self._stats[stat]}"
|
||||||
|
)
|
||||||
|
self._stats_win.box()
|
||||||
|
self._stats_win.noutrefresh()
|
||||||
|
|
||||||
|
curses.doupdate()
|
||||||
@@ -0,0 +1,263 @@
|
|||||||
|
import argparse
|
||||||
|
import curses
|
||||||
|
import datetime
|
||||||
|
import typing as T
|
||||||
|
|
||||||
|
from pqcli.mechanic import Player, SignalMixin, Simulation
|
||||||
|
from pqcli.roster import Roster
|
||||||
|
from pqcli.ui.curses.event_handler import EventHandler
|
||||||
|
from pqcli.ui.curses.util import (
|
||||||
|
KEYS_CANCEL,
|
||||||
|
KEYS_CYCLE,
|
||||||
|
KEYS_DOWN,
|
||||||
|
KEYS_LEFT,
|
||||||
|
KEYS_RIGHT,
|
||||||
|
KEYS_UP,
|
||||||
|
)
|
||||||
|
from pqcli.ui.curses.views.base_view import BaseView
|
||||||
|
from pqcli.ui.curses.widgets import Focusable, Scrollable, Widget
|
||||||
|
|
||||||
|
from .character_sheet_window import CharacterSheetWindow
|
||||||
|
from .equipment_window import EquipmentWindow
|
||||||
|
from .inventory_window import InventoryWindow
|
||||||
|
from .plot_window import PlotWindow
|
||||||
|
from .quest_book_window import QuestBookWindow
|
||||||
|
from .spell_book_window import SpellBookWindow
|
||||||
|
from .task_progress_window import TaskProgressWindow
|
||||||
|
|
||||||
|
|
||||||
|
class GameView(BaseView):
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
screen: T.Any,
|
||||||
|
roster: Roster,
|
||||||
|
player: Player,
|
||||||
|
args: argparse.Namespace,
|
||||||
|
) -> None:
|
||||||
|
super().__init__(screen)
|
||||||
|
self.on_exit = EventHandler()
|
||||||
|
|
||||||
|
self._roster = roster
|
||||||
|
self._player = player
|
||||||
|
self._args = args
|
||||||
|
|
||||||
|
self._simulation = Simulation(player)
|
||||||
|
self._last_tick = datetime.datetime.now()
|
||||||
|
|
||||||
|
def keypress(self, key: int) -> None:
|
||||||
|
focused = self.focused
|
||||||
|
|
||||||
|
if key in KEYS_CANCEL:
|
||||||
|
self.on_exit()
|
||||||
|
|
||||||
|
elif key in {curses.KEY_PPAGE}:
|
||||||
|
if hasattr(focused, "scroll_page_up"):
|
||||||
|
focused.scroll_page_up()
|
||||||
|
|
||||||
|
elif key in {curses.KEY_NPAGE}:
|
||||||
|
if hasattr(focused, "scroll_page_down"):
|
||||||
|
focused.scroll_page_down()
|
||||||
|
|
||||||
|
elif key in KEYS_CYCLE:
|
||||||
|
if focused == self._char_sheet_win:
|
||||||
|
self.focus(self._spell_book_win)
|
||||||
|
elif focused == self._spell_book_win:
|
||||||
|
self.focus(self._equipment_win)
|
||||||
|
elif focused == self._equipment_win:
|
||||||
|
self.focus(self._inventory_win)
|
||||||
|
elif focused == self._inventory_win:
|
||||||
|
self.focus(self._plot_win)
|
||||||
|
elif focused == self._plot_win:
|
||||||
|
self.focus(self._quest_book_win)
|
||||||
|
elif focused == self._quest_book_win:
|
||||||
|
self.focus(self._char_sheet_win)
|
||||||
|
|
||||||
|
elif key in KEYS_DOWN:
|
||||||
|
if focused == self._char_sheet_win:
|
||||||
|
self.focus(self._spell_book_win)
|
||||||
|
elif focused == self._equipment_win:
|
||||||
|
self.focus(self._inventory_win)
|
||||||
|
elif focused == self._plot_win:
|
||||||
|
self.focus(self._quest_book_win)
|
||||||
|
|
||||||
|
elif key in KEYS_UP:
|
||||||
|
if focused == self._spell_book_win:
|
||||||
|
self.focus(self._char_sheet_win)
|
||||||
|
elif focused == self._inventory_win:
|
||||||
|
self.focus(self._equipment_win)
|
||||||
|
elif focused == self._quest_book_win:
|
||||||
|
self.focus(self._plot_win)
|
||||||
|
|
||||||
|
elif key in KEYS_LEFT:
|
||||||
|
if focused == self._equipment_win:
|
||||||
|
self.focus(self._char_sheet_win)
|
||||||
|
elif focused == self._inventory_win:
|
||||||
|
self.focus(self._spell_book_win)
|
||||||
|
elif focused == self._plot_win:
|
||||||
|
self.focus(self._equipment_win)
|
||||||
|
elif focused == self._quest_book_win:
|
||||||
|
self.focus(self._inventory_win)
|
||||||
|
|
||||||
|
elif key in KEYS_RIGHT:
|
||||||
|
if focused == self._char_sheet_win:
|
||||||
|
self.focus(self._equipment_win)
|
||||||
|
elif focused == self._spell_book_win:
|
||||||
|
self.focus(self._inventory_win)
|
||||||
|
elif focused == self._equipment_win:
|
||||||
|
self.focus(self._plot_win)
|
||||||
|
elif focused == self._inventory_win:
|
||||||
|
self.focus(self._quest_book_win)
|
||||||
|
|
||||||
|
elif self._args.cheats and key == ord("t"):
|
||||||
|
self._simulation.tick()
|
||||||
|
|
||||||
|
elif self._args.cheats and key == ord("T"):
|
||||||
|
for _ in range(100):
|
||||||
|
self._simulation.tick()
|
||||||
|
|
||||||
|
elif self._args.cheats and key == curses.ascii.DC4: # ^t
|
||||||
|
old = SignalMixin.emit
|
||||||
|
SignalMixin.emit = lambda *_: None
|
||||||
|
for _ in range(10000):
|
||||||
|
self._simulation.tick()
|
||||||
|
SignalMixin.emit = old
|
||||||
|
for child in self._children:
|
||||||
|
child.sync()
|
||||||
|
|
||||||
|
else:
|
||||||
|
super().keypress(key)
|
||||||
|
|
||||||
|
curses.doupdate()
|
||||||
|
|
||||||
|
def start(self) -> None:
|
||||||
|
scr_height, scr_width = self.screen.getmaxyx()
|
||||||
|
col1_width = int(scr_width * 0.3)
|
||||||
|
col3_width = col1_width
|
||||||
|
col2_width = scr_width - col1_width - col3_width
|
||||||
|
|
||||||
|
self.screen.erase()
|
||||||
|
self.screen.noutrefresh()
|
||||||
|
|
||||||
|
self._big_win = curses.newwin(scr_height, scr_width)
|
||||||
|
self._col1_win = self._big_win.derwin(scr_height - 2, col1_width, 0, 0)
|
||||||
|
self._col2_win = self._big_win.derwin(
|
||||||
|
self._col1_win.getmaxyx()[0],
|
||||||
|
col2_width,
|
||||||
|
0,
|
||||||
|
self._col1_win.getmaxyx()[1],
|
||||||
|
)
|
||||||
|
self._col3_win = self._big_win.derwin(
|
||||||
|
self._col1_win.getmaxyx()[0],
|
||||||
|
col3_width,
|
||||||
|
0,
|
||||||
|
self._col1_win.getmaxyx()[1] + self._col2_win.getmaxyx()[1],
|
||||||
|
)
|
||||||
|
|
||||||
|
self._char_sheet_win = CharacterSheetWindow(
|
||||||
|
player=self._player,
|
||||||
|
parent=self._col1_win,
|
||||||
|
h=min(17, self._col1_win.getmaxyx()[0]),
|
||||||
|
w=self._col1_win.getmaxyx()[1],
|
||||||
|
y=0,
|
||||||
|
x=0,
|
||||||
|
)
|
||||||
|
|
||||||
|
self._spell_book_win = SpellBookWindow(
|
||||||
|
player=self._player,
|
||||||
|
parent=self._col1_win,
|
||||||
|
h=self._col1_win.getmaxyx()[0]
|
||||||
|
- self._char_sheet_win.getmaxyx()[0],
|
||||||
|
w=self._col1_win.getmaxyx()[1],
|
||||||
|
y=self._char_sheet_win.getmaxyx()[0],
|
||||||
|
x=0,
|
||||||
|
)
|
||||||
|
|
||||||
|
self._equipment_win = EquipmentWindow(
|
||||||
|
player=self._player,
|
||||||
|
parent=self._col2_win,
|
||||||
|
h=min(15, self._col2_win.getmaxyx()[0]),
|
||||||
|
w=self._col2_win.getmaxyx()[1],
|
||||||
|
y=0,
|
||||||
|
x=0,
|
||||||
|
)
|
||||||
|
|
||||||
|
self._inventory_win = InventoryWindow(
|
||||||
|
player=self._player,
|
||||||
|
parent=self._col2_win,
|
||||||
|
h=max(
|
||||||
|
1,
|
||||||
|
self._col2_win.getmaxyx()[0]
|
||||||
|
- self._equipment_win.getmaxyx()[0],
|
||||||
|
),
|
||||||
|
w=self._col2_win.getmaxyx()[1],
|
||||||
|
y=self._equipment_win.getmaxyx()[0],
|
||||||
|
x=0,
|
||||||
|
)
|
||||||
|
|
||||||
|
self._plot_win = PlotWindow(
|
||||||
|
player=self._player,
|
||||||
|
parent=self._col3_win,
|
||||||
|
h=self._equipment_win.getmaxyx()[0],
|
||||||
|
w=self._col3_win.getmaxyx()[1],
|
||||||
|
y=0,
|
||||||
|
x=0,
|
||||||
|
)
|
||||||
|
|
||||||
|
self._quest_book_win = QuestBookWindow(
|
||||||
|
player=self._player,
|
||||||
|
parent=self._col3_win,
|
||||||
|
h=max(
|
||||||
|
1, self._col3_win.getmaxyx()[0] - self._plot_win.getmaxyx()[0]
|
||||||
|
),
|
||||||
|
w=self._col3_win.getmaxyx()[1],
|
||||||
|
y=self._plot_win.getmaxyx()[0],
|
||||||
|
x=0,
|
||||||
|
)
|
||||||
|
|
||||||
|
self._task_win = TaskProgressWindow(
|
||||||
|
player=self._player,
|
||||||
|
parent=self._big_win,
|
||||||
|
h=2,
|
||||||
|
w=scr_width,
|
||||||
|
y=scr_height - 2,
|
||||||
|
x=0,
|
||||||
|
)
|
||||||
|
|
||||||
|
self._focusable_children: T.List[Focusable] = [
|
||||||
|
self._char_sheet_win,
|
||||||
|
self._spell_book_win,
|
||||||
|
self._equipment_win,
|
||||||
|
self._inventory_win,
|
||||||
|
self._plot_win,
|
||||||
|
self._quest_book_win,
|
||||||
|
]
|
||||||
|
self._children: T.List[Widget] = self._focusable_children + [
|
||||||
|
self._task_win
|
||||||
|
]
|
||||||
|
|
||||||
|
def stop(self) -> None:
|
||||||
|
for child in self._children:
|
||||||
|
child.stop()
|
||||||
|
del self._col1_win
|
||||||
|
del self._col2_win
|
||||||
|
del self._col3_win
|
||||||
|
del self._big_win
|
||||||
|
|
||||||
|
def idle(self) -> None:
|
||||||
|
elapsed = datetime.datetime.now() - self._last_tick
|
||||||
|
self._simulation.tick(elapsed.total_seconds() * 1000)
|
||||||
|
self._last_tick = datetime.datetime.now()
|
||||||
|
if self._args.use_saves:
|
||||||
|
self._roster.save_periodically()
|
||||||
|
curses.doupdate()
|
||||||
|
|
||||||
|
@property
|
||||||
|
def focused(self) -> Focusable:
|
||||||
|
for widget in self._focusable_children:
|
||||||
|
if widget.focused:
|
||||||
|
return widget
|
||||||
|
raise AssertionError
|
||||||
|
|
||||||
|
def focus(self, widget: Focusable) -> None:
|
||||||
|
self.focused.focused = False
|
||||||
|
widget.focused = True
|
||||||
@@ -0,0 +1,66 @@
|
|||||||
|
import typing as T
|
||||||
|
|
||||||
|
from pqcli.mechanic import Player, StatType
|
||||||
|
from pqcli.ui.curses.widgets import Focusable
|
||||||
|
|
||||||
|
from .progress_bar_window import DataTableProgressBarWindow
|
||||||
|
|
||||||
|
|
||||||
|
class CharacterSheetWindow(Focusable, DataTableProgressBarWindow):
|
||||||
|
def __init__(
|
||||||
|
self, player: Player, parent: T.Any, h: int, w: int, y: int, x: int
|
||||||
|
) -> None:
|
||||||
|
super().__init__(
|
||||||
|
parent,
|
||||||
|
h,
|
||||||
|
w,
|
||||||
|
y,
|
||||||
|
x,
|
||||||
|
" Character Sheet ",
|
||||||
|
align_right=False,
|
||||||
|
show_time=True,
|
||||||
|
)
|
||||||
|
self._on_focus_change += self._render
|
||||||
|
|
||||||
|
self._focused = True
|
||||||
|
|
||||||
|
self._player = player
|
||||||
|
self._player.connect("level_up", self._sync_traits)
|
||||||
|
self._player.stats.connect("change", self._sync_traits)
|
||||||
|
self._player.exp_bar.connect("change", self._sync_exp)
|
||||||
|
|
||||||
|
self.sync()
|
||||||
|
|
||||||
|
def stop(self) -> None:
|
||||||
|
super().stop()
|
||||||
|
|
||||||
|
self._player.disconnect("level_up", self._sync_traits)
|
||||||
|
self._player.stats.disconnect("change", self._sync_traits)
|
||||||
|
self._player.exp_bar.disconnect("change", self._sync_exp)
|
||||||
|
|
||||||
|
def sync(self) -> None:
|
||||||
|
self._sync_traits()
|
||||||
|
self._sync_exp()
|
||||||
|
|
||||||
|
def _sync_traits(self) -> None:
|
||||||
|
if not self._win:
|
||||||
|
return
|
||||||
|
|
||||||
|
self._data_table.clear()
|
||||||
|
self._data_table.add("Name", self._player.name)
|
||||||
|
self._data_table.add("Race", self._player.race.name)
|
||||||
|
self._data_table.add("Class", self._player.class_.name)
|
||||||
|
self._data_table.add("Level", str(self._player.level))
|
||||||
|
self._data_table.add(" " * 15, "")
|
||||||
|
for stat in StatType:
|
||||||
|
self._data_table.add(stat.value, str(self._player.stats[stat]))
|
||||||
|
|
||||||
|
self._render_data_table()
|
||||||
|
|
||||||
|
def _sync_exp(self) -> None:
|
||||||
|
self._cur_pos = self._player.exp_bar.position
|
||||||
|
self._max_pos = self._player.exp_bar.max_
|
||||||
|
self._progress_title = (
|
||||||
|
f"Experience ({self._max_pos-self._cur_pos:.0f} XP to go)"
|
||||||
|
)
|
||||||
|
self._render_progress_bar()
|
||||||
@@ -0,0 +1,59 @@
|
|||||||
|
import typing as T
|
||||||
|
|
||||||
|
from pqcli.mechanic import EquipmentType, Player
|
||||||
|
from pqcli.ui.curses.widgets import DataTable, Focusable, WindowWrapper
|
||||||
|
|
||||||
|
|
||||||
|
class EquipmentWindow(Focusable, WindowWrapper):
|
||||||
|
def __init__(
|
||||||
|
self, player: Player, parent: T.Any, h: int, w: int, y: int, x: int
|
||||||
|
) -> None:
|
||||||
|
super().__init__(parent, h, w, y, x)
|
||||||
|
self._on_focus_change += self._render
|
||||||
|
|
||||||
|
self._data_table = DataTable(
|
||||||
|
self._win, h - 2, w - 2, 1, 1, align_right=False
|
||||||
|
)
|
||||||
|
|
||||||
|
self._player = player
|
||||||
|
self._player.equipment.connect("change", self._sync_equipment_change)
|
||||||
|
|
||||||
|
self.sync()
|
||||||
|
|
||||||
|
def stop(self) -> None:
|
||||||
|
super().stop()
|
||||||
|
self._data_table.stop()
|
||||||
|
|
||||||
|
self._player.equipment.disconnect(
|
||||||
|
"change", self._sync_equipment_change
|
||||||
|
)
|
||||||
|
|
||||||
|
def sync(self) -> None:
|
||||||
|
self._data_table.clear()
|
||||||
|
for equipment_type in EquipmentType:
|
||||||
|
self._data_table.add(
|
||||||
|
equipment_type.value.ljust(15),
|
||||||
|
self._player.equipment[equipment_type] or "",
|
||||||
|
)
|
||||||
|
self._data_table.select(None)
|
||||||
|
self._render()
|
||||||
|
|
||||||
|
def _sync_equipment_change(
|
||||||
|
self, equipment_type: EquipmentType, item_name: T.Optional[str]
|
||||||
|
) -> None:
|
||||||
|
self._data_table.set(equipment_type.value, item_name or "")
|
||||||
|
self._data_table.select(equipment_type.value)
|
||||||
|
self._render()
|
||||||
|
|
||||||
|
def _render(self) -> None:
|
||||||
|
if not self._win:
|
||||||
|
return
|
||||||
|
|
||||||
|
with self.focus_standout(self._win):
|
||||||
|
self._win.box()
|
||||||
|
text = " Equipment "
|
||||||
|
x = max(0, (self.getmaxyx()[1] - len(text)) // 2)
|
||||||
|
self._win.addnstr(0, x, text, min(len(text), self.getmaxyx()[1]))
|
||||||
|
|
||||||
|
self._win.noutrefresh()
|
||||||
|
self._data_table.render()
|
||||||
@@ -0,0 +1,113 @@
|
|||||||
|
import curses
|
||||||
|
import typing as T
|
||||||
|
|
||||||
|
from pqcli.lingo import to_roman
|
||||||
|
from pqcli.mechanic import InventoryItem, Player
|
||||||
|
from pqcli.ui.curses.widgets import (
|
||||||
|
DataTable,
|
||||||
|
Focusable,
|
||||||
|
ProgressBar,
|
||||||
|
WindowWrapper,
|
||||||
|
)
|
||||||
|
|
||||||
|
from .progress_bar_window import DataTableProgressBarWindow
|
||||||
|
|
||||||
|
|
||||||
|
class InventoryWindow(Focusable, DataTableProgressBarWindow):
|
||||||
|
def __init__(
|
||||||
|
self, player: Player, parent: T.Any, h: int, w: int, y: int, x: int
|
||||||
|
) -> None:
|
||||||
|
super().__init__(
|
||||||
|
parent,
|
||||||
|
h,
|
||||||
|
w,
|
||||||
|
y,
|
||||||
|
x,
|
||||||
|
" Inventory ",
|
||||||
|
align_right=True,
|
||||||
|
show_time=False,
|
||||||
|
)
|
||||||
|
self._on_focus_change += self._render
|
||||||
|
|
||||||
|
self._player = player
|
||||||
|
self._player.connect("new_task", self._on_new_task)
|
||||||
|
self._player.inventory.connect("gold_change", self._sync_gold)
|
||||||
|
self._player.inventory.connect("item_add", self._sync_item_add)
|
||||||
|
self._player.inventory.connect("item_del", self._sync_item_del)
|
||||||
|
self._player.inventory.connect("item_change", self._sync_item_change)
|
||||||
|
self._player.inventory.encum_bar.connect(
|
||||||
|
"change", self._sync_encumbrance
|
||||||
|
)
|
||||||
|
|
||||||
|
self.sync()
|
||||||
|
|
||||||
|
def stop(self) -> None:
|
||||||
|
super().stop()
|
||||||
|
|
||||||
|
self._player.disconnect("new_task", self._on_new_task)
|
||||||
|
self._player.inventory.disconnect("gold_change", self._sync_gold)
|
||||||
|
self._player.inventory.disconnect("item_add", self._sync_item_add)
|
||||||
|
self._player.inventory.disconnect("item_del", self._sync_item_del)
|
||||||
|
self._player.inventory.disconnect(
|
||||||
|
"item_change", self._sync_item_change
|
||||||
|
)
|
||||||
|
self._player.inventory.encum_bar.disconnect(
|
||||||
|
"change", self._sync_encumbrance
|
||||||
|
)
|
||||||
|
|
||||||
|
def scroll_page_up(self) -> None:
|
||||||
|
self._data_table.scroll_page_up()
|
||||||
|
self._render()
|
||||||
|
|
||||||
|
def scroll_page_down(self) -> None:
|
||||||
|
self._data_table.scroll_page_down()
|
||||||
|
self._render()
|
||||||
|
|
||||||
|
def sync(self) -> None:
|
||||||
|
self._sync_gold()
|
||||||
|
self._sync_items()
|
||||||
|
self._sync_encumbrance()
|
||||||
|
|
||||||
|
def _sync_encumbrance(self) -> None:
|
||||||
|
self._cur_pos = self._player.inventory.encum_bar.position
|
||||||
|
self._max_pos = self._player.inventory.encum_bar.max_
|
||||||
|
self._progress_title = (
|
||||||
|
f"Encumbrance ({self._cur_pos:.0f}/{self._max_pos} cubits)"
|
||||||
|
)
|
||||||
|
self._render_progress_bar()
|
||||||
|
|
||||||
|
def _sync_gold(self) -> None:
|
||||||
|
self._data_table.set("Gold", str(self._player.inventory.gold))
|
||||||
|
self._data_table.select("Gold")
|
||||||
|
self._render_data_table()
|
||||||
|
|
||||||
|
def _sync_items(self) -> None:
|
||||||
|
self._data_table.clear()
|
||||||
|
self._sync_gold()
|
||||||
|
for item in self._player.inventory:
|
||||||
|
self._data_table.add(item.name, str(item.quantity))
|
||||||
|
self._data_table.scroll_to_item(0)
|
||||||
|
self._data_table.select(None)
|
||||||
|
self._render_data_table()
|
||||||
|
|
||||||
|
def _sync_item_add(self, item: InventoryItem) -> None:
|
||||||
|
self._data_table.add(item.name, str(item.quantity))
|
||||||
|
self._data_table.select(item.name)
|
||||||
|
self._render_data_table()
|
||||||
|
|
||||||
|
def _sync_item_del(self, item: InventoryItem) -> None:
|
||||||
|
self._data_table.delete(item.name)
|
||||||
|
self._data_table.select(None)
|
||||||
|
self._render_data_table()
|
||||||
|
|
||||||
|
def _sync_item_change(self, item: InventoryItem) -> None:
|
||||||
|
self._data_table.set(item.name, str(item.quantity))
|
||||||
|
self._data_table.select(item.name)
|
||||||
|
self._render_data_table()
|
||||||
|
|
||||||
|
def _on_new_task(self) -> None:
|
||||||
|
description = (
|
||||||
|
self._player.task.description if self._player.task else "?"
|
||||||
|
)
|
||||||
|
if description.lower().startswith("sell"):
|
||||||
|
self._data_table.scroll_to_item(0)
|
||||||
@@ -0,0 +1,69 @@
|
|||||||
|
import typing as T
|
||||||
|
|
||||||
|
from pqcli.lingo import act_name
|
||||||
|
from pqcli.mechanic import Player
|
||||||
|
from pqcli.ui.curses.widgets import Focusable, ListBox, WindowWrapper
|
||||||
|
|
||||||
|
from .progress_bar_window import ListBoxProgressBarWindow
|
||||||
|
|
||||||
|
|
||||||
|
class PlotWindow(Focusable, ListBoxProgressBarWindow):
|
||||||
|
cutoff = 100
|
||||||
|
|
||||||
|
def __init__(
|
||||||
|
self, player: Player, parent: T.Any, h: int, w: int, y: int, x: int
|
||||||
|
) -> None:
|
||||||
|
super().__init__(
|
||||||
|
parent, h, w, y, x, " Plot Development ", show_time=True
|
||||||
|
)
|
||||||
|
self._on_focus_change += self._render
|
||||||
|
|
||||||
|
self._player = player
|
||||||
|
self._player.quest_book.connect("start_act", self._sync_act_add)
|
||||||
|
self._player.quest_book.plot_bar.connect("change", self._sync_position)
|
||||||
|
|
||||||
|
self.sync()
|
||||||
|
|
||||||
|
def stop(self) -> None:
|
||||||
|
super().stop()
|
||||||
|
self._list_box.stop()
|
||||||
|
|
||||||
|
self._player.quest_book.disconnect("start_act", self._sync_act_add)
|
||||||
|
self._player.quest_book.plot_bar.disconnect(
|
||||||
|
"change", self._sync_position
|
||||||
|
)
|
||||||
|
|
||||||
|
def scroll_page_up(self) -> None:
|
||||||
|
self._list_box.scroll_page_up()
|
||||||
|
self._render()
|
||||||
|
|
||||||
|
def scroll_page_down(self) -> None:
|
||||||
|
self._list_box.scroll_page_down()
|
||||||
|
self._render()
|
||||||
|
|
||||||
|
def sync(self) -> None:
|
||||||
|
self._sync_acts()
|
||||||
|
self._sync_position()
|
||||||
|
|
||||||
|
def _sync_acts(self) -> None:
|
||||||
|
self._list_box.clear()
|
||||||
|
for act_number in range(
|
||||||
|
max(0, self._player.quest_book.act - self.cutoff),
|
||||||
|
self._player.quest_book.act + 1,
|
||||||
|
):
|
||||||
|
self._sync_act_add(act_number)
|
||||||
|
self._render_list_box()
|
||||||
|
|
||||||
|
def _sync_act_add(self, act_number: int) -> None:
|
||||||
|
self._list_box.delete(0, max(0, len(self._list_box) - self.cutoff))
|
||||||
|
prev = self._list_box.get(-1)
|
||||||
|
if prev is not None:
|
||||||
|
self._list_box.set(-1, "[X] " + prev[4:])
|
||||||
|
self._list_box.add("[ ] " + act_name(act_number))
|
||||||
|
self._list_box.select(-1)
|
||||||
|
self._render_list_box()
|
||||||
|
|
||||||
|
def _sync_position(self) -> None:
|
||||||
|
self._cur_pos = self._player.quest_book.plot_bar.position
|
||||||
|
self._max_pos = self._player.quest_book.plot_bar.max_
|
||||||
|
self._render_progress_bar()
|
||||||
@@ -0,0 +1,155 @@
|
|||||||
|
import curses
|
||||||
|
import typing as T
|
||||||
|
|
||||||
|
from pqcli.ui.curses.widgets import (
|
||||||
|
DataTable,
|
||||||
|
Focusable,
|
||||||
|
ListBox,
|
||||||
|
ProgressBar,
|
||||||
|
WindowWrapper,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class ProgressBarWindow(WindowWrapper):
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
parent: T.Any,
|
||||||
|
h: int,
|
||||||
|
w: int,
|
||||||
|
y: int,
|
||||||
|
x: int,
|
||||||
|
title: str,
|
||||||
|
show_time: bool,
|
||||||
|
) -> None:
|
||||||
|
super().__init__(parent, h, w, y, x)
|
||||||
|
|
||||||
|
self._title = title
|
||||||
|
self._cur_pos = 0.0
|
||||||
|
self._max_pos = 1.0
|
||||||
|
self._progress_title = ""
|
||||||
|
|
||||||
|
try:
|
||||||
|
self._progress_bar_win = (
|
||||||
|
self._win.derwin(3, w, h - 3, 0) if self._win else None
|
||||||
|
)
|
||||||
|
except curses.error:
|
||||||
|
self._progress_bar_win = None
|
||||||
|
self._progress_bar = ProgressBar(
|
||||||
|
self._progress_bar_win, 1, w - 2, 1, 1, show_time=show_time
|
||||||
|
)
|
||||||
|
|
||||||
|
def stop(self) -> None:
|
||||||
|
super().stop()
|
||||||
|
self._progress_bar.stop()
|
||||||
|
|
||||||
|
del self._progress_bar_win
|
||||||
|
self._progress_bar_win = None
|
||||||
|
|
||||||
|
def _render_progress_bar(self) -> None:
|
||||||
|
if not self._progress_bar_win:
|
||||||
|
return
|
||||||
|
|
||||||
|
self._progress_bar_win.erase()
|
||||||
|
|
||||||
|
with Focusable.focus_standout(self, self._progress_bar_win):
|
||||||
|
self._progress_bar_win.border(
|
||||||
|
curses.ACS_VLINE,
|
||||||
|
curses.ACS_VLINE,
|
||||||
|
curses.ACS_HLINE,
|
||||||
|
curses.ACS_HLINE,
|
||||||
|
curses.ACS_LTEE,
|
||||||
|
curses.ACS_RTEE,
|
||||||
|
curses.ACS_LLCORNER,
|
||||||
|
curses.ACS_LRCORNER,
|
||||||
|
)
|
||||||
|
|
||||||
|
text = self._progress_title
|
||||||
|
x = max(0, (self.getmaxyx()[1] - len(text)) // 2)
|
||||||
|
self._progress_bar_win.addnstr(
|
||||||
|
0,
|
||||||
|
x,
|
||||||
|
text,
|
||||||
|
min(len(text), self._progress_bar_win.getmaxyx()[1]),
|
||||||
|
)
|
||||||
|
|
||||||
|
self._progress_bar.set_position(self._cur_pos, self._max_pos)
|
||||||
|
self._progress_bar_win.noutrefresh()
|
||||||
|
|
||||||
|
|
||||||
|
class DataTableProgressBarWindow(ProgressBarWindow):
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
parent: T.Any,
|
||||||
|
h: int,
|
||||||
|
w: int,
|
||||||
|
y: int,
|
||||||
|
x: int,
|
||||||
|
title: str,
|
||||||
|
align_right: bool,
|
||||||
|
show_time: bool,
|
||||||
|
) -> None:
|
||||||
|
super().__init__(parent, h, w, y, x, title, show_time)
|
||||||
|
|
||||||
|
self._data_table = DataTable(
|
||||||
|
self._win, h - 4, w - 2, 1, 1, align_right
|
||||||
|
)
|
||||||
|
|
||||||
|
def stop(self) -> None:
|
||||||
|
super().stop()
|
||||||
|
self._data_table.stop()
|
||||||
|
|
||||||
|
def _render_data_table(self) -> None:
|
||||||
|
if not self._win:
|
||||||
|
return
|
||||||
|
|
||||||
|
with Focusable.focus_standout(self, self._win):
|
||||||
|
self._win.box()
|
||||||
|
x = max(0, (self.getmaxyx()[1] - len(self._title)) // 2)
|
||||||
|
self._win.addnstr(
|
||||||
|
0, x, self._title, min(len(self._title), self.getmaxyx()[1])
|
||||||
|
)
|
||||||
|
|
||||||
|
self._win.noutrefresh()
|
||||||
|
self._data_table.render()
|
||||||
|
|
||||||
|
def _render(self) -> None:
|
||||||
|
self._render_data_table()
|
||||||
|
self._render_progress_bar()
|
||||||
|
|
||||||
|
|
||||||
|
class ListBoxProgressBarWindow(ProgressBarWindow):
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
parent: T.Any,
|
||||||
|
h: int,
|
||||||
|
w: int,
|
||||||
|
y: int,
|
||||||
|
x: int,
|
||||||
|
title: str,
|
||||||
|
show_time: bool,
|
||||||
|
) -> None:
|
||||||
|
super().__init__(parent, h, w, y, x, title, show_time)
|
||||||
|
|
||||||
|
self._list_box = ListBox(self._win, h - 4, w - 2, 1, 1)
|
||||||
|
|
||||||
|
def stop(self) -> None:
|
||||||
|
super().stop()
|
||||||
|
self._list_box.stop()
|
||||||
|
|
||||||
|
def _render_list_box(self) -> None:
|
||||||
|
if not self._win:
|
||||||
|
return
|
||||||
|
|
||||||
|
with Focusable.focus_standout(self, self._win):
|
||||||
|
self._win.box()
|
||||||
|
x = max(0, (self.getmaxyx()[1] - len(self._title)) // 2)
|
||||||
|
self._win.addnstr(
|
||||||
|
0, x, self._title, min(len(self._title), self.getmaxyx()[1])
|
||||||
|
)
|
||||||
|
|
||||||
|
self._win.noutrefresh()
|
||||||
|
self._list_box.render()
|
||||||
|
|
||||||
|
def _render(self) -> None:
|
||||||
|
self._render_list_box()
|
||||||
|
self._render_progress_bar()
|
||||||
@@ -0,0 +1,66 @@
|
|||||||
|
import typing as T
|
||||||
|
|
||||||
|
from pqcli.mechanic import Player
|
||||||
|
from pqcli.ui.curses.widgets import Focusable, ListBox, WindowWrapper
|
||||||
|
|
||||||
|
from .progress_bar_window import ListBoxProgressBarWindow
|
||||||
|
|
||||||
|
|
||||||
|
class QuestBookWindow(Focusable, ListBoxProgressBarWindow):
|
||||||
|
cutoff = 100
|
||||||
|
|
||||||
|
def __init__(
|
||||||
|
self, player: Player, parent: T.Any, h: int, w: int, y: int, x: int
|
||||||
|
) -> None:
|
||||||
|
super().__init__(parent, h, w, y, x, " Quests ", show_time=True)
|
||||||
|
self._on_focus_change += self._render
|
||||||
|
|
||||||
|
self._player = player
|
||||||
|
self._player.quest_book.connect("start_quest", self._sync_quest_add)
|
||||||
|
self._player.quest_book.quest_bar.connect(
|
||||||
|
"change", self._sync_position
|
||||||
|
)
|
||||||
|
|
||||||
|
self.sync()
|
||||||
|
|
||||||
|
def stop(self) -> None:
|
||||||
|
super().stop()
|
||||||
|
self._list_box.stop()
|
||||||
|
|
||||||
|
self._player.quest_book.disconnect("start_quest", self._sync_quest_add)
|
||||||
|
self._player.quest_book.quest_bar.disconnect(
|
||||||
|
"change", self._sync_position
|
||||||
|
)
|
||||||
|
|
||||||
|
def scroll_page_up(self) -> None:
|
||||||
|
self._list_box.scroll_page_up()
|
||||||
|
self._render()
|
||||||
|
|
||||||
|
def scroll_page_down(self) -> None:
|
||||||
|
self._list_box.scroll_page_down()
|
||||||
|
self._render()
|
||||||
|
|
||||||
|
def sync(self) -> None:
|
||||||
|
self._sync_quests()
|
||||||
|
self._sync_position()
|
||||||
|
|
||||||
|
def _sync_quests(self) -> None:
|
||||||
|
self._list_box.clear()
|
||||||
|
for quest_name in self._player.quest_book.quests[-self.cutoff :]:
|
||||||
|
self._sync_quest_add(quest_name)
|
||||||
|
self._list_box.select(-1)
|
||||||
|
self._render_list_box()
|
||||||
|
|
||||||
|
def _sync_quest_add(self, quest_name: str) -> None:
|
||||||
|
self._list_box.delete(0, max(0, len(self._list_box) - self.cutoff))
|
||||||
|
prev = self._list_box.get(-1)
|
||||||
|
if prev is not None:
|
||||||
|
self._list_box.set(-1, "[X] " + prev[4:])
|
||||||
|
self._list_box.add("[ ] " + quest_name)
|
||||||
|
self._list_box.select(-1)
|
||||||
|
self._render_list_box()
|
||||||
|
|
||||||
|
def _sync_position(self) -> None:
|
||||||
|
self._cur_pos = self._player.quest_book.quest_bar.position
|
||||||
|
self._max_pos = self._player.quest_book.quest_bar.max_
|
||||||
|
self._render_progress_bar()
|
||||||
@@ -0,0 +1,68 @@
|
|||||||
|
import typing as T
|
||||||
|
|
||||||
|
from pqcli.lingo import to_roman
|
||||||
|
from pqcli.mechanic import Player, Spell
|
||||||
|
from pqcli.ui.curses.widgets import DataTable, Focusable, WindowWrapper
|
||||||
|
|
||||||
|
|
||||||
|
class SpellBookWindow(Focusable, WindowWrapper):
|
||||||
|
def __init__(
|
||||||
|
self, player: Player, parent: T.Any, h: int, w: int, y: int, x: int
|
||||||
|
) -> None:
|
||||||
|
super().__init__(parent, h, w, y, x)
|
||||||
|
self._on_focus_change += self._render
|
||||||
|
|
||||||
|
self._data_table = DataTable(
|
||||||
|
self._win, h - 2, w - 2, 1, 1, align_right=True
|
||||||
|
)
|
||||||
|
|
||||||
|
self._player = player
|
||||||
|
self._player.spell_book.connect("add", self._sync_spell_add)
|
||||||
|
self._player.spell_book.connect("change", self._sync_spell_change)
|
||||||
|
|
||||||
|
self.sync()
|
||||||
|
|
||||||
|
def stop(self) -> None:
|
||||||
|
super().stop()
|
||||||
|
self._data_table.stop()
|
||||||
|
|
||||||
|
self._player.spell_book.disconnect("add", self._sync_spell_add)
|
||||||
|
self._player.spell_book.disconnect("change", self._sync_spell_change)
|
||||||
|
|
||||||
|
def scroll_page_up(self) -> None:
|
||||||
|
self._data_table.scroll_page_up()
|
||||||
|
self._render()
|
||||||
|
|
||||||
|
def scroll_page_down(self) -> None:
|
||||||
|
self._data_table.scroll_page_down()
|
||||||
|
self._render()
|
||||||
|
|
||||||
|
def sync(self) -> None:
|
||||||
|
self._data_table.clear()
|
||||||
|
for spell in self._player.spell_book:
|
||||||
|
self._data_table.add(spell.name, to_roman(spell.level))
|
||||||
|
self._data_table.select(None)
|
||||||
|
self._render()
|
||||||
|
|
||||||
|
def _sync_spell_add(self, spell: Spell) -> None:
|
||||||
|
self._data_table.add(spell.name, to_roman(spell.level))
|
||||||
|
self._data_table.select(spell.name)
|
||||||
|
self._render()
|
||||||
|
|
||||||
|
def _sync_spell_change(self, spell: Spell) -> None:
|
||||||
|
self._data_table.set(spell.name, to_roman(spell.level))
|
||||||
|
self._data_table.select(spell.name)
|
||||||
|
self._render()
|
||||||
|
|
||||||
|
def _render(self) -> None:
|
||||||
|
if not self._win:
|
||||||
|
return
|
||||||
|
|
||||||
|
with self.focus_standout(self._win):
|
||||||
|
self._win.box()
|
||||||
|
text = " Spell Book "
|
||||||
|
x = max(0, (self.getmaxyx()[1] - len(text)) // 2)
|
||||||
|
self._win.addnstr(0, x, text, min(len(text), self.getmaxyx()[1]))
|
||||||
|
|
||||||
|
self._win.noutrefresh()
|
||||||
|
self._data_table.render()
|
||||||
@@ -0,0 +1,50 @@
|
|||||||
|
import typing as T
|
||||||
|
|
||||||
|
from pqcli.mechanic import Player
|
||||||
|
from pqcli.ui.curses.widgets import (
|
||||||
|
Focusable,
|
||||||
|
Label,
|
||||||
|
ProgressBar,
|
||||||
|
WindowWrapper,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class TaskProgressWindow(Focusable, WindowWrapper):
|
||||||
|
def __init__(
|
||||||
|
self, player: Player, parent: T.Any, h: int, w: int, y: int, x: int
|
||||||
|
) -> None:
|
||||||
|
super().__init__(parent, h, w, y, x)
|
||||||
|
|
||||||
|
self._desc_win = Label(self._win, 1, w, 0, 0)
|
||||||
|
self._progress_bar = ProgressBar(
|
||||||
|
self._win, 1, w, 1, 0, show_time=False
|
||||||
|
)
|
||||||
|
|
||||||
|
self._player = player
|
||||||
|
self._player.connect("new_task", self._sync_task_name)
|
||||||
|
self._player.task_bar.connect("change", self._sync_position)
|
||||||
|
|
||||||
|
self.sync()
|
||||||
|
|
||||||
|
def stop(self) -> None:
|
||||||
|
super().stop()
|
||||||
|
|
||||||
|
self._desc_win.stop()
|
||||||
|
self._progress_bar.stop()
|
||||||
|
|
||||||
|
self._player.disconnect("new_task", self._sync_task_name)
|
||||||
|
self._player.task_bar.disconnect("change", self._sync_position)
|
||||||
|
|
||||||
|
def sync(self) -> None:
|
||||||
|
self._sync_task_name()
|
||||||
|
self._sync_position()
|
||||||
|
|
||||||
|
def _sync_task_name(self) -> None:
|
||||||
|
self._desc_win.set_text(
|
||||||
|
f"{self._player.task.description}..." if self._player.task else "?"
|
||||||
|
)
|
||||||
|
|
||||||
|
def _sync_position(self) -> None:
|
||||||
|
self._progress_bar.set_position(
|
||||||
|
self._player.task_bar.position, self._player.task_bar.max_
|
||||||
|
)
|
||||||
@@ -0,0 +1,42 @@
|
|||||||
|
import functools
|
||||||
|
import typing as T
|
||||||
|
|
||||||
|
from pqcli.ui.curses.event_handler import EventHandler
|
||||||
|
from pqcli.ui.curses.util import Choice
|
||||||
|
from pqcli.ui.curses.views.base_view import BaseView
|
||||||
|
from pqcli.ui.curses.widgets import Menu
|
||||||
|
|
||||||
|
|
||||||
|
class MenuView(BaseView):
|
||||||
|
def __init__(
|
||||||
|
self, screen: T.Any, title: str, active_choice: int = 0
|
||||||
|
) -> None:
|
||||||
|
super().__init__(screen)
|
||||||
|
|
||||||
|
self.on_cancel = EventHandler()
|
||||||
|
self.on_confirm = EventHandler()
|
||||||
|
|
||||||
|
self._title = f"{title}:"
|
||||||
|
self._choices: T.List[Choice] = []
|
||||||
|
self._active_choice = active_choice
|
||||||
|
|
||||||
|
def start(self) -> None:
|
||||||
|
scr_height, scr_width = self.screen.getmaxyx()
|
||||||
|
self.screen.erase()
|
||||||
|
self.screen.noutrefresh()
|
||||||
|
|
||||||
|
self.main_menu = Menu(
|
||||||
|
header=self._title,
|
||||||
|
choices=self._choices,
|
||||||
|
active_choice=self._active_choice,
|
||||||
|
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()
|
||||||
@@ -0,0 +1,99 @@
|
|||||||
|
import curses
|
||||||
|
import typing as T
|
||||||
|
|
||||||
|
from pqcli.ui.curses.colors import COLOR_LOGO, COLOR_LOGO_ALT, has_colors
|
||||||
|
from pqcli.ui.curses.event_handler import EventHandler
|
||||||
|
from pqcli.ui.curses.util import KEYS_CANCEL, Choice
|
||||||
|
from pqcli.ui.curses.widgets import Menu
|
||||||
|
|
||||||
|
from .base_view import BaseView
|
||||||
|
|
||||||
|
LOGO = """
|
||||||
|
█▀▀▄ ▄▀▀█
|
||||||
|
▀▄ ▀▄ ▄▀ ▄▀
|
||||||
|
▀▄ ▀▄ ▄▀ ▄▀
|
||||||
|
▀▄ ▄▀ ▄▀
|
||||||
|
▄▀ ▄▀▄
|
||||||
|
▄▄▄▄▀ ▄▀▄ ▀▄▄▄▄
|
||||||
|
▀▄ █▄▄▀ ▀▄▄█ ▄▀
|
||||||
|
▄▀█▄ █ █ ▄█▀▄
|
||||||
|
▄▄▀▄▀ ▀▀ ▀▀ ▀▄▀▄▄
|
||||||
|
█▄█ █▄█
|
||||||
|
|
||||||
|
█▀▀▄ ▄▀▀▄ █
|
||||||
|
█▄▄▀ █▀▀ ▄▀▀▄ ▄▀▀▄ █▀▀ ▄▀▀▄ ▄▀▀▀ ▄▀▀▀ █ █ █ █ ▄▀▀▄ ▄▀▀▀ ▀█▀
|
||||||
|
█ █ █ █ ▀▄▄█ █ █▀▀ ▀▀▄ ▀▀▄ █ ▌█ █ █ █▀▀ ▀▀▄ █
|
||||||
|
▀ ▀ ▀▀ ▄▄▀ ▀ ▀▀ ▀▀▀ ▀▀▀ ▀▀▌ ▀▀ ▀▀ ▀▀▀ ▀
|
||||||
|
""".strip(
|
||||||
|
"\n"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class MainMenu(Menu):
|
||||||
|
def __init__(
|
||||||
|
self, choices: T.List[Choice], scr_height: int, scr_width: int
|
||||||
|
) -> None:
|
||||||
|
super().__init__(
|
||||||
|
header=LOGO,
|
||||||
|
choices=choices,
|
||||||
|
active_choice=0,
|
||||||
|
scr_height=scr_height,
|
||||||
|
scr_width=scr_width,
|
||||||
|
)
|
||||||
|
if has_colors():
|
||||||
|
for y in range(len(self._header_lines)):
|
||||||
|
self._pad.chgat(y, 0, curses.color_pair(COLOR_LOGO))
|
||||||
|
self._pad.chgat(3, 29, 5, curses.color_pair(COLOR_LOGO_ALT))
|
||||||
|
self._pad.chgat(4, 29, 5, curses.color_pair(COLOR_LOGO_ALT))
|
||||||
|
self._pad.chgat(5, 32, 4, curses.color_pair(COLOR_LOGO_ALT))
|
||||||
|
self._pad.chgat(5, 27, 4, curses.color_pair(COLOR_LOGO_ALT))
|
||||||
|
self._pad.chgat(6, 34, 2, curses.color_pair(COLOR_LOGO_ALT))
|
||||||
|
self._pad.chgat(6, 27, 2, curses.color_pair(COLOR_LOGO_ALT))
|
||||||
|
|
||||||
|
|
||||||
|
class RosterView(BaseView):
|
||||||
|
def __init__(self, screen: T.Any) -> None:
|
||||||
|
super().__init__(screen)
|
||||||
|
|
||||||
|
self.on_create = EventHandler()
|
||||||
|
self.on_play = EventHandler()
|
||||||
|
self.on_delete = EventHandler()
|
||||||
|
self.on_quit = EventHandler()
|
||||||
|
|
||||||
|
self.choices: T.List[Choice] = [
|
||||||
|
Choice(
|
||||||
|
keys=list(map(ord, "cC")),
|
||||||
|
desc="[C] Create new character",
|
||||||
|
callback=self.on_create,
|
||||||
|
),
|
||||||
|
Choice(
|
||||||
|
keys=list(map(ord, "pP")),
|
||||||
|
desc="[P] Play",
|
||||||
|
callback=self.on_play,
|
||||||
|
),
|
||||||
|
Choice(
|
||||||
|
keys=list(map(ord, "dD")),
|
||||||
|
desc="[D] Delete character",
|
||||||
|
callback=self.on_delete,
|
||||||
|
),
|
||||||
|
Choice(
|
||||||
|
keys=list(KEYS_CANCEL), desc="[Q] Quit", callback=self.on_quit
|
||||||
|
),
|
||||||
|
]
|
||||||
|
|
||||||
|
def start(self) -> None:
|
||||||
|
scr_height, scr_width = self.screen.getmaxyx()
|
||||||
|
self.screen.erase()
|
||||||
|
self.screen.noutrefresh()
|
||||||
|
|
||||||
|
self.main_menu = MainMenu(
|
||||||
|
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()
|
||||||
@@ -0,0 +1,8 @@
|
|||||||
|
from .base import Widget, WindowWrapper
|
||||||
|
from .data_table import DataTable
|
||||||
|
from .focusable import Focusable
|
||||||
|
from .label import Label
|
||||||
|
from .list_box import ListBox
|
||||||
|
from .menu import Menu
|
||||||
|
from .progress_bar import ProgressBar
|
||||||
|
from .scrollable import Scrollable
|
||||||
@@ -0,0 +1,41 @@
|
|||||||
|
import curses
|
||||||
|
import typing as T
|
||||||
|
|
||||||
|
|
||||||
|
class Widget:
|
||||||
|
def __del__(self) -> None:
|
||||||
|
self.stop()
|
||||||
|
|
||||||
|
def stop(self) -> None:
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class WindowWrapper(Widget):
|
||||||
|
def __init__(
|
||||||
|
self, parent: T.Optional[T.Any], h: int, w: int, y: int, x: int
|
||||||
|
) -> None:
|
||||||
|
self._parent = parent
|
||||||
|
self._win: T.Optional[T.Any]
|
||||||
|
|
||||||
|
if not parent:
|
||||||
|
self._win = None
|
||||||
|
else:
|
||||||
|
try:
|
||||||
|
self._win = parent.derwin(h, w, y, x)
|
||||||
|
self._win.noutrefresh()
|
||||||
|
except curses.error:
|
||||||
|
self._win = None
|
||||||
|
|
||||||
|
def stop(self) -> None:
|
||||||
|
del self._win
|
||||||
|
self._win = None
|
||||||
|
|
||||||
|
def getbegyx(self) -> T.Tuple[int, int]:
|
||||||
|
if not self._win:
|
||||||
|
return (0, 0)
|
||||||
|
return self._win.getbegyx()
|
||||||
|
|
||||||
|
def getmaxyx(self) -> T.Tuple[int, int]:
|
||||||
|
if not self._win:
|
||||||
|
return (0, 0)
|
||||||
|
return self._win.getmaxyx()
|
||||||
@@ -0,0 +1,81 @@
|
|||||||
|
import curses
|
||||||
|
import enum
|
||||||
|
import typing as T
|
||||||
|
|
||||||
|
from pqcli.ui.curses.colors import COLOR_HIGHLIGHT, has_colors
|
||||||
|
|
||||||
|
from .scrollable import Scrollable
|
||||||
|
|
||||||
|
|
||||||
|
class DataTable(Scrollable):
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
parent: T.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: 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))
|
||||||
|
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: 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():
|
||||||
|
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))
|
||||||
|
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])
|
||||||
@@ -0,0 +1,42 @@
|
|||||||
|
import contextlib
|
||||||
|
import curses
|
||||||
|
import typing as T
|
||||||
|
|
||||||
|
from pqcli.ui.curses.colors import COLOR_FOCUSED, has_colors
|
||||||
|
from pqcli.ui.curses.event_handler import EventHandler
|
||||||
|
|
||||||
|
|
||||||
|
@contextlib.contextmanager
|
||||||
|
def focus_standout(focused: bool, win: T.Any) -> T.Generator:
|
||||||
|
if focused:
|
||||||
|
if has_colors():
|
||||||
|
win.attron(curses.color_pair(COLOR_FOCUSED))
|
||||||
|
else:
|
||||||
|
win.standout()
|
||||||
|
yield
|
||||||
|
if focused:
|
||||||
|
if has_colors():
|
||||||
|
win.attroff(curses.color_pair(COLOR_FOCUSED))
|
||||||
|
else:
|
||||||
|
win.standend()
|
||||||
|
|
||||||
|
|
||||||
|
class Focusable:
|
||||||
|
def __init__(self, *args: T.Any, **kwargs: T.Any) -> None:
|
||||||
|
self._focused = False
|
||||||
|
self._on_focus_change = EventHandler()
|
||||||
|
super().__init__(*args, **kwargs)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def focused(self) -> bool:
|
||||||
|
return self._focused
|
||||||
|
|
||||||
|
@focused.setter
|
||||||
|
def focused(self, focused: bool) -> None:
|
||||||
|
self._focused = focused
|
||||||
|
self._on_focus_change()
|
||||||
|
|
||||||
|
@contextlib.contextmanager
|
||||||
|
def focus_standout(self, win: T.Any) -> T.Generator:
|
||||||
|
with focus_standout(self._focused, win):
|
||||||
|
yield
|
||||||
@@ -0,0 +1,10 @@
|
|||||||
|
from .base import WindowWrapper
|
||||||
|
|
||||||
|
|
||||||
|
class Label(WindowWrapper):
|
||||||
|
def set_text(self, text: str) -> None:
|
||||||
|
if not self._win:
|
||||||
|
return
|
||||||
|
self._win.erase()
|
||||||
|
self._win.addnstr(text, min(len(text), self.getmaxyx()[1] - 1))
|
||||||
|
self._win.noutrefresh()
|
||||||
@@ -0,0 +1,51 @@
|
|||||||
|
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:
|
||||||
|
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))
|
||||||
@@ -0,0 +1,95 @@
|
|||||||
|
import curses
|
||||||
|
import curses.ascii
|
||||||
|
import typing as T
|
||||||
|
|
||||||
|
from pqcli.ui.curses.util import KEYS_CYCLE, KEYS_DOWN, KEYS_UP, Choice
|
||||||
|
from pqcli.ui.curses.widgets.focusable import focus_standout
|
||||||
|
|
||||||
|
from .base import Widget
|
||||||
|
|
||||||
|
|
||||||
|
class Menu(Widget):
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
header: str,
|
||||||
|
choices: T.List[Choice],
|
||||||
|
active_choice: int,
|
||||||
|
scr_height: int,
|
||||||
|
scr_width: int,
|
||||||
|
) -> None:
|
||||||
|
self._header_lines = header.split("\n")
|
||||||
|
self._choices = choices
|
||||||
|
self._scr_width = scr_width
|
||||||
|
self._scr_height = scr_height
|
||||||
|
|
||||||
|
all_lines = self._header_lines + sum(
|
||||||
|
[choice.desc.splitlines() for choice in choices], []
|
||||||
|
)
|
||||||
|
|
||||||
|
w = max(map(len, all_lines)) + 1
|
||||||
|
h = len(all_lines) + 1
|
||||||
|
self._pad: T.Optional[T.Any] = curses.newpad(h, w)
|
||||||
|
|
||||||
|
for y, line in enumerate(self._header_lines):
|
||||||
|
self._pad.move(y, 0)
|
||||||
|
self._pad.addstr(line)
|
||||||
|
|
||||||
|
self._active_choice = active_choice
|
||||||
|
|
||||||
|
def stop(self) -> None:
|
||||||
|
del self._pad
|
||||||
|
self._pad = None
|
||||||
|
|
||||||
|
def getmaxyx(self) -> T.Tuple[int, int]:
|
||||||
|
if not self._pad:
|
||||||
|
return (0, 0)
|
||||||
|
return self._pad.getmaxyx()
|
||||||
|
|
||||||
|
def next(self) -> None:
|
||||||
|
self._active_choice = min(
|
||||||
|
len(self._choices) - 1, self._active_choice + 1
|
||||||
|
)
|
||||||
|
|
||||||
|
def prev(self) -> None:
|
||||||
|
self._active_choice = max(0, self._active_choice - 1)
|
||||||
|
|
||||||
|
def cycle(self) -> None:
|
||||||
|
self._active_choice = (self._active_choice + 1) % len(self._choices)
|
||||||
|
|
||||||
|
def keypress(self, key: int) -> None:
|
||||||
|
for choice in self._choices:
|
||||||
|
if key in choice.keys:
|
||||||
|
choice.callback()
|
||||||
|
return
|
||||||
|
|
||||||
|
if key == curses.ascii.NL:
|
||||||
|
self._choices[self._active_choice].callback()
|
||||||
|
elif key in KEYS_DOWN:
|
||||||
|
self.next()
|
||||||
|
elif key in KEYS_UP:
|
||||||
|
self.prev()
|
||||||
|
elif key in KEYS_CYCLE:
|
||||||
|
self.cycle()
|
||||||
|
|
||||||
|
def render(self) -> None:
|
||||||
|
if not self._pad:
|
||||||
|
return
|
||||||
|
|
||||||
|
y = len(self._header_lines) + 1
|
||||||
|
for i, choice in enumerate(self._choices):
|
||||||
|
self._pad.move(y, 0)
|
||||||
|
with focus_standout(i == self._active_choice, self._pad):
|
||||||
|
lines = choice.desc.splitlines()
|
||||||
|
max_len = max(map(len, lines))
|
||||||
|
lines = [line.ljust(max_len) for line in lines]
|
||||||
|
self._pad.addstr("\n".join(lines))
|
||||||
|
y += len(choice.desc.splitlines())
|
||||||
|
|
||||||
|
self._pad.refresh(
|
||||||
|
0,
|
||||||
|
0,
|
||||||
|
(self._scr_height - self._pad.getmaxyx()[0]) // 2,
|
||||||
|
(self._scr_width - self._pad.getmaxyx()[1]) // 2,
|
||||||
|
self._scr_height - 1,
|
||||||
|
self._scr_width - 1,
|
||||||
|
)
|
||||||
@@ -0,0 +1,60 @@
|
|||||||
|
import curses
|
||||||
|
import datetime
|
||||||
|
import typing as T
|
||||||
|
|
||||||
|
from pqcli.lingo import format_timespan
|
||||||
|
from pqcli.ui.curses.colors import COLOR_PROGRESSBAR, has_colors
|
||||||
|
from pqcli.ui.curses.widgets.base import WindowWrapper
|
||||||
|
|
||||||
|
|
||||||
|
class ProgressBar(WindowWrapper):
|
||||||
|
def __init__(
|
||||||
|
self, parent: T.Any, h: int, w: int, y: int, x: int, show_time: bool
|
||||||
|
) -> None:
|
||||||
|
super().__init__(parent, h, w, y, x)
|
||||||
|
|
||||||
|
self._cur_pos = 0.0
|
||||||
|
self._max_pos = 1.0
|
||||||
|
|
||||||
|
self._show_time = show_time
|
||||||
|
self._last_tick: T.Optional[T.Tuple[datetime.datetime, float]] = None
|
||||||
|
|
||||||
|
@property
|
||||||
|
def time_left(self) -> T.Optional[datetime.timedelta]:
|
||||||
|
if self._last_tick is None:
|
||||||
|
return None
|
||||||
|
time_then, pos_then = self._last_tick
|
||||||
|
time_now, pos_now = datetime.datetime.now(), self._cur_pos
|
||||||
|
speed = (pos_now - pos_then) / (time_now - time_then).total_seconds()
|
||||||
|
if not speed:
|
||||||
|
return None
|
||||||
|
return datetime.timedelta(seconds=(self._max_pos - pos_now) / speed)
|
||||||
|
|
||||||
|
def set_position(self, cur_pos: float, max_pos: float) -> None:
|
||||||
|
if not self._win:
|
||||||
|
return
|
||||||
|
self._win.erase()
|
||||||
|
|
||||||
|
if self._last_tick is None or cur_pos == 0 or max_pos != self._max_pos:
|
||||||
|
self._last_tick = (datetime.datetime.now(), cur_pos)
|
||||||
|
self._cur_pos = cur_pos
|
||||||
|
self._max_pos = max_pos
|
||||||
|
|
||||||
|
text = f"{cur_pos / max_pos:.02%}"
|
||||||
|
if self.time_left and self._show_time:
|
||||||
|
text += f" ({format_timespan(self.time_left)})"
|
||||||
|
|
||||||
|
x = max(0, (self.getmaxyx()[1] - len(text)) // 2)
|
||||||
|
self._win.addnstr(0, x, text, min(len(text), self.getmaxyx()[1] - 1))
|
||||||
|
x = int(cur_pos * self.getmaxyx()[1] // max_pos)
|
||||||
|
if x > 0:
|
||||||
|
self._win.chgat(
|
||||||
|
0,
|
||||||
|
0,
|
||||||
|
curses.color_pair(COLOR_PROGRESSBAR)
|
||||||
|
if has_colors()
|
||||||
|
else curses.A_REVERSE,
|
||||||
|
)
|
||||||
|
if x < self.getmaxyx()[1]:
|
||||||
|
self._win.chgat(0, x, curses.A_NORMAL)
|
||||||
|
self._win.noutrefresh()
|
||||||
@@ -0,0 +1,114 @@
|
|||||||
|
import curses
|
||||||
|
import typing as T
|
||||||
|
|
||||||
|
from pqcli.ui.curses.colors import (
|
||||||
|
COLOR_SCROLLBAR_THUMB,
|
||||||
|
COLOR_SCROLLBAR_TRACK,
|
||||||
|
has_colors,
|
||||||
|
)
|
||||||
|
|
||||||
|
from .base import WindowWrapper
|
||||||
|
|
||||||
|
|
||||||
|
class Scrollable(WindowWrapper):
|
||||||
|
def __init__(self, parent: T.Any, h: int, w: int, y: int, x: int) -> None:
|
||||||
|
super().__init__(parent, h, w, y, x)
|
||||||
|
self._scroll_pos = 0
|
||||||
|
self._pad: T.Optional[T.Any] = curses.newpad(1, 1)
|
||||||
|
self._items: T.List[T.Any] = []
|
||||||
|
|
||||||
|
def clear(self) -> None:
|
||||||
|
self._items = []
|
||||||
|
self.scroll_to_item(0)
|
||||||
|
|
||||||
|
def scroll_page_up(self) -> None:
|
||||||
|
self.scroll_to_item(max(0, self._scroll_pos - self.getmaxyx()[0]))
|
||||||
|
|
||||||
|
def scroll_page_down(self) -> None:
|
||||||
|
self.scroll_to_item(
|
||||||
|
min(
|
||||||
|
len(self._items) - 1,
|
||||||
|
self._scroll_pos + self.getmaxyx()[0] * 2 - 1,
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
def scroll_to_item(self, y: int) -> None:
|
||||||
|
if not self._items or not self._win:
|
||||||
|
return
|
||||||
|
if y < 0:
|
||||||
|
y %= len(self._items)
|
||||||
|
|
||||||
|
h = self.getmaxyx()[0]
|
||||||
|
delta = 1 if self._scroll_pos < y else -1
|
||||||
|
while not self._scroll_pos <= y < self._scroll_pos + h:
|
||||||
|
self._scroll_pos += delta
|
||||||
|
|
||||||
|
def stop(self) -> None:
|
||||||
|
super().stop()
|
||||||
|
del self._pad
|
||||||
|
self._pad = None
|
||||||
|
|
||||||
|
def __len__(self) -> int:
|
||||||
|
return len(self._items)
|
||||||
|
|
||||||
|
def render(self) -> None:
|
||||||
|
if not self._pad or not self._win:
|
||||||
|
return
|
||||||
|
|
||||||
|
if self._items:
|
||||||
|
self._pad.resize(
|
||||||
|
max(1, len(self._items)), max(1, self.getmaxyx()[1] + 1)
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
self._pad.resize(1, 1)
|
||||||
|
|
||||||
|
self._win.erase()
|
||||||
|
self._win.noutrefresh()
|
||||||
|
|
||||||
|
self._pad.erase()
|
||||||
|
h, w = self.getmaxyx()
|
||||||
|
y, x = self.getbegyx()
|
||||||
|
|
||||||
|
try:
|
||||||
|
if self._scroll_pos > 0 or self._scroll_pos + h < len(self._items):
|
||||||
|
y1 = self._scroll_pos
|
||||||
|
y2 = self._scroll_pos + h
|
||||||
|
thumb_y1 = int(y1 * h // len(self._items))
|
||||||
|
thumb_y2 = int(y2 * h // len(self._items))
|
||||||
|
|
||||||
|
attr_thumb = (
|
||||||
|
curses.color_pair(COLOR_SCROLLBAR_THUMB)
|
||||||
|
if has_colors()
|
||||||
|
else curses.A_REVERSE
|
||||||
|
)
|
||||||
|
attr_track = (
|
||||||
|
curses.color_pair(COLOR_SCROLLBAR_TRACK)
|
||||||
|
if has_colors()
|
||||||
|
else curses.A_NORMAL
|
||||||
|
)
|
||||||
|
|
||||||
|
for win_y in range(h):
|
||||||
|
self._win.chgat(
|
||||||
|
win_y,
|
||||||
|
w - 1,
|
||||||
|
attr_thumb
|
||||||
|
if thumb_y1 <= win_y <= thumb_y2
|
||||||
|
else attr_track,
|
||||||
|
)
|
||||||
|
self._win.noutrefresh()
|
||||||
|
|
||||||
|
self._render_impl(h, w - 1)
|
||||||
|
self._pad.noutrefresh(
|
||||||
|
self._scroll_pos, 0, y, x, y + h - 1, x + w - 2
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
self._render_impl(h, w)
|
||||||
|
self._pad.noutrefresh(
|
||||||
|
self._scroll_pos, 0, y, x, y + h - 1, x + w - 1
|
||||||
|
)
|
||||||
|
|
||||||
|
except curses.error:
|
||||||
|
pass
|
||||||
|
|
||||||
|
def _render_impl(self, h: int, w: int) -> None:
|
||||||
|
raise NotImplementedError("not implemented")
|
||||||
@@ -1,31 +0,0 @@
|
|||||||
import typing as T
|
|
||||||
|
|
||||||
import urwid
|
|
||||||
|
|
||||||
_MAGIC = 999 # where to display the cursor (high value = don't display it)
|
|
||||||
|
|
||||||
|
|
||||||
class CustomButton(urwid.Button):
|
|
||||||
def __init__(
|
|
||||||
self,
|
|
||||||
label: str,
|
|
||||||
*args: T.Any,
|
|
||||||
hint: T.Optional[str] = None,
|
|
||||||
**kwargs: T.Any,
|
|
||||||
) -> None:
|
|
||||||
super().__init__("", *args, **kwargs)
|
|
||||||
self._w = urwid.AttrMap(
|
|
||||||
urwid.Columns(
|
|
||||||
[
|
|
||||||
(urwid.PACK, urwid.Text("< ")),
|
|
||||||
urwid.SelectableIcon([label], _MAGIC),
|
|
||||||
(
|
|
||||||
urwid.PACK,
|
|
||||||
urwid.Text(f"[{hint}]" if hint else "", align="right"),
|
|
||||||
),
|
|
||||||
(urwid.PACK, urwid.Text(" >")),
|
|
||||||
]
|
|
||||||
),
|
|
||||||
"button",
|
|
||||||
"button-focus",
|
|
||||||
)
|
|
||||||
@@ -1,18 +0,0 @@
|
|||||||
import typing as T
|
|
||||||
|
|
||||||
import urwid
|
|
||||||
|
|
||||||
|
|
||||||
class CustomLineBox(urwid.AttrMap):
|
|
||||||
def __init__(
|
|
||||||
self, widget: urwid.Widget, *args: T.Any, **kwargs: T.Any
|
|
||||||
) -> None:
|
|
||||||
super().__init__(
|
|
||||||
urwid.LineBox(
|
|
||||||
urwid.AttrMap(widget, "linebox-content", "linebox-content"),
|
|
||||||
*args,
|
|
||||||
**kwargs,
|
|
||||||
),
|
|
||||||
"linebox",
|
|
||||||
"linebox-focus",
|
|
||||||
)
|
|
||||||
@@ -1,119 +0,0 @@
|
|||||||
import datetime
|
|
||||||
import typing as T
|
|
||||||
|
|
||||||
import urwid
|
|
||||||
|
|
||||||
|
|
||||||
def format_float(num: float) -> str:
|
|
||||||
ret = f"{num:.01f}"
|
|
||||||
if ret.endswith("0"):
|
|
||||||
ret = ret[:-2]
|
|
||||||
return ret
|
|
||||||
|
|
||||||
|
|
||||||
def format_timespan(timespan: datetime.timedelta) -> str:
|
|
||||||
num = timespan.total_seconds()
|
|
||||||
if num < 60.0:
|
|
||||||
return f"~{int(num)}s"
|
|
||||||
num /= 60
|
|
||||||
if num < 60.0:
|
|
||||||
return f"~{int(num)}m"
|
|
||||||
num /= 60
|
|
||||||
if num < 24.0:
|
|
||||||
return f"~{format_float(num)}h"
|
|
||||||
num /= 24
|
|
||||||
return f"~{format_float(num)}d"
|
|
||||||
|
|
||||||
|
|
||||||
class CustomProgressBar(urwid.Widget):
|
|
||||||
_sizing = frozenset(["flow"])
|
|
||||||
text_align = "center"
|
|
||||||
eighths = " ▏▎▍▌▋▊▉"
|
|
||||||
|
|
||||||
def __init__(self, show_time: bool) -> None:
|
|
||||||
self.show_time = show_time
|
|
||||||
self.position = 0.0
|
|
||||||
self.max_ = 1.0
|
|
||||||
self.normal = "progressbar-normal"
|
|
||||||
self.complete = "progressbar-done"
|
|
||||||
self.smooth = "progressbar-smooth"
|
|
||||||
self.last_tick: T.Optional[datetime.datetime, float] = None
|
|
||||||
|
|
||||||
@property
|
|
||||||
def time_left(self) -> T.Optional[datetime.timedelta]:
|
|
||||||
if self.last_tick is None:
|
|
||||||
return None
|
|
||||||
time_now = datetime.datetime.now()
|
|
||||||
time_then = self.last_tick[0]
|
|
||||||
pos_now = self.position
|
|
||||||
pos_then = self.last_tick[1]
|
|
||||||
speed = (pos_now - pos_then) / (time_now - time_then).total_seconds()
|
|
||||||
if not speed:
|
|
||||||
return None
|
|
||||||
return datetime.timedelta(seconds=(self.max_ - pos_now) / speed)
|
|
||||||
|
|
||||||
def set_completion(self, value: float) -> None:
|
|
||||||
if self.last_tick is None or value == 0:
|
|
||||||
self.last_tick = (datetime.datetime.now(), value)
|
|
||||||
self.position = value
|
|
||||||
self._invalidate()
|
|
||||||
|
|
||||||
def set_max(self, value: T.Union[int, float]) -> None:
|
|
||||||
if self.max_ == value:
|
|
||||||
return
|
|
||||||
self.last_tick = None
|
|
||||||
self.max_ = value
|
|
||||||
self._invalidate()
|
|
||||||
|
|
||||||
def reset(self, cur: float, max_: T.Union[int, float]) -> None:
|
|
||||||
if self.last_tick is None or cur == 0 or max_ != self.max_:
|
|
||||||
self.last_tick = (datetime.datetime.now(), cur)
|
|
||||||
self.position = cur
|
|
||||||
self.max_ = max_
|
|
||||||
self._invalidate()
|
|
||||||
|
|
||||||
def rows(self, size: T.Tuple[int, int], focus: bool = False) -> int:
|
|
||||||
return 1
|
|
||||||
|
|
||||||
def get_text(self) -> str:
|
|
||||||
percent = min(100.0, max(0.0, self.position * 100.0 / self.max_))
|
|
||||||
ret = f"{percent:.02f}%"
|
|
||||||
if self.time_left and self.show_time:
|
|
||||||
ret += f" ({format_timespan(self.time_left)})"
|
|
||||||
return ret
|
|
||||||
|
|
||||||
# workaround for https://github.com/urwid/urwid/issues/317
|
|
||||||
def render(self, size: T.Tuple[int, int], focus: bool = False) -> T.Any:
|
|
||||||
(maxcol,) = size
|
|
||||||
txt = urwid.Text(self.get_text(), self.text_align, "clip")
|
|
||||||
c = txt.render((maxcol,))
|
|
||||||
t = c._text[0]
|
|
||||||
|
|
||||||
cf = self.position * maxcol / self.max_
|
|
||||||
ccol_dirty = int(cf)
|
|
||||||
ccol = len(
|
|
||||||
c._text[0][:ccol_dirty].decode("utf-8", "ignore").encode("utf-8")
|
|
||||||
)
|
|
||||||
|
|
||||||
cs = 0
|
|
||||||
cs = int((cf - ccol) * 8)
|
|
||||||
|
|
||||||
if ccol < 0 or (ccol == 0 and cs == 0):
|
|
||||||
c._attr = [[(self.normal, maxcol)]]
|
|
||||||
elif ccol >= maxcol:
|
|
||||||
c._attr = [[(self.complete, maxcol)]]
|
|
||||||
elif cs and c._text[0][ccol] == 32:
|
|
||||||
t = c._text[0]
|
|
||||||
cenc = self.eighths[cs].encode("utf-8")
|
|
||||||
c._text[0] = t[:ccol] + cenc + t[ccol + 1 :]
|
|
||||||
a = []
|
|
||||||
if ccol > 0:
|
|
||||||
a.append((self.complete, ccol))
|
|
||||||
a.append((self.smooth, len(cenc)))
|
|
||||||
if maxcol - ccol - 1 > 0:
|
|
||||||
a.append((self.normal, maxcol - ccol - 1))
|
|
||||||
c._attr = [a]
|
|
||||||
c._cs = [[(None, len(c._text[0]))]]
|
|
||||||
else:
|
|
||||||
c._attr = [[(self.complete, ccol), (self.normal, maxcol - ccol)]]
|
|
||||||
return c
|
|
||||||
@@ -1,64 +0,0 @@
|
|||||||
import typing as T
|
|
||||||
|
|
||||||
import urwid
|
|
||||||
|
|
||||||
from pqcli.ui.layout import NColumns, NPile
|
|
||||||
|
|
||||||
|
|
||||||
class DataTable(NPile):
|
|
||||||
def __init__(self, columns: T.Any) -> None:
|
|
||||||
self.column_count = len(columns)
|
|
||||||
self._columns = columns
|
|
||||||
self._rows: T.List[T.List[urwid.Widget]] = []
|
|
||||||
super().__init__([])
|
|
||||||
|
|
||||||
@property
|
|
||||||
def row_count(self) -> int:
|
|
||||||
return len(self._rows)
|
|
||||||
|
|
||||||
@property
|
|
||||||
def data_rows(self) -> T.Iterable[T.List[urwid.Widget]]:
|
|
||||||
# "rows" was already used by urwid
|
|
||||||
return self._rows
|
|
||||||
|
|
||||||
def add_row(self, *widgets: T.List[urwid.Widget]) -> None:
|
|
||||||
if len(widgets) != self.column_count:
|
|
||||||
raise VauleError(f"Expected {self.column_count} widgets")
|
|
||||||
|
|
||||||
self.contents.append(
|
|
||||||
(
|
|
||||||
NColumns(
|
|
||||||
list((*x, y) for x, y in zip(self._columns, widgets))
|
|
||||||
),
|
|
||||||
(urwid.PACK, 1),
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
self.resize_columns()
|
|
||||||
self._rows.append(widgets)
|
|
||||||
|
|
||||||
def delete_row(self, idx: int) -> None:
|
|
||||||
del self.contents[idx]
|
|
||||||
del self._rows[idx]
|
|
||||||
self.resize_columns()
|
|
||||||
|
|
||||||
def delete_rows(self, start_idx: int, end_idx: int) -> None:
|
|
||||||
del self.contents[start_idx:end_idx]
|
|
||||||
del self._rows[start_idx:end_idx]
|
|
||||||
self.resize_columns()
|
|
||||||
|
|
||||||
def resize_columns(self) -> None:
|
|
||||||
if not self._rows:
|
|
||||||
return
|
|
||||||
|
|
||||||
for i, sizing in enumerate(self._columns):
|
|
||||||
if isinstance(sizing, tuple) and sizing[0] == urwid.PACK:
|
|
||||||
width = max(
|
|
||||||
row_widgets[i].pack()[0] for row_widgets in self._rows
|
|
||||||
)
|
|
||||||
for item in self.contents:
|
|
||||||
columns = item[0]
|
|
||||||
columns.contents[i] = (
|
|
||||||
columns.contents[i][0],
|
|
||||||
(urwid.GIVEN, width, None),
|
|
||||||
)
|
|
||||||
@@ -1,34 +0,0 @@
|
|||||||
import urwid
|
|
||||||
|
|
||||||
from pqcli.ui.layout import NPile
|
|
||||||
|
|
||||||
|
|
||||||
class DoubleLineBox(urwid.AttrMap):
|
|
||||||
def __init__(
|
|
||||||
self,
|
|
||||||
top_widget: urwid.Widget,
|
|
||||||
bottom_widget: urwid.Widget,
|
|
||||||
top_title: str = "",
|
|
||||||
bottom_title: str = "",
|
|
||||||
) -> None:
|
|
||||||
self.top_line_box = urwid.LineBox(
|
|
||||||
urwid.AttrMap(top_widget, "linebox-content", "linebox-content"),
|
|
||||||
title=top_title,
|
|
||||||
bline="",
|
|
||||||
)
|
|
||||||
|
|
||||||
self.bottom_line_box = urwid.LineBox(
|
|
||||||
urwid.AttrMap(bottom_widget, "linebox-content", "linebox-content"),
|
|
||||||
title=bottom_title,
|
|
||||||
tlcorner="├",
|
|
||||||
trcorner="┤",
|
|
||||||
)
|
|
||||||
|
|
||||||
super().__init__(
|
|
||||||
NPile([self.top_line_box, (urwid.PACK, self.bottom_line_box)]),
|
|
||||||
"linebox",
|
|
||||||
"linebox-focus",
|
|
||||||
)
|
|
||||||
|
|
||||||
def set_bottom_title(self, text: str) -> None:
|
|
||||||
self.bottom_line_box.set_title(text)
|
|
||||||
@@ -1,446 +0,0 @@
|
|||||||
import argparse
|
|
||||||
import contextlib
|
|
||||||
import datetime
|
|
||||||
import typing as T
|
|
||||||
|
|
||||||
import urwid
|
|
||||||
|
|
||||||
from pqcli.config import EquipmentType, StatType
|
|
||||||
from pqcli.lingo import act_name, to_roman
|
|
||||||
from pqcli.mechanic import (
|
|
||||||
InventoryItem,
|
|
||||||
Player,
|
|
||||||
SignalMixin,
|
|
||||||
Simulation,
|
|
||||||
Spell,
|
|
||||||
)
|
|
||||||
from pqcli.ui.custom_line_box import CustomLineBox
|
|
||||||
from pqcli.ui.custom_progress_bar import CustomProgressBar
|
|
||||||
from pqcli.ui.data_table import DataTable
|
|
||||||
from pqcli.ui.double_line_box import DoubleLineBox
|
|
||||||
from pqcli.ui.layout import NColumns, NPile
|
|
||||||
from pqcli.ui.read_only_check_box import ReadOnlyCheckBox
|
|
||||||
from pqcli.ui.scrollable import Scrollable, ScrollBar
|
|
||||||
|
|
||||||
|
|
||||||
class CharacterSheetView(DoubleLineBox):
|
|
||||||
def __init__(self, player: Player) -> None:
|
|
||||||
self.player = player
|
|
||||||
|
|
||||||
self.level_text = urwid.Text("")
|
|
||||||
self.stat_texts = {stat: urwid.Text("") for stat in list(StatType)}
|
|
||||||
self.exp_bar = CustomProgressBar(show_time=True)
|
|
||||||
|
|
||||||
data_table = DataTable(columns=[(urwid.WEIGHT, 1), (urwid.WEIGHT, 2)])
|
|
||||||
data_table.add_row(urwid.Text("Name"), urwid.Text(player.name))
|
|
||||||
data_table.add_row(urwid.Text("Race"), urwid.Text(player.race.name))
|
|
||||||
data_table.add_row(urwid.Text("Class"), urwid.Text(player.class_.name))
|
|
||||||
data_table.add_row(urwid.Text("Level"), self.level_text)
|
|
||||||
data_table.add_row(urwid.Divider(), urwid.Divider())
|
|
||||||
for stat in StatType:
|
|
||||||
data_table.add_row(urwid.Text(stat.value), self.stat_texts[stat])
|
|
||||||
|
|
||||||
self.player.connect("level_up", self.sync_level)
|
|
||||||
self.player.stats.connect("change", self.sync_stats)
|
|
||||||
self.player.exp_bar.connect("change", self.sync_exp)
|
|
||||||
|
|
||||||
super().__init__(
|
|
||||||
top_widget=ScrollBar(Scrollable(data_table)),
|
|
||||||
bottom_widget=self.exp_bar,
|
|
||||||
top_title="Character Sheet",
|
|
||||||
bottom_title="Experience",
|
|
||||||
)
|
|
||||||
self.sync()
|
|
||||||
|
|
||||||
def sync(self) -> None:
|
|
||||||
self.sync_level()
|
|
||||||
self.sync_stats()
|
|
||||||
self.sync_exp()
|
|
||||||
|
|
||||||
def sync_level(self) -> None:
|
|
||||||
self.level_text.set_text(str(self.player.level))
|
|
||||||
|
|
||||||
def sync_stats(self) -> None:
|
|
||||||
for stat in StatType:
|
|
||||||
self.stat_texts[stat].set_text(str(self.player.stats[stat]))
|
|
||||||
|
|
||||||
def sync_exp(self) -> None:
|
|
||||||
cur = self.player.exp_bar.position
|
|
||||||
max_ = self.player.exp_bar.max_
|
|
||||||
self.exp_bar.reset(cur, max_)
|
|
||||||
self.set_bottom_title(f"Experience ({max_-cur:.0f} XP to go)")
|
|
||||||
|
|
||||||
|
|
||||||
class SpellBookView(CustomLineBox):
|
|
||||||
def __init__(self, player: Player) -> None:
|
|
||||||
self.player = player
|
|
||||||
|
|
||||||
self.data_table = DataTable(columns=[(urwid.WEIGHT, 3), (urwid.PACK,)])
|
|
||||||
self.scrollable = Scrollable(self.data_table)
|
|
||||||
|
|
||||||
self.player.spell_book.connect("add", self.sync_spell_add)
|
|
||||||
self.player.spell_book.connect("change", self.sync_spell_change)
|
|
||||||
|
|
||||||
super().__init__(ScrollBar(self.scrollable), title="Spell Book")
|
|
||||||
self.sync()
|
|
||||||
|
|
||||||
def sync(self) -> None:
|
|
||||||
self.data_table.delete_rows(0, len(self.data_table.data_rows))
|
|
||||||
for spell in self.player.spell_book:
|
|
||||||
self.sync_spell_add(spell)
|
|
||||||
|
|
||||||
def sync_spell_add(self, spell: Spell) -> None:
|
|
||||||
self.data_table.add_row(
|
|
||||||
urwid.Text(spell.name), urwid.Text(to_roman(spell.level))
|
|
||||||
)
|
|
||||||
self.scrollable.set_scrollpos(len(self.data_table.data_rows) - 1)
|
|
||||||
|
|
||||||
def sync_spell_change(self, spell: Spell) -> None:
|
|
||||||
for row_widgets in self.data_table.data_rows:
|
|
||||||
if row_widgets[0].text == spell.name:
|
|
||||||
row_widgets[1].set_text(to_roman(spell.level))
|
|
||||||
self.data_table.resize_columns()
|
|
||||||
|
|
||||||
|
|
||||||
class EquipmentView(CustomLineBox):
|
|
||||||
def __init__(self, player: Player) -> None:
|
|
||||||
self.player = player
|
|
||||||
|
|
||||||
self.value_texts = {
|
|
||||||
equipment_type: urwid.Text("") for equipment_type in EquipmentType
|
|
||||||
}
|
|
||||||
data_table = DataTable(columns=[(urwid.WEIGHT, 1), (urwid.WEIGHT, 3)])
|
|
||||||
for equipment_type in EquipmentType:
|
|
||||||
data_table.add_row(
|
|
||||||
urwid.Text(equipment_type.value),
|
|
||||||
self.value_texts[equipment_type],
|
|
||||||
)
|
|
||||||
|
|
||||||
self.player.equipment.connect("change", self.sync_equipment_change)
|
|
||||||
|
|
||||||
super().__init__(ScrollBar(Scrollable(data_table)), title="Equipment")
|
|
||||||
self.sync()
|
|
||||||
|
|
||||||
def sync(self) -> None:
|
|
||||||
for equipment_type in EquipmentType:
|
|
||||||
self.sync_equipment_change(
|
|
||||||
equipment_type, self.player.equipment[equipment_type]
|
|
||||||
)
|
|
||||||
|
|
||||||
def sync_equipment_change(
|
|
||||||
self, equipment_type: EquipmentType, item_name: T.Optional[str]
|
|
||||||
) -> None:
|
|
||||||
self.value_texts[equipment_type].set_text(item_name or "")
|
|
||||||
|
|
||||||
|
|
||||||
class InventoryView(DoubleLineBox):
|
|
||||||
def __init__(self, player: Player) -> None:
|
|
||||||
self.player = player
|
|
||||||
|
|
||||||
self.gold_text = urwid.Text("", align=urwid.RIGHT)
|
|
||||||
self.data_table = DataTable(columns=[(urwid.WEIGHT, 3), (urwid.PACK,)])
|
|
||||||
self.data_table.add_row(urwid.Text("Gold"), self.gold_text)
|
|
||||||
self.scrollable = Scrollable(self.data_table)
|
|
||||||
self.encumbrance_bar = CustomProgressBar(show_time=False)
|
|
||||||
|
|
||||||
self.player.connect("new_task", self.on_new_task)
|
|
||||||
self.player.inventory.connect("gold_change", self.sync_gold)
|
|
||||||
self.player.inventory.connect("item_add", self.sync_item_add)
|
|
||||||
self.player.inventory.connect("item_del", self.sync_item_del)
|
|
||||||
self.player.inventory.connect("item_change", self.sync_item_change)
|
|
||||||
self.player.inventory.encum_bar.connect(
|
|
||||||
"change", self.sync_encumbrance_position
|
|
||||||
)
|
|
||||||
|
|
||||||
super().__init__(
|
|
||||||
top_widget=ScrollBar(self.scrollable),
|
|
||||||
top_title="Inventory",
|
|
||||||
bottom_widget=self.encumbrance_bar,
|
|
||||||
bottom_title="Encumbrance",
|
|
||||||
)
|
|
||||||
self.sync()
|
|
||||||
|
|
||||||
def sync(self) -> None:
|
|
||||||
self.sync_gold()
|
|
||||||
self.sync_items()
|
|
||||||
self.sync_encumbrance_position()
|
|
||||||
|
|
||||||
def sync_gold(self) -> None:
|
|
||||||
self.gold_text.set_text(str(self.player.inventory.gold))
|
|
||||||
|
|
||||||
def sync_items(self) -> None:
|
|
||||||
self.data_table.delete_rows(1, len(self.data_table.data_rows) - 1)
|
|
||||||
for item in self.player.inventory:
|
|
||||||
self.sync_item_add(item)
|
|
||||||
|
|
||||||
def sync_encumbrance_position(self) -> None:
|
|
||||||
cur = self.player.inventory.encum_bar.position
|
|
||||||
max_ = self.player.inventory.encum_bar.max_
|
|
||||||
self.encumbrance_bar.reset(cur, max_)
|
|
||||||
self.set_bottom_title(f"Encumbrance ({cur:.0f}/{max_} cubits)")
|
|
||||||
|
|
||||||
def sync_item_add(self, item: InventoryItem) -> None:
|
|
||||||
self.data_table.add_row(
|
|
||||||
urwid.Text(item.name),
|
|
||||||
urwid.Text(str(item.quantity), align=urwid.RIGHT),
|
|
||||||
)
|
|
||||||
self.scrollable.set_scrollpos(self.scrollable.rows_max() - 1)
|
|
||||||
|
|
||||||
def sync_item_del(self, item: InventoryItem) -> None:
|
|
||||||
idx = self.get_item_idx_by_name(item.name)
|
|
||||||
if idx is not None:
|
|
||||||
self.data_table.delete_row(idx)
|
|
||||||
|
|
||||||
def sync_item_change(self, item: InventoryItem) -> None:
|
|
||||||
idx = self.get_item_idx_by_name(item.name)
|
|
||||||
if idx is not None:
|
|
||||||
self.data_table.data_rows[idx][1].set_text(str(item.quantity))
|
|
||||||
|
|
||||||
def get_item_idx_by_name(self, name: str) -> T.Optional[int]:
|
|
||||||
for i, row_widgets in enumerate(self.data_table.data_rows):
|
|
||||||
if row_widgets[0].text == name:
|
|
||||||
return i
|
|
||||||
return None
|
|
||||||
|
|
||||||
def on_new_task(self) -> None:
|
|
||||||
description = self.player.task.description if self.player.task else "?"
|
|
||||||
if description.lower().startswith("sell"):
|
|
||||||
self.scrollable.set_scrollpos(0)
|
|
||||||
|
|
||||||
|
|
||||||
class PlotView(DoubleLineBox):
|
|
||||||
cutoff = 100
|
|
||||||
|
|
||||||
def __init__(self, player: Player) -> None:
|
|
||||||
self.player = player
|
|
||||||
|
|
||||||
self.data_table = DataTable(columns=[(urwid.WEIGHT, 1)])
|
|
||||||
self.scrollable = Scrollable(self.data_table)
|
|
||||||
self.plot_bar = CustomProgressBar(show_time=True)
|
|
||||||
|
|
||||||
self.player.quest_book.connect("start_act", self.sync_act_add)
|
|
||||||
self.player.quest_book.plot_bar.connect("change", self.sync_position)
|
|
||||||
|
|
||||||
super().__init__(
|
|
||||||
top_widget=ScrollBar(self.scrollable),
|
|
||||||
top_title="Plot Development",
|
|
||||||
bottom_widget=self.plot_bar,
|
|
||||||
)
|
|
||||||
self.sync()
|
|
||||||
|
|
||||||
def sync(self) -> None:
|
|
||||||
self.sync_acts()
|
|
||||||
self.sync_position()
|
|
||||||
|
|
||||||
def sync_acts(self) -> None:
|
|
||||||
self.data_table.delete_rows(0, len(self.data_table.data_rows))
|
|
||||||
for act_number in range(
|
|
||||||
max(0, self.player.quest_book.act - self.cutoff),
|
|
||||||
self.player.quest_book.act + 1,
|
|
||||||
):
|
|
||||||
self.sync_act_add(act_number)
|
|
||||||
|
|
||||||
def sync_act_add(self, act_number: int) -> None:
|
|
||||||
self.data_table.delete_rows(
|
|
||||||
0, max(0, self.data_table.row_count - self.cutoff)
|
|
||||||
)
|
|
||||||
if self.data_table.data_rows:
|
|
||||||
self.data_table.data_rows[-1][0].set_state(True)
|
|
||||||
self.data_table.add_row(
|
|
||||||
ReadOnlyCheckBox(act_name(act_number), state=False)
|
|
||||||
)
|
|
||||||
self.scrollable.set_scrollpos(self.scrollable.rows_max() - 1)
|
|
||||||
|
|
||||||
def sync_position(self) -> None:
|
|
||||||
self.plot_bar.reset(
|
|
||||||
self.player.quest_book.plot_bar.position,
|
|
||||||
self.player.quest_book.plot_bar.max_,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
class QuestBookView(DoubleLineBox):
|
|
||||||
cutoff = 100
|
|
||||||
|
|
||||||
def __init__(self, player: Player) -> None:
|
|
||||||
self.player = player
|
|
||||||
|
|
||||||
self.data_table = DataTable(columns=[(urwid.WEIGHT, 1)])
|
|
||||||
self.quest_bar = CustomProgressBar(show_time=True)
|
|
||||||
self.scrollable = Scrollable(self.data_table)
|
|
||||||
|
|
||||||
self.player.quest_book.connect("start_quest", self.sync_quest_add)
|
|
||||||
self.player.quest_book.quest_bar.connect("change", self.sync_position)
|
|
||||||
|
|
||||||
super().__init__(
|
|
||||||
top_widget=ScrollBar(self.scrollable),
|
|
||||||
top_title="Quests",
|
|
||||||
bottom_widget=self.quest_bar,
|
|
||||||
)
|
|
||||||
self.sync()
|
|
||||||
|
|
||||||
def sync(self) -> None:
|
|
||||||
self.sync_quests()
|
|
||||||
self.sync_position()
|
|
||||||
|
|
||||||
def sync_quests(self) -> None:
|
|
||||||
for quest_name in self.player.quest_book.quests[-self.cutoff :]:
|
|
||||||
self.sync_quest_add(quest_name)
|
|
||||||
|
|
||||||
def sync_quest_add(self, quest_name: str) -> None:
|
|
||||||
self.data_table.delete_rows(
|
|
||||||
0, max(0, self.data_table.row_count - self.cutoff)
|
|
||||||
)
|
|
||||||
if self.data_table.data_rows:
|
|
||||||
self.data_table.data_rows[-1][0].set_state(True)
|
|
||||||
self.data_table.add_row(ReadOnlyCheckBox(quest_name, state=False))
|
|
||||||
self.scrollable.set_scrollpos(self.scrollable.rows_max() - 1)
|
|
||||||
|
|
||||||
def sync_position(self) -> None:
|
|
||||||
self.quest_bar.reset(
|
|
||||||
self.player.quest_book.quest_bar.position,
|
|
||||||
self.player.quest_book.quest_bar.max_,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
class TaskView(NPile):
|
|
||||||
def __init__(self, player: Player) -> None:
|
|
||||||
self.player = player
|
|
||||||
|
|
||||||
self.current_task_text = urwid.Text("...")
|
|
||||||
self.current_task_bar = CustomProgressBar(show_time=False)
|
|
||||||
|
|
||||||
self.player.connect("new_task", self.sync_task_name)
|
|
||||||
self.player.task_bar.connect("change", self.sync_position)
|
|
||||||
|
|
||||||
super().__init__([self.current_task_text, self.current_task_bar])
|
|
||||||
self.sync()
|
|
||||||
|
|
||||||
def sync(self) -> None:
|
|
||||||
self.sync_task_name()
|
|
||||||
self.sync_position()
|
|
||||||
|
|
||||||
def sync_task_name(self) -> None:
|
|
||||||
self.current_task_text.set_text(
|
|
||||||
f"{self.player.task.description}..." if self.player.task else "?"
|
|
||||||
)
|
|
||||||
|
|
||||||
def sync_position(self) -> None:
|
|
||||||
self.current_task_bar.reset(
|
|
||||||
self.player.task_bar.position, self.player.task_bar.max_
|
|
||||||
)
|
|
||||||
|
|
||||||
def pack(self, size: T.Tuple[int, int]) -> T.Tuple[int, int]:
|
|
||||||
return (size[0], 2)
|
|
||||||
|
|
||||||
|
|
||||||
class GameView(NPile):
|
|
||||||
signals = ["cancel"]
|
|
||||||
|
|
||||||
def __init__(
|
|
||||||
self, loop: urwid.MainLoop, player: Player, args: argparse.Namespace
|
|
||||||
) -> None:
|
|
||||||
self.loop = loop
|
|
||||||
self.player = player
|
|
||||||
self.args = args
|
|
||||||
self.simulation = Simulation(player)
|
|
||||||
self.last_tick = datetime.datetime.now()
|
|
||||||
|
|
||||||
self.character_sheet_view = CharacterSheetView(player)
|
|
||||||
self.spell_book_view = SpellBookView(player)
|
|
||||||
self.equipment_view = EquipmentView(player)
|
|
||||||
self.inventory_view = InventoryView(player)
|
|
||||||
self.plot_view = PlotView(player)
|
|
||||||
self.quest_book_view = QuestBookView(player)
|
|
||||||
self.task_view = TaskView(player)
|
|
||||||
|
|
||||||
self.columns = NColumns(
|
|
||||||
[
|
|
||||||
(
|
|
||||||
urwid.WEIGHT,
|
|
||||||
4,
|
|
||||||
NPile(
|
|
||||||
[(19, self.character_sheet_view), self.spell_book_view]
|
|
||||||
),
|
|
||||||
),
|
|
||||||
(
|
|
||||||
urwid.WEIGHT,
|
|
||||||
5,
|
|
||||||
NPile([(15, self.equipment_view), self.inventory_view]),
|
|
||||||
),
|
|
||||||
(
|
|
||||||
urwid.WEIGHT,
|
|
||||||
4,
|
|
||||||
NPile([(15, self.plot_view), self.quest_book_view]),
|
|
||||||
),
|
|
||||||
]
|
|
||||||
)
|
|
||||||
|
|
||||||
super().__init__(
|
|
||||||
[self.columns, (urwid.PACK, self.task_view)], outermost=True
|
|
||||||
)
|
|
||||||
|
|
||||||
def cancel(self) -> None:
|
|
||||||
self._emit("cancel")
|
|
||||||
|
|
||||||
def tick(self) -> None:
|
|
||||||
now = datetime.datetime.now()
|
|
||||||
elapsed = (now - self.last_tick).total_seconds()
|
|
||||||
self.simulation.tick(elapsed * 1000)
|
|
||||||
self.last_tick = datetime.datetime.now()
|
|
||||||
|
|
||||||
def keypress(self, size: T.Tuple[int, int], key: str) -> bool:
|
|
||||||
if key == "esc":
|
|
||||||
self.cancel()
|
|
||||||
return None
|
|
||||||
|
|
||||||
if self._command_map[key] == urwid.CURSOR_RIGHT:
|
|
||||||
old_focus_position = self.columns.focus.focus_position
|
|
||||||
self.columns.focus_position = min(
|
|
||||||
self.columns.focus_position + 1, len(self.columns.contents) - 1
|
|
||||||
)
|
|
||||||
self.columns.focus.focus_position = old_focus_position
|
|
||||||
return None
|
|
||||||
|
|
||||||
if self._command_map[key] == urwid.CURSOR_LEFT:
|
|
||||||
old_focus_position = self.columns.focus.focus_position
|
|
||||||
self.columns.focus_position = max(
|
|
||||||
self.columns.focus_position - 1, 0
|
|
||||||
)
|
|
||||||
self.columns.focus.focus_position = old_focus_position
|
|
||||||
return None
|
|
||||||
|
|
||||||
if self.args.cheats:
|
|
||||||
if key == "e":
|
|
||||||
self.simulation.player.exp_bar.increment(1)
|
|
||||||
return None
|
|
||||||
|
|
||||||
fast_forward = {"t": 1, "T": 100, "ctrl t": 10000}
|
|
||||||
if key in fast_forward:
|
|
||||||
iterations = fast_forward[key]
|
|
||||||
ctx_mgr = (
|
|
||||||
self.suppress_updates()
|
|
||||||
if iterations >= 1000
|
|
||||||
else contextlib.suppress()
|
|
||||||
)
|
|
||||||
|
|
||||||
with ctx_mgr:
|
|
||||||
for _ in range(iterations):
|
|
||||||
self.simulation.tick()
|
|
||||||
|
|
||||||
return None
|
|
||||||
|
|
||||||
return super().keypress(size, key)
|
|
||||||
|
|
||||||
@contextlib.contextmanager
|
|
||||||
def suppress_updates(self) -> T.Generator:
|
|
||||||
old = SignalMixin.emit
|
|
||||||
SignalMixin.emit = lambda *_: None
|
|
||||||
yield
|
|
||||||
self.character_sheet_view.sync()
|
|
||||||
self.spell_book_view.sync()
|
|
||||||
self.equipment_view.sync()
|
|
||||||
self.inventory_view.sync()
|
|
||||||
self.plot_view.sync()
|
|
||||||
self.quest_book_view.sync()
|
|
||||||
self.task_view.sync()
|
|
||||||
SignalMixin.emit = old
|
|
||||||
@@ -1,119 +0,0 @@
|
|||||||
# Copyright (c) 2013-2016 Christian Geier et al.
|
|
||||||
#
|
|
||||||
# Permission is hereby granted, free of charge, to any person obtaining
|
|
||||||
# a copy of this software and associated documentation files (the
|
|
||||||
# "Software"), to deal in the Software without restriction, including
|
|
||||||
# without limitation the rights to use, copy, modify, merge, publish,
|
|
||||||
# distribute, sublicense, and/or sell copies of the Software, and to
|
|
||||||
# permit persons to whom the Software is furnished to do so, subject to
|
|
||||||
# the following conditions:
|
|
||||||
#
|
|
||||||
# The above copyright notice and this permission notice shall be
|
|
||||||
# included in all copies or substantial portions of the Software.
|
|
||||||
#
|
|
||||||
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
||||||
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
||||||
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
||||||
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
|
||||||
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
|
||||||
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
|
||||||
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
||||||
|
|
||||||
import typing as T
|
|
||||||
|
|
||||||
import urwid
|
|
||||||
|
|
||||||
|
|
||||||
class SupportsNext:
|
|
||||||
"""
|
|
||||||
Classes inheriting from SupportsNext must implement the following methods:
|
|
||||||
_select_first_selectable
|
|
||||||
_select_last_selectable
|
|
||||||
"""
|
|
||||||
|
|
||||||
def __init__(self, *args, **kwargs):
|
|
||||||
self.outermost = kwargs.get("outermost", False)
|
|
||||||
if "outermost" in kwargs:
|
|
||||||
kwargs.pop("outermost")
|
|
||||||
super().__init__(*args, **kwargs)
|
|
||||||
|
|
||||||
|
|
||||||
class NextMixin(SupportsNext):
|
|
||||||
"""Implements SupportsNext for urwid.Pile and urwid.Columns"""
|
|
||||||
|
|
||||||
def _select_first_selectable(self):
|
|
||||||
"""
|
|
||||||
select our first selectable item
|
|
||||||
(recursively if that item SupportsNext)
|
|
||||||
"""
|
|
||||||
i = self._first_selectable()
|
|
||||||
self.set_focus(i)
|
|
||||||
if isinstance(self.contents[i][0], SupportsNext):
|
|
||||||
self.contents[i][0]._select_first_selectable()
|
|
||||||
|
|
||||||
def _select_last_selectable(self):
|
|
||||||
"""
|
|
||||||
select our last selectable item
|
|
||||||
(recursively if that item SupportsNext)
|
|
||||||
"""
|
|
||||||
i = self._last_selectable()
|
|
||||||
self.set_focus(i)
|
|
||||||
if isinstance(self._contents[i][0], SupportsNext):
|
|
||||||
self.contents[i][0]._select_last_selectable()
|
|
||||||
|
|
||||||
def _first_selectable(self):
|
|
||||||
"""return sequence number of self.contents last selectable item"""
|
|
||||||
for j in range(0, len(self._contents)):
|
|
||||||
if self._contents[j][0].selectable():
|
|
||||||
return j
|
|
||||||
return False
|
|
||||||
|
|
||||||
def _last_selectable(self):
|
|
||||||
"""return sequence number of self._contents last selectable item"""
|
|
||||||
for j in range(len(self._contents) - 1, -1, -1):
|
|
||||||
if self._contents[j][0].selectable():
|
|
||||||
return j
|
|
||||||
return False
|
|
||||||
|
|
||||||
def keypress(self, size, key) -> T.Optional[str]:
|
|
||||||
key = super().keypress(size, key)
|
|
||||||
|
|
||||||
if key == "tab":
|
|
||||||
if (
|
|
||||||
self.outermost
|
|
||||||
and self.focus_position == self._last_selectable()
|
|
||||||
):
|
|
||||||
self._select_first_selectable()
|
|
||||||
return None
|
|
||||||
|
|
||||||
for i in range(self.focus_position + 1, len(self._contents)):
|
|
||||||
if self._contents[i][0].selectable():
|
|
||||||
self.set_focus(i)
|
|
||||||
if isinstance(self._contents[i][0], SupportsNext):
|
|
||||||
self._contents[i][0]._select_first_selectable()
|
|
||||||
return None
|
|
||||||
|
|
||||||
elif key == "shift tab":
|
|
||||||
if (
|
|
||||||
self.outermost
|
|
||||||
and self.focus_position == self._first_selectable()
|
|
||||||
):
|
|
||||||
self._select_last_selectable()
|
|
||||||
return None
|
|
||||||
|
|
||||||
for i in range(self.focus_position - 1, 0 - 1, -1):
|
|
||||||
if self._contents[i][0].selectable():
|
|
||||||
self.set_focus(i)
|
|
||||||
if isinstance(self._contents[i][0], SupportsNext):
|
|
||||||
self._contents[i][0]._select_last_selectable()
|
|
||||||
return None
|
|
||||||
|
|
||||||
return key
|
|
||||||
|
|
||||||
|
|
||||||
class NPile(NextMixin, urwid.Pile):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class NColumns(NextMixin, urwid.Columns):
|
|
||||||
pass
|
|
||||||
@@ -1,223 +0,0 @@
|
|||||||
import random
|
|
||||||
import typing as T
|
|
||||||
|
|
||||||
import urwid
|
|
||||||
import urwid_readline
|
|
||||||
|
|
||||||
from pqcli.config import CLASSES, PRIME_STATS, RACES
|
|
||||||
from pqcli.mechanic import StatsBuilder, create_player, generate_name
|
|
||||||
from pqcli.ui.custom_button import CustomButton
|
|
||||||
from pqcli.ui.custom_line_box import CustomLineBox
|
|
||||||
from pqcli.ui.layout import NColumns, NPile
|
|
||||||
|
|
||||||
|
|
||||||
class StatsBox(CustomLineBox):
|
|
||||||
def __init__(self) -> None:
|
|
||||||
self.stats_builder = StatsBuilder()
|
|
||||||
self.stats = self.stats_builder.roll()
|
|
||||||
self.stat_labels = {stat: urwid.Text("0") for stat in PRIME_STATS}
|
|
||||||
self.total_label = urwid.Text("0")
|
|
||||||
self.update_values()
|
|
||||||
|
|
||||||
roll_button = CustomButton(
|
|
||||||
"Roll", hint="F5", on_press=self.on_roll_press
|
|
||||||
)
|
|
||||||
unroll_button = CustomButton(
|
|
||||||
"Unroll", hint="F6", on_press=self.on_unroll_press
|
|
||||||
)
|
|
||||||
|
|
||||||
value_texts = list(self.stat_labels.values())
|
|
||||||
label_texts = [urwid.Text(f"{stat.value}: ") for stat in PRIME_STATS]
|
|
||||||
|
|
||||||
super().__init__(
|
|
||||||
urwid.Filler(
|
|
||||||
NPile(
|
|
||||||
[
|
|
||||||
NColumns([NPile(label_texts), NPile(value_texts)]),
|
|
||||||
urwid.Divider(),
|
|
||||||
NColumns([urwid.Text("Total: "), self.total_label]),
|
|
||||||
urwid.Divider(),
|
|
||||||
roll_button,
|
|
||||||
unroll_button,
|
|
||||||
]
|
|
||||||
)
|
|
||||||
),
|
|
||||||
title="Stats",
|
|
||||||
)
|
|
||||||
|
|
||||||
def roll(self) -> None:
|
|
||||||
self.stats = self.stats_builder.roll()
|
|
||||||
self.update_values()
|
|
||||||
|
|
||||||
def unroll(self) -> None:
|
|
||||||
try:
|
|
||||||
self.stats = self.stats_builder.unroll()
|
|
||||||
except IndexError:
|
|
||||||
return
|
|
||||||
self.update_values()
|
|
||||||
|
|
||||||
def on_roll_press(self, _user_data: T.Any) -> None:
|
|
||||||
self.roll()
|
|
||||||
|
|
||||||
def on_unroll_press(self, _user_data: T.Any) -> None:
|
|
||||||
self.unroll()
|
|
||||||
|
|
||||||
def update_values(self) -> None:
|
|
||||||
for stat, edit in self.stat_labels.items():
|
|
||||||
edit.set_text(str(self.stats[stat]))
|
|
||||||
self.total_label.set_text(
|
|
||||||
str(sum(self.stats[stat] for stat in self.stat_labels.keys()))
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
class RaceBox(CustomLineBox):
|
|
||||||
def __init__(self) -> None:
|
|
||||||
self.race = random.choice(RACES)
|
|
||||||
|
|
||||||
radio_buttons: T.List[urwid.RadioButton] = []
|
|
||||||
for race in RACES:
|
|
||||||
urwid.RadioButton(
|
|
||||||
group=radio_buttons,
|
|
||||||
label=race.name,
|
|
||||||
state=race.name == self.race.name,
|
|
||||||
on_state_change=self.on_state_change,
|
|
||||||
user_data=race,
|
|
||||||
)
|
|
||||||
|
|
||||||
super().__init__(urwid.ListBox(radio_buttons), title="Race")
|
|
||||||
|
|
||||||
def on_state_change(
|
|
||||||
self, _widget: urwid.RadioButton, new_state: bool, user_data: T.Any
|
|
||||||
) -> None:
|
|
||||||
if new_state:
|
|
||||||
self.race = user_data
|
|
||||||
|
|
||||||
|
|
||||||
class ClassBox(CustomLineBox):
|
|
||||||
def __init__(self) -> None:
|
|
||||||
self.class_ = random.choice(CLASSES)
|
|
||||||
|
|
||||||
radio_buttons: T.List[urwid.RadioButton] = []
|
|
||||||
for class_ in CLASSES:
|
|
||||||
urwid.RadioButton(
|
|
||||||
group=radio_buttons,
|
|
||||||
label=class_.name,
|
|
||||||
state=class_.name == self.class_.name,
|
|
||||||
on_state_change=self.on_state_change,
|
|
||||||
user_data=class_,
|
|
||||||
)
|
|
||||||
|
|
||||||
super().__init__(urwid.ListBox(radio_buttons), title="Class")
|
|
||||||
|
|
||||||
def on_state_change(
|
|
||||||
self, _widget: urwid.RadioButton, new_state: bool, user_data: T.Any
|
|
||||||
) -> None:
|
|
||||||
if new_state:
|
|
||||||
self.class_ = user_data
|
|
||||||
|
|
||||||
|
|
||||||
class NewGameView(NPile):
|
|
||||||
signals = ["confirm", "cancel"]
|
|
||||||
|
|
||||||
def __init__(self) -> None:
|
|
||||||
self.race_box = RaceBox()
|
|
||||||
self.class_box = ClassBox()
|
|
||||||
self.stats_box = StatsBox()
|
|
||||||
|
|
||||||
self.char_name_edit = urwid_readline.ReadlineEdit(
|
|
||||||
"Name: ", generate_name()
|
|
||||||
)
|
|
||||||
generate_char_name_btn = CustomButton(
|
|
||||||
"Generate random name",
|
|
||||||
hint="F7",
|
|
||||||
on_press=self.on_generate_char_name_press,
|
|
||||||
)
|
|
||||||
|
|
||||||
buttons_box = urwid.Filler(
|
|
||||||
urwid.Padding(
|
|
||||||
NPile(
|
|
||||||
[
|
|
||||||
CustomButton(
|
|
||||||
"Sold!", hint="F10", on_press=self.on_confirm_press
|
|
||||||
),
|
|
||||||
CustomButton(
|
|
||||||
"Cancel", hint="Esc", on_press=self.on_cancel_press
|
|
||||||
),
|
|
||||||
]
|
|
||||||
),
|
|
||||||
width=20,
|
|
||||||
align="center",
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
super().__init__(
|
|
||||||
[
|
|
||||||
(1, urwid.Filler(self.char_name_edit)),
|
|
||||||
(
|
|
||||||
1,
|
|
||||||
urwid.Filler(
|
|
||||||
urwid.Padding(
|
|
||||||
generate_char_name_btn,
|
|
||||||
left=len(self.char_name_edit.caption),
|
|
||||||
width=25,
|
|
||||||
)
|
|
||||||
),
|
|
||||||
),
|
|
||||||
NColumns(
|
|
||||||
[
|
|
||||||
(urwid.WEIGHT, 2, self.race_box),
|
|
||||||
(urwid.WEIGHT, 2, self.class_box),
|
|
||||||
NPile([(13, self.stats_box), buttons_box]),
|
|
||||||
]
|
|
||||||
),
|
|
||||||
],
|
|
||||||
outermost=True,
|
|
||||||
)
|
|
||||||
|
|
||||||
def generate_random_char_name(self) -> None:
|
|
||||||
self.char_name_edit.edit_text = generate_name()
|
|
||||||
self.char_name_edit.edit_pos = len(self.char_name_edit.edit_text)
|
|
||||||
|
|
||||||
def confirm(self) -> None:
|
|
||||||
player = create_player(
|
|
||||||
name=self.char_name_edit.edit_text,
|
|
||||||
race=self.race_box.race,
|
|
||||||
class_=self.class_box.class_,
|
|
||||||
stats=self.stats_box.stats,
|
|
||||||
)
|
|
||||||
self._emit("confirm", player)
|
|
||||||
|
|
||||||
def cancel(self) -> None:
|
|
||||||
self._emit("cancel")
|
|
||||||
|
|
||||||
def on_generate_char_name_press(self, _user_data: T.Any) -> None:
|
|
||||||
self.generate_random_char_name()
|
|
||||||
|
|
||||||
def on_confirm_press(self, _user_data: T.Any) -> None:
|
|
||||||
self.confirm()
|
|
||||||
|
|
||||||
def on_cancel_press(self, _user_data: T.Any) -> None:
|
|
||||||
self.cancel()
|
|
||||||
|
|
||||||
def unhandled_input(self, key: str) -> bool:
|
|
||||||
if key == "esc":
|
|
||||||
self.cancel()
|
|
||||||
return True
|
|
||||||
|
|
||||||
if key == "f7":
|
|
||||||
self.generate_random_char_name()
|
|
||||||
return True
|
|
||||||
|
|
||||||
if key == "f5":
|
|
||||||
self.stats_box.roll()
|
|
||||||
return True
|
|
||||||
|
|
||||||
if key == "f6":
|
|
||||||
self.stats_box.unroll()
|
|
||||||
return True
|
|
||||||
|
|
||||||
if key == "f10":
|
|
||||||
self.confirm()
|
|
||||||
return True
|
|
||||||
|
|
||||||
return False
|
|
||||||
@@ -1,16 +0,0 @@
|
|||||||
import urwid
|
|
||||||
|
|
||||||
|
|
||||||
class ReadOnlyCheckBox(urwid.Text):
|
|
||||||
def __init__(self, text: str, state: bool = False) -> None:
|
|
||||||
self.icons = {
|
|
||||||
key: widget.text for key, widget in urwid.CheckBox.states.items()
|
|
||||||
}
|
|
||||||
self._caption = text
|
|
||||||
|
|
||||||
super().__init__("")
|
|
||||||
|
|
||||||
self.set_state(state)
|
|
||||||
|
|
||||||
def set_state(self, state: bool) -> None:
|
|
||||||
self.set_text(self.icons[state] + " " + self._caption)
|
|
||||||
@@ -1,115 +0,0 @@
|
|||||||
import typing as T
|
|
||||||
|
|
||||||
import urwid
|
|
||||||
|
|
||||||
from pqcli.lingo import act_name, to_roman
|
|
||||||
from pqcli.roster import Roster
|
|
||||||
from pqcli.ui.custom_button import CustomButton
|
|
||||||
from pqcli.ui.layout import NColumns, NPile
|
|
||||||
|
|
||||||
|
|
||||||
class RosterView(urwid.Filler):
|
|
||||||
signals = ["load_game", "new_game", "delete_game", "exit"]
|
|
||||||
|
|
||||||
def __init__(self, roster: Roster) -> None:
|
|
||||||
self.roster = roster
|
|
||||||
|
|
||||||
logo = urwid.BigText("Progress Quest", urwid.HalfBlock5x4Font())
|
|
||||||
|
|
||||||
buttons = []
|
|
||||||
for player_idx, player in enumerate(self.roster.players):
|
|
||||||
best_equip = player.equipment.best
|
|
||||||
|
|
||||||
best_spell = player.spell_book.best
|
|
||||||
if best_spell:
|
|
||||||
best_spell_name = (
|
|
||||||
best_spell.name + " " + to_roman(best_spell.level)
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
best_spell_name = "-"
|
|
||||||
|
|
||||||
best_stat_name = (
|
|
||||||
f"{player.stats.best_prime.value} "
|
|
||||||
f"{player.stats[player.stats.best_prime]}"
|
|
||||||
)
|
|
||||||
|
|
||||||
label = (
|
|
||||||
f"{player.name} the {player.race.name} "
|
|
||||||
f"({act_name(player.quest_book.act)})\n"
|
|
||||||
f"Level {player.level} {player.class_.name}\n"
|
|
||||||
f"{best_equip} / {best_spell_name} / {best_stat_name}"
|
|
||||||
)
|
|
||||||
|
|
||||||
buttons.append(
|
|
||||||
NColumns(
|
|
||||||
[
|
|
||||||
CustomButton(
|
|
||||||
label=label,
|
|
||||||
on_press=self.on_resume_game_press,
|
|
||||||
user_data=player_idx,
|
|
||||||
),
|
|
||||||
(
|
|
||||||
10,
|
|
||||||
CustomButton(
|
|
||||||
label="Delete",
|
|
||||||
on_press=self.on_delete_game_press,
|
|
||||||
user_data=player_idx,
|
|
||||||
),
|
|
||||||
),
|
|
||||||
]
|
|
||||||
)
|
|
||||||
)
|
|
||||||
buttons.append(urwid.Divider())
|
|
||||||
|
|
||||||
buttons.append(
|
|
||||||
CustomButton(
|
|
||||||
label="Create a new character",
|
|
||||||
hint="F1",
|
|
||||||
on_press=self.on_new_game_press,
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
buttons.append(
|
|
||||||
CustomButton(label="Exit", hint="Esc", on_press=self.on_exit_press)
|
|
||||||
)
|
|
||||||
|
|
||||||
super().__init__(
|
|
||||||
urwid.Padding(
|
|
||||||
NPile(
|
|
||||||
[urwid.Padding(logo, width="clip"), urwid.Divider()]
|
|
||||||
+ buttons,
|
|
||||||
outermost=True,
|
|
||||||
),
|
|
||||||
align="center",
|
|
||||||
width=logo.pack()[0],
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
def on_resume_game_press(
|
|
||||||
self, _widget: urwid.Widget, user_data: T.Any
|
|
||||||
) -> None:
|
|
||||||
player_idx = T.cast(int, user_data)
|
|
||||||
self._emit("load_game", player_idx)
|
|
||||||
|
|
||||||
def on_delete_game_press(
|
|
||||||
self, _widget: urwid.Widget, user_data: T.Any
|
|
||||||
) -> None:
|
|
||||||
player_idx = T.cast(int, user_data)
|
|
||||||
self._emit("delete_game", player_idx)
|
|
||||||
|
|
||||||
def on_new_game_press(self, _user_data: T.Any) -> None:
|
|
||||||
self._emit("new_game")
|
|
||||||
|
|
||||||
def on_exit_press(self, _user_data: T.Any) -> None:
|
|
||||||
self._emit("exit")
|
|
||||||
|
|
||||||
def unhandled_input(self, key: str) -> bool:
|
|
||||||
if key == "esc":
|
|
||||||
self._emit("exit")
|
|
||||||
return True
|
|
||||||
|
|
||||||
if key == "f1":
|
|
||||||
self._emit("new_game")
|
|
||||||
return True
|
|
||||||
|
|
||||||
return False
|
|
||||||
@@ -1,498 +0,0 @@
|
|||||||
# This program is free software: you can redistribute it and/or modify
|
|
||||||
# it under the terms of the GNU General Public License as published by
|
|
||||||
# the Free Software Foundation, either version 3 of the License, or
|
|
||||||
# (at your option) any later version.
|
|
||||||
#
|
|
||||||
# This program is distributed in the hope that it will be useful
|
|
||||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
||||||
# GNU General Public License for more details
|
|
||||||
# http://www.gnu.org/licenses/gpl-3.0.txt
|
|
||||||
|
|
||||||
# Taken from stig (stig.tui.scroll).
|
|
||||||
# Changes compared to upstream:
|
|
||||||
# - reformatted the code
|
|
||||||
# - added AttrMap wrapping
|
|
||||||
# - added padding support
|
|
||||||
# - removed up/down arrow navigation (for widget navigation)
|
|
||||||
|
|
||||||
import typing as T
|
|
||||||
|
|
||||||
import urwid
|
|
||||||
from urwid.widget import BOX, FIXED, FLOW
|
|
||||||
|
|
||||||
# Scroll actions
|
|
||||||
SCROLL_LINE_UP = "line up"
|
|
||||||
SCROLL_LINE_DOWN = "line down"
|
|
||||||
SCROLL_PAGE_UP = "page up"
|
|
||||||
SCROLL_PAGE_DOWN = "page down"
|
|
||||||
SCROLL_TO_TOP = "to top"
|
|
||||||
SCROLL_TO_END = "to end"
|
|
||||||
|
|
||||||
# Scrollbar positions
|
|
||||||
SCROLLBAR_LEFT = "left"
|
|
||||||
SCROLLBAR_RIGHT = "right"
|
|
||||||
|
|
||||||
|
|
||||||
class EmptyCanvas(urwid.Canvas):
|
|
||||||
def __init__(self, cols: int, rows: int) -> None:
|
|
||||||
super().__init__()
|
|
||||||
self._text, cs = urwid.apply_target_encoding(" ")
|
|
||||||
self._cs = cs[0][0]
|
|
||||||
self.size = cols, rows
|
|
||||||
self.cursor = None
|
|
||||||
|
|
||||||
def cols(self) -> int:
|
|
||||||
return self.size[0]
|
|
||||||
|
|
||||||
def rows(self) -> int:
|
|
||||||
return self.size[1]
|
|
||||||
|
|
||||||
def content(
|
|
||||||
self,
|
|
||||||
trim_left: int = 0,
|
|
||||||
trim_top: int = 0,
|
|
||||||
cols: T.Optional[int] = None,
|
|
||||||
rows: T.Optional[int] = None,
|
|
||||||
attr: T.Optional[str] = None,
|
|
||||||
) -> T.Generator:
|
|
||||||
if cols is None:
|
|
||||||
cols = self.size[0]
|
|
||||||
if rows is None:
|
|
||||||
rows = self.size[1]
|
|
||||||
|
|
||||||
line = [(None, self._cs, self._text * cols)]
|
|
||||||
for i in range(rows):
|
|
||||||
yield line
|
|
||||||
|
|
||||||
def content_delta(self, other: T.Any) -> T.Any:
|
|
||||||
if other is self:
|
|
||||||
return [self.cols()] * self.rows()
|
|
||||||
return self.content()
|
|
||||||
|
|
||||||
|
|
||||||
class _Scrollable(urwid.WidgetDecoration):
|
|
||||||
def __init__(self, widget):
|
|
||||||
"""Box widget that makes a fixed or flow widget vertically scrollable
|
|
||||||
|
|
||||||
TODO: Focusable widgets are handled, including switching focus, but
|
|
||||||
possibly not intuitively, depending on the arrangement of widgets.
|
|
||||||
When switching focus to a widget that is ouside of the visible part of
|
|
||||||
the original widget, the canvas scrolls up/down to the focused widget.
|
|
||||||
It would be better to scroll until the next focusable widget is in
|
|
||||||
sight first. But for that to work we must somehow obtain a list of
|
|
||||||
focusable rows in the original canvas.
|
|
||||||
"""
|
|
||||||
if not any(s in widget.sizing() for s in (FIXED, FLOW)):
|
|
||||||
raise ValueError("Not a fixed or flow widget: %r" % widget)
|
|
||||||
self._trim_top = 0
|
|
||||||
self._scroll_action = None
|
|
||||||
self._forward_keypress = None
|
|
||||||
self._old_cursor_coords = None
|
|
||||||
self._rows_max_cached = 0
|
|
||||||
super().__init__(widget)
|
|
||||||
|
|
||||||
def sizing(self):
|
|
||||||
return frozenset([BOX])
|
|
||||||
|
|
||||||
def selectable(self):
|
|
||||||
return True
|
|
||||||
|
|
||||||
def render(self, size, focus=False):
|
|
||||||
maxcol, maxrow = size
|
|
||||||
|
|
||||||
# Render complete original widget
|
|
||||||
ow = self._original_widget
|
|
||||||
ow_size = self._get_original_widget_size(size)
|
|
||||||
canv_full = ow.render(ow_size, focus)
|
|
||||||
|
|
||||||
# Make full canvas editable
|
|
||||||
canv = urwid.CompositeCanvas(canv_full)
|
|
||||||
canv_cols, canv_rows = canv.cols(), canv.rows()
|
|
||||||
|
|
||||||
if canv_cols <= maxcol:
|
|
||||||
pad_width = maxcol - canv_cols
|
|
||||||
if pad_width > 0:
|
|
||||||
# Canvas is narrower than available horizontal space
|
|
||||||
canv.pad_trim_left_right(0, pad_width)
|
|
||||||
|
|
||||||
if canv_rows <= maxrow:
|
|
||||||
fill_height = maxrow - canv_rows
|
|
||||||
if fill_height > 0:
|
|
||||||
# Canvas is lower than available vertical space
|
|
||||||
canv.pad_trim_top_bottom(0, fill_height)
|
|
||||||
|
|
||||||
if canv_cols <= maxcol and canv_rows <= maxrow:
|
|
||||||
# Canvas is small enough to fit without trimming
|
|
||||||
return canv
|
|
||||||
|
|
||||||
self._adjust_trim_top(canv, size)
|
|
||||||
|
|
||||||
# Trim canvas if necessary
|
|
||||||
trim_top = self._trim_top
|
|
||||||
trim_end = canv_rows - maxrow - trim_top
|
|
||||||
trim_right = canv_cols - maxcol
|
|
||||||
if trim_top > 0:
|
|
||||||
canv.trim(trim_top)
|
|
||||||
if trim_end > 0:
|
|
||||||
canv.trim_end(trim_end)
|
|
||||||
if trim_right > 0:
|
|
||||||
canv.pad_trim_left_right(0, -trim_right)
|
|
||||||
|
|
||||||
# Disable cursor display if cursor is outside of visible canvas parts
|
|
||||||
if canv.cursor is not None:
|
|
||||||
curscol, cursrow = canv.cursor
|
|
||||||
if cursrow >= maxrow or cursrow < 0:
|
|
||||||
canv.cursor = None
|
|
||||||
|
|
||||||
# Figure out whether we should forward keypresses to original widget
|
|
||||||
if canv.cursor is not None:
|
|
||||||
# Trimmed canvas contains the cursor, e.g. in an Edit widget
|
|
||||||
self._forward_keypress = True
|
|
||||||
else:
|
|
||||||
if canv_full.cursor is not None:
|
|
||||||
# Full canvas contains the cursor, but scrolled out of view
|
|
||||||
self._forward_keypress = False
|
|
||||||
else:
|
|
||||||
# Original widget does not have a cursor, but may be selectable
|
|
||||||
|
|
||||||
# FIXME: Using ow.selectable() is bad because the original
|
|
||||||
# widget may be selectable because it's a container widget with
|
|
||||||
# a key-grabbing widget that is scrolled out of view.
|
|
||||||
# ow.selectable() returns True anyway because it doesn't know
|
|
||||||
# how we trimmed our canvas.
|
|
||||||
#
|
|
||||||
# To fix this, we need to resolve ow.focus and somehow
|
|
||||||
# ask canv whether it contains bits of the focused widget.
|
|
||||||
# I can't see a way to do that.
|
|
||||||
if ow.selectable():
|
|
||||||
self._forward_keypress = True
|
|
||||||
else:
|
|
||||||
self._forward_keypress = False
|
|
||||||
|
|
||||||
return canv
|
|
||||||
|
|
||||||
def keypress(self, size, key):
|
|
||||||
# Maybe offer key to original widget
|
|
||||||
if self._forward_keypress:
|
|
||||||
ow = self._original_widget
|
|
||||||
ow_size = self._get_original_widget_size(size)
|
|
||||||
|
|
||||||
# Remember previous cursor position if possible
|
|
||||||
if hasattr(ow, "get_cursor_coords"):
|
|
||||||
self._old_cursor_coords = ow.get_cursor_coords(ow_size)
|
|
||||||
|
|
||||||
key = ow.keypress(ow_size, key)
|
|
||||||
if key is None:
|
|
||||||
return None
|
|
||||||
|
|
||||||
# Handle up/down, page up/down, etc
|
|
||||||
command_map = self._command_map
|
|
||||||
if command_map[key] == urwid.CURSOR_PAGE_UP:
|
|
||||||
self._scroll_action = SCROLL_PAGE_UP
|
|
||||||
elif command_map[key] == urwid.CURSOR_PAGE_DOWN:
|
|
||||||
self._scroll_action = SCROLL_PAGE_DOWN
|
|
||||||
|
|
||||||
elif command_map[key] == urwid.CURSOR_MAX_LEFT: # 'home'
|
|
||||||
self._scroll_action = SCROLL_TO_TOP
|
|
||||||
elif command_map[key] == urwid.CURSOR_MAX_RIGHT: # 'end'
|
|
||||||
self._scroll_action = SCROLL_TO_END
|
|
||||||
|
|
||||||
else:
|
|
||||||
return key
|
|
||||||
|
|
||||||
self._invalidate()
|
|
||||||
|
|
||||||
def mouse_event(self, size, event, button, col, row, focus):
|
|
||||||
ow = self._original_widget
|
|
||||||
if hasattr(ow, "mouse_event"):
|
|
||||||
ow_size = self._get_original_widget_size(size)
|
|
||||||
row += self._trim_top
|
|
||||||
return ow.mouse_event(ow_size, event, button, col, row, focus)
|
|
||||||
else:
|
|
||||||
return False
|
|
||||||
|
|
||||||
def _adjust_trim_top(self, canv, size):
|
|
||||||
"""Adjust self._trim_top according to self._scroll_action"""
|
|
||||||
action = self._scroll_action
|
|
||||||
self._scroll_action = None
|
|
||||||
|
|
||||||
maxcol, maxrow = size
|
|
||||||
trim_top = self._trim_top
|
|
||||||
canv_rows = canv.rows()
|
|
||||||
|
|
||||||
if trim_top < 0:
|
|
||||||
# Negative trim_top values use bottom of canvas as reference
|
|
||||||
trim_top = canv_rows - maxrow + trim_top + 1
|
|
||||||
|
|
||||||
if canv_rows <= maxrow:
|
|
||||||
self._trim_top = 0 # Reset scroll position
|
|
||||||
return
|
|
||||||
|
|
||||||
def ensure_bounds(new_trim_top):
|
|
||||||
return max(0, min(canv_rows - maxrow, new_trim_top))
|
|
||||||
|
|
||||||
if action == SCROLL_LINE_UP:
|
|
||||||
self._trim_top = ensure_bounds(trim_top - 1)
|
|
||||||
elif action == SCROLL_LINE_DOWN:
|
|
||||||
self._trim_top = ensure_bounds(trim_top + 1)
|
|
||||||
|
|
||||||
elif action == SCROLL_PAGE_UP:
|
|
||||||
self._trim_top = ensure_bounds(trim_top - maxrow + 1)
|
|
||||||
elif action == SCROLL_PAGE_DOWN:
|
|
||||||
self._trim_top = ensure_bounds(trim_top + maxrow - 1)
|
|
||||||
|
|
||||||
elif action == SCROLL_TO_TOP:
|
|
||||||
self._trim_top = 0
|
|
||||||
elif action == SCROLL_TO_END:
|
|
||||||
self._trim_top = canv_rows - maxrow
|
|
||||||
|
|
||||||
else:
|
|
||||||
self._trim_top = ensure_bounds(trim_top)
|
|
||||||
|
|
||||||
# If the cursor was moved by the most recent keypress, adjust trim_top
|
|
||||||
# so that the new cursor position is within the displayed canvas part.
|
|
||||||
# But don't do this if the cursor is at the top/bottom edge so we can still scroll out
|
|
||||||
if (
|
|
||||||
self._old_cursor_coords is not None
|
|
||||||
and self._old_cursor_coords != canv.cursor
|
|
||||||
):
|
|
||||||
self._old_cursor_coords = None
|
|
||||||
curscol, cursrow = canv.cursor
|
|
||||||
if cursrow < self._trim_top:
|
|
||||||
self._trim_top = cursrow
|
|
||||||
elif cursrow >= self._trim_top + maxrow:
|
|
||||||
self._trim_top = max(0, cursrow - maxrow + 1)
|
|
||||||
|
|
||||||
def _get_original_widget_size(self, size):
|
|
||||||
ow = self._original_widget
|
|
||||||
sizing = ow.sizing()
|
|
||||||
if FIXED in sizing:
|
|
||||||
return ()
|
|
||||||
elif FLOW in sizing:
|
|
||||||
return (size[0],)
|
|
||||||
|
|
||||||
def get_scrollpos(self, size=None, focus=False):
|
|
||||||
"""Current scrolling position
|
|
||||||
|
|
||||||
Lower limit is 0, upper limit is the maximum number of rows with the
|
|
||||||
given maxcol minus maxrow.
|
|
||||||
|
|
||||||
NOTE: The returned value may be too low or too high if the position has
|
|
||||||
changed but the widget wasn't rendered yet.
|
|
||||||
"""
|
|
||||||
return self._trim_top
|
|
||||||
|
|
||||||
def set_scrollpos(self, position):
|
|
||||||
"""Set scrolling position
|
|
||||||
|
|
||||||
If `position` is positive it is interpreted as lines from the top.
|
|
||||||
If `position` is negative it is interpreted as lines from the bottom.
|
|
||||||
|
|
||||||
Values that are too high or too low values are automatically adjusted
|
|
||||||
during rendering.
|
|
||||||
"""
|
|
||||||
self._trim_top = int(position)
|
|
||||||
self._invalidate()
|
|
||||||
|
|
||||||
def rows_max(self, size=None, focus=False):
|
|
||||||
"""Return the number of rows for `size`
|
|
||||||
|
|
||||||
If `size` is not given, the currently rendered number of rows is returned.
|
|
||||||
"""
|
|
||||||
if size is not None:
|
|
||||||
ow = self._original_widget
|
|
||||||
ow_size = self._get_original_widget_size(size)
|
|
||||||
sizing = ow.sizing()
|
|
||||||
if FIXED in sizing:
|
|
||||||
self._rows_max_cached = ow.pack(ow_size, focus)[1]
|
|
||||||
elif FLOW in sizing:
|
|
||||||
self._rows_max_cached = ow.rows(ow_size, focus)
|
|
||||||
else:
|
|
||||||
raise RuntimeError(
|
|
||||||
"Not a flow/box widget: %r" % self._original_widget
|
|
||||||
)
|
|
||||||
return self._rows_max_cached
|
|
||||||
|
|
||||||
|
|
||||||
class _ScrollBar(urwid.WidgetDecoration):
|
|
||||||
def __init__(
|
|
||||||
self,
|
|
||||||
widget,
|
|
||||||
thumb_char="\N{FULL BLOCK}",
|
|
||||||
trough_char=" ",
|
|
||||||
side=SCROLLBAR_RIGHT,
|
|
||||||
width=1,
|
|
||||||
padding=0,
|
|
||||||
):
|
|
||||||
"""Box widget that adds a scrollbar to `widget`
|
|
||||||
|
|
||||||
`widget` must be a box widget with the following methods:
|
|
||||||
- `get_scrollpos` takes the arguments `size` and `focus` and returns
|
|
||||||
the index of the first visible row.
|
|
||||||
- `set_scrollpos` (optional; needed for mouse click support) takes the
|
|
||||||
index of the first visible row.
|
|
||||||
- `rows_max` takes `size` and `focus` and returns the total number of
|
|
||||||
rows `widget` can render.
|
|
||||||
|
|
||||||
`thumb_char` is the character used for the scrollbar handle.
|
|
||||||
`trough_char` is used for the space above and below the handle.
|
|
||||||
`side` must be 'left' or 'right'.
|
|
||||||
`width` specifies the number of columns the scrollbar uses.
|
|
||||||
`padding` specifies the number of columns the scrollbar is separated
|
|
||||||
from the content by.
|
|
||||||
"""
|
|
||||||
if BOX not in widget.sizing():
|
|
||||||
raise ValueError("Not a box widget: %r" % widget)
|
|
||||||
super().__init__(widget)
|
|
||||||
self._thumb_char = thumb_char
|
|
||||||
self._trough_char = trough_char
|
|
||||||
self._scrollbar_side = side
|
|
||||||
self._scrollbar_width = max(1, width)
|
|
||||||
self._padding_width = padding
|
|
||||||
self._original_widget_size = (0, 0)
|
|
||||||
|
|
||||||
def sizing(self):
|
|
||||||
return frozenset((BOX,))
|
|
||||||
|
|
||||||
def selectable(self):
|
|
||||||
return True
|
|
||||||
|
|
||||||
def render(self, size, focus=False):
|
|
||||||
maxcol, maxrow = size
|
|
||||||
sb_width = self._scrollbar_width
|
|
||||||
pad_width = self._padding_width
|
|
||||||
self._original_widget_size = ow_size = (
|
|
||||||
maxcol - sb_width - pad_width,
|
|
||||||
maxrow,
|
|
||||||
)
|
|
||||||
|
|
||||||
ow = self._original_widget
|
|
||||||
ow_base = self.scrolling_base_widget
|
|
||||||
ow_rows_max = ow_base.rows_max(size, focus)
|
|
||||||
if ow_rows_max <= maxrow:
|
|
||||||
# Canvas fits without scrolling - no scrollbar needed
|
|
||||||
self._original_widget_size = size
|
|
||||||
return ow.render(size, focus)
|
|
||||||
ow_rows_max = ow_base.rows_max(ow_size, focus)
|
|
||||||
|
|
||||||
ow_canv = ow.render(ow_size, focus)
|
|
||||||
|
|
||||||
pos = ow_base.get_scrollpos(ow_size, focus)
|
|
||||||
posmax = ow_rows_max - maxrow
|
|
||||||
|
|
||||||
# Thumb shrinks/grows according to the ratio of
|
|
||||||
# <number of visible lines> / <number of total lines>
|
|
||||||
thumb_weight = min(1, maxrow / max(1, ow_rows_max))
|
|
||||||
thumb_height = max(1, round(thumb_weight * maxrow))
|
|
||||||
|
|
||||||
# Thumb may only touch top/bottom if the first/last row is visible
|
|
||||||
top_weight = float(pos) / max(1, posmax)
|
|
||||||
top_height = int((maxrow - thumb_height) * top_weight)
|
|
||||||
if top_height == 0 and top_weight > 0:
|
|
||||||
top_height = 1
|
|
||||||
|
|
||||||
# Bottom part is remaining space
|
|
||||||
bottom_height = maxrow - thumb_height - top_height
|
|
||||||
assert thumb_height + top_height + bottom_height == maxrow
|
|
||||||
|
|
||||||
# Create scrollbar canvas
|
|
||||||
top = urwid.SolidCanvas(self._trough_char, sb_width, 1)
|
|
||||||
thumb = urwid.SolidCanvas(self._thumb_char, sb_width, 1)
|
|
||||||
bottom = urwid.SolidCanvas(self._trough_char, sb_width, 1)
|
|
||||||
sb_canv = urwid.CanvasCombine(
|
|
||||||
[(top, None, False)] * top_height
|
|
||||||
+ [(thumb, None, False)] * thumb_height
|
|
||||||
+ [(bottom, None, False)] * bottom_height
|
|
||||||
)
|
|
||||||
|
|
||||||
# Create padding canvas
|
|
||||||
pad_canv = EmptyCanvas(pad_width, maxrow)
|
|
||||||
|
|
||||||
combinelist = [
|
|
||||||
(ow_canv, None, True, ow_size[0]),
|
|
||||||
(pad_canv, None, False, pad_width),
|
|
||||||
(sb_canv, None, False, sb_width),
|
|
||||||
]
|
|
||||||
if self._scrollbar_side != SCROLLBAR_LEFT:
|
|
||||||
return urwid.CanvasJoin(combinelist)
|
|
||||||
else:
|
|
||||||
return urwid.CanvasJoin(reversed(combinelist))
|
|
||||||
|
|
||||||
@property
|
|
||||||
def scrollbar_width(self):
|
|
||||||
"""Columns the scrollbar uses"""
|
|
||||||
return max(1, self._scrollbar_width)
|
|
||||||
|
|
||||||
@scrollbar_width.setter
|
|
||||||
def scrollbar_width(self, width):
|
|
||||||
self._scrollbar_width = max(1, int(width))
|
|
||||||
self._invalidate()
|
|
||||||
|
|
||||||
@property
|
|
||||||
def scrollbar_side(self):
|
|
||||||
"""Where to display the scrollbar; must be 'left' or 'right'"""
|
|
||||||
return self._scrollbar_side
|
|
||||||
|
|
||||||
@scrollbar_side.setter
|
|
||||||
def scrollbar_side(self, side):
|
|
||||||
if side not in (SCROLLBAR_LEFT, SCROLLBAR_RIGHT):
|
|
||||||
raise ValueError(
|
|
||||||
'scrollbar_side must be "left" or "right", not %r' % side
|
|
||||||
)
|
|
||||||
self._scrollbar_side = side
|
|
||||||
self._invalidate()
|
|
||||||
|
|
||||||
@property
|
|
||||||
def scrolling_base_widget(self):
|
|
||||||
"""Nearest `original_widget` that is compatible with the scrolling API"""
|
|
||||||
|
|
||||||
def orig_iter(w):
|
|
||||||
while hasattr(w, "original_widget"):
|
|
||||||
w = w.original_widget
|
|
||||||
yield w
|
|
||||||
yield w
|
|
||||||
|
|
||||||
def is_scrolling_widget(w):
|
|
||||||
return hasattr(w, "get_scrollpos") and hasattr(w, "rows_max")
|
|
||||||
|
|
||||||
for w in orig_iter(self):
|
|
||||||
if is_scrolling_widget(w):
|
|
||||||
return w
|
|
||||||
raise ValueError("Not compatible to be wrapped by ScrollBar: %r" % w)
|
|
||||||
|
|
||||||
def keypress(self, size, key):
|
|
||||||
return self._original_widget.keypress(self._original_widget_size, key)
|
|
||||||
|
|
||||||
def mouse_event(self, size, event, button, col, row, focus):
|
|
||||||
ow = self._original_widget
|
|
||||||
ow_size = self._original_widget_size
|
|
||||||
handled = False
|
|
||||||
if hasattr(ow, "mouse_event"):
|
|
||||||
handled = ow.mouse_event(ow_size, event, button, col, row, focus)
|
|
||||||
|
|
||||||
if not handled and hasattr(ow, "set_scrollpos"):
|
|
||||||
if button == 4: # scroll wheel up
|
|
||||||
pos = ow.get_scrollpos(ow_size)
|
|
||||||
ow.set_scrollpos(pos - 1)
|
|
||||||
return True
|
|
||||||
elif button == 5: # scroll wheel down
|
|
||||||
pos = ow.get_scrollpos(ow_size)
|
|
||||||
ow.set_scrollpos(pos + 1)
|
|
||||||
return True
|
|
||||||
|
|
||||||
return False
|
|
||||||
|
|
||||||
|
|
||||||
class Scrollable(urwid.AttrWrap):
|
|
||||||
def __init__(self, *args, **kwargs):
|
|
||||||
super().__init__(
|
|
||||||
_Scrollable(*args, **kwargs), "scrollable", "scrollable-focus"
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
class ScrollBar(urwid.AttrWrap):
|
|
||||||
def __init__(self, *args, **kwargs):
|
|
||||||
super().__init__(
|
|
||||||
_ScrollBar(*args, **kwargs), "scrollbar", "scrollbar-focus"
|
|
||||||
)
|
|
||||||
@@ -0,0 +1,8 @@
|
|||||||
|
[tool.black]
|
||||||
|
line-length = 79
|
||||||
|
py36 = true
|
||||||
|
|
||||||
|
[tool.isort]
|
||||||
|
known_first_party = "pqcli"
|
||||||
|
multi_line_output = 3
|
||||||
|
include_trailing_comma = true
|
||||||
Binary file not shown.
|
After Width: | Height: | Size: 104 KiB |
Binary file not shown.
|
After Width: | Height: | Size: 8.3 KiB |
Binary file not shown.
|
After Width: | Height: | Size: 98 KiB |
BIN
Binary file not shown.
|
Before Width: | Height: | Size: 80 KiB |
@@ -5,7 +5,7 @@ setup(
|
|||||||
author_email="rr-@sakuya.pl",
|
author_email="rr-@sakuya.pl",
|
||||||
name="pqcli",
|
name="pqcli",
|
||||||
long_description="Progress Quest: the CLI edition",
|
long_description="Progress Quest: the CLI edition",
|
||||||
version="0.3",
|
version="1.0",
|
||||||
url="https://github.com/rr-/pq-cli.git",
|
url="https://github.com/rr-/pq-cli.git",
|
||||||
packages=find_packages(),
|
packages=find_packages(),
|
||||||
entry_points={"console_scripts": ["pqcli = pqcli.__main__:main"]},
|
entry_points={"console_scripts": ["pqcli = pqcli.__main__:main"]},
|
||||||
|
|||||||
Reference in New Issue
Block a user