mirror of
https://github.com/rr-/pq-cli.git
synced 2026-09-29 14:54:00 -06:00
142 lines
4.4 KiB
Python
142 lines
4.4 KiB
Python
import random
|
|
import typing as T
|
|
|
|
import urwid
|
|
import urwid_readline
|
|
|
|
from pqcli.game_state import Player, StatsBuilder, generate_name, create_player
|
|
from pqcli.config import RACES, CLASSES, PRIME_STATS
|
|
|
|
|
|
class StatsGroupBox(urwid.Pile):
|
|
def __init__(self, player: Player, stats_builder: StatsBuilder) -> None:
|
|
self.player = player
|
|
self.stats_builder = stats_builder
|
|
self.stat_labels = {stat: urwid.Text("0") for stat in PRIME_STATS}
|
|
self.total_label = urwid.Text("0")
|
|
self.update_values()
|
|
|
|
roll_button = urwid.Button("Roll", on_press=self.on_roll_press)
|
|
unroll_button = urwid.Button("Unroll", on_press=self.on_unroll_press)
|
|
|
|
value_texts = list(self.stat_labels.values()) + [self.total_label]
|
|
label_texts = [
|
|
urwid.Text(f"{stat.value}: ") for stat in PRIME_STATS
|
|
] + [urwid.Text("Total: ")]
|
|
|
|
super().__init__(
|
|
[
|
|
urwid.Columns(
|
|
[
|
|
(8, urwid.ListBox(label_texts)),
|
|
urwid.ListBox(value_texts),
|
|
]
|
|
),
|
|
urwid.Filler(roll_button),
|
|
urwid.Filler(unroll_button),
|
|
]
|
|
)
|
|
|
|
def on_roll_press(self, _user_data: T.Any) -> None:
|
|
self.player.stats = self.stats_builder.roll()
|
|
self.update_values()
|
|
|
|
def on_unroll_press(self, _user_data: T.Any) -> None:
|
|
try:
|
|
self.player.stats = self.stats_builder.unroll()
|
|
except IndexError:
|
|
return
|
|
self.update_values()
|
|
|
|
def update_values(self) -> None:
|
|
for stat, edit in self.stat_labels.items():
|
|
edit.set_text(str(self.player.stats[stat]))
|
|
self.total_label.set_text(
|
|
str(
|
|
sum(
|
|
self.player.stats[stat] for stat in self.stat_labels.keys()
|
|
)
|
|
)
|
|
)
|
|
|
|
|
|
class NewGameView(urwid.Pile):
|
|
def __init__(self, on_confirm: T.Callable, on_cancel: T.Callable) -> None:
|
|
self.on_confirm = on_confirm
|
|
self.on_cancel = on_cancel
|
|
|
|
stats_builder = StatsBuilder()
|
|
self.player = create_player(stats_builder)
|
|
|
|
race_checkboxes = []
|
|
for race in RACES:
|
|
urwid.RadioButton(
|
|
group=race_checkboxes,
|
|
label=race.name,
|
|
state=race.name == self.player.race.name,
|
|
)
|
|
|
|
class_checkboxes = []
|
|
for class_ in CLASSES:
|
|
urwid.RadioButton(
|
|
group=class_checkboxes,
|
|
label=class_.name,
|
|
state=class_.name == self.player.class_.name,
|
|
)
|
|
|
|
self.char_name_edit = urwid_readline.ReadlineEdit(
|
|
"Name: ", self.player.name
|
|
)
|
|
generate_char_name_btn = urwid.Button(
|
|
"?", on_press=self.on_generate_char_name_press
|
|
)
|
|
|
|
urwid.connect_signal(
|
|
self.char_name_edit, "postchange", self.on_char_name_change
|
|
)
|
|
|
|
super().__init__(
|
|
[
|
|
(
|
|
1,
|
|
urwid.Columns(
|
|
[
|
|
urwid.Filler(self.char_name_edit),
|
|
urwid.Filler(generate_char_name_btn),
|
|
]
|
|
),
|
|
),
|
|
urwid.Columns(
|
|
[
|
|
urwid.LineBox(
|
|
urwid.ListBox(race_checkboxes), title="Race"
|
|
),
|
|
urwid.LineBox(
|
|
urwid.ListBox(class_checkboxes), title="Class"
|
|
),
|
|
urwid.LineBox(
|
|
StatsGroupBox(self.player, stats_builder),
|
|
title="Stats",
|
|
),
|
|
]
|
|
),
|
|
]
|
|
)
|
|
|
|
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 on_char_name_change(self, widget: urwid.Text, old_text: str) -> None:
|
|
self.player.name = widget.edit_text
|
|
|
|
def on_generate_char_name_press(self, _user_data: T.Any) -> None:
|
|
self.generate_random_char_name()
|
|
|
|
def unhandled_input(self, key: str) -> bool:
|
|
if key == "esc":
|
|
self.on_cancel()
|
|
return True
|
|
|
|
return False
|