squirrel-battle/squirrelbattle/display/statsdisplay.py

59 lines
2.1 KiB
Python
Raw Normal View History

2020-11-27 15:33:17 +00:00
# Copyright (C) 2020 by ÿnérant, eichhornchen, nicomarg, charlse
# SPDX-License-Identifier: GPL-3.0-or-later
2020-11-11 00:17:00 +00:00
import curses
2020-11-27 19:42:19 +00:00
from ..entities.player import Player
from ..translations import gettext as _
from .display import Display
2020-11-06 18:53:27 +00:00
2020-11-06 20:15:09 +00:00
class StatsDisplay(Display):
2020-11-10 09:49:11 +00:00
player: Player
2020-11-10 19:34:22 +00:00
2020-11-10 19:43:30 +00:00
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.pad = self.newpad(self.rows, self.cols)
2020-11-11 00:17:00 +00:00
self.init_pair(3, curses.COLOR_RED, curses.COLOR_BLACK)
2020-11-10 19:34:22 +00:00
def update_player(self, p: Player) -> None:
self.player = p
2020-11-06 20:15:09 +00:00
def update_pad(self) -> None:
string2 = "Player -- LVL {}\nEXP {}/{}\nHP {}/{}"\
2020-11-06 20:15:09 +00:00
.format(self.player.level, self.player.current_xp,
self.player.max_xp, self.player.health,
2020-11-07 14:00:24 +00:00
self.player.maxhealth)
self.addstr(self.pad, 0, 0, string2)
string3 = "STR {}\nINT {}\nCHR {}\nDEX {}\nCON {}"\
2020-11-07 14:00:24 +00:00
.format(self.player.strength,
2020-11-06 20:15:09 +00:00
self.player.intelligence, self.player.charisma,
self.player.dexterity, self.player.constitution)
self.addstr(self.pad, 3, 0, string3)
inventory_str = _("Inventory:") + " "
# Stack items by type instead of displaying each item
item_types = [item.name for item in self.player.inventory]
item_types.sort(key=item_types.count, reverse=True)
printed_items = []
for item in item_types:
if item in printed_items:
continue
count = item_types.count(item)
inventory_str += self.pack[item.upper()]
if count > 1:
inventory_str += f"x{count} "
printed_items.append(item)
self.addstr(self.pad, 8, 0, inventory_str)
2020-11-11 00:17:00 +00:00
if self.player.dead:
2020-11-27 19:42:19 +00:00
self.addstr(self.pad, 10, 0, _("YOU ARE DEAD"),
curses.A_BOLD | curses.A_BLINK | curses.A_STANDOUT
| self.color_pair(3))
2020-11-06 20:15:09 +00:00
def display(self) -> None:
self.pad.erase()
self.update_pad()
self.refresh_pad(self.pad, 0, 0, self.y, self.x,
self.y + self.height - 1, self.width + self.x - 1)