royal-mifia/roles/Vigilante.py

from .Role import Role
import strings as s

class Vigilante(Role):
    """Il vigilante può uccidere una persona a sua scelta.
    Possibilmente un mifioso."""
    icon = s.vigilante_icon
    team = 'Good'
    name = s.vigilante_name
    powerdesc = s.vigilante_power_description

    def __init__(self, player):
        super().__init__(player)
        self.target = None
        self.power_was_used = False

    def __repr__(self) -> str:
        if self.target is None:
            return "<Role: Angelo>"
        else:
            return "<Role: Angelo, protecting {target}>".format(target=self.target.tusername)

    def power(self, arg):
        # Imposta qualcuno come bersaglio
        selected = self.player.game.findplayerbyusername(arg)
        if self.power_was_used:
            self.player.message(s.error_no_uses)
            return
        if selected is None:
            self.player.message(s.error_username)
            return
        # Bersaglia il nuovo giocatore selezionato
        selected.protectedby = self.player
        self.target = selected
        self.player.message(s.vigilante_target_selected.format(target=self.target.tusername))

    def onendday(self):
        # Resetta la protezione
        if self.target is not None:
            self.target.kill()
            self.player.game.message()
            self.power_was_used = True
            self.target = None