Source code for VegansDeluxe.rebuild.Weapons.Axe

from VegansDeluxe.core import AttachedAction, RegisterWeapon
from VegansDeluxe.core import MeleeAttack
from VegansDeluxe.core.Translator.LocalizedString import ls
from VegansDeluxe.core.Weapons.Weapon import MeleeWeapon
from VegansDeluxe.rebuild.States.DamageThreshold import DamageThreshold


@RegisterWeapon
class Axe(MeleeWeapon):
    id = 'axe'
    name = ls("rebuild.weapon.axe.name")
    description = ls("rebuild.weapon.axe.description")

    cubes = 3
    accuracy_bonus = 2
    energy_cost = 2
    damage_bonus = 0


@AttachedAction(Axe)
class AxeAttack(MeleeAttack):
    async def func(self, source, target):
        damage = (await super().attack(source, target)).dealt
        if not damage:
            return damage
        threshold = target.get_state(DamageThreshold)
        self.session.say(ls("rebuild.weapon.axe.effect").format(target.name))

        threshold.threshold -= 1
        return damage