mirror of
https://github.com/rr-/pq-cli.git
synced 2026-09-28 22:33:59 -06:00
ui/urwid: split to files
This commit is contained in:
rr-
@@ -0,0 +1,4 @@
|
||||
from .confirm_view import ConfirmView
|
||||
from .game_view import GameView
|
||||
from .new_game_view import NewGameView
|
||||
from .roster_view import RosterView
|
||||
@@ -0,0 +1,45 @@
|
||||
import typing as T
|
||||
|
||||
import urwid
|
||||
|
||||
from pqcli.ui.urwid.widgets import CustomButton, CustomLineBox, NPile
|
||||
|
||||
|
||||
class ConfirmView(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,459 @@
|
||||
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.roster import Roster
|
||||
from pqcli.ui.urwid.widgets import (
|
||||
CustomLineBox,
|
||||
CustomProgressBar,
|
||||
DataTable,
|
||||
DoubleLineBox,
|
||||
NColumns,
|
||||
NPile,
|
||||
ReadOnlyCheckBox,
|
||||
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,
|
||||
roster: Roster,
|
||||
player: Player,
|
||||
args: argparse.Namespace,
|
||||
) -> None:
|
||||
self.loop = loop
|
||||
self.roster = roster
|
||||
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()
|
||||
if self.args.use_saves:
|
||||
self.roster.save_periodically()
|
||||
|
||||
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
|
||||
@@ -0,0 +1,218 @@
|
||||
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.urwid.widgets import CustomButton, CustomLineBox, 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:
|
||||
self.stats = self.stats_builder.unroll()
|
||||
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
|
||||
@@ -0,0 +1,114 @@
|
||||
import typing as T
|
||||
|
||||
import urwid
|
||||
|
||||
from pqcli.lingo import act_name, to_roman
|
||||
from pqcli.roster import Roster
|
||||
from pqcli.ui.urwid.widgets import CustomButton, 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
|
||||
Reference in New Issue
Block a user