squirrel-battle/squirrelbattle/display/statsdisplay.py

# Copyright (C) 2020 by ÿnérant, eichhornchen, nicomarg, charlse
# SPDX-License-Identifier: GPL-3.0-or-later

import curses

from ..entities.player import Player
from ..game import Game
from ..translations import gettext as _
from .display import Display


class StatsDisplay(Display):
    player: Player

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.pad = self.newpad(self.rows, self.cols)

    def update(self, game: Game) -> None:
        self.player = game.player

    def update_pad(self) -> None:
        string2 = _("player").capitalize() + " -- LVL {}\nEXP {}/{}\nHP {}/{}"\
            .format(self.player.level, self.player.current_xp,
                    self.player.max_xp, self.player.health,
                    self.player.maxhealth)
        self.addstr(self.pad, 0, 0, string2)
        string3 = "STR {}\nINT {}\nCHR {}\nDEX {}\nCON {}\nCRI {}%"\
            .format(self.player.strength,
                    self.player.intelligence, self.player.charisma,
                    self.player.dexterity, self.player.constitution,\
                    self.player.critical)
        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, 9, 0, inventory_str)

        if self.player.equipped_main:
            self.addstr(self.pad, 10, 0,
                        _("Equipped main:") + " "
                        f"{self.pack[self.player.equipped_main.name.upper()]}")
        if self.player.equipped_secondary:
            self.addstr(self.pad, 11, 0,
                        _("Equipped secondary:") + " "
                        f"{self.pack[self.player.equipped_secondary.name.upper()]}")
        if self.player.equipped_armor:
            self.addstr(self.pad, 12, 0,
                        _("Equipped chestplate:") + " "
                        f"{self.pack[self.player.equipped_armor.name.upper()]}")
        if self.player.equipped_helmet:
            self.addstr(self.pad, 13, 0,
                        _("Equipped helmet:") + " "
                        f"{self.pack[self.player.equipped_helmet.name.upper()]}")

        self.addstr(self.pad, 14, 0, f"{self.pack.HAZELNUT} "
                                     f"x{self.player.hazel}")

        if self.player.dead:
            self.addstr(self.pad, 15, 0, _("YOU ARE DEAD"), curses.COLOR_RED,
                        bold=True, blink=True, standout=True)

    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)