Source code for VegansDeluxe.rebuild.Weapons.Saw

from VegansDeluxe.core import AttachedAction, RegisterWeapon
from VegansDeluxe.core import EventContext
from VegansDeluxe.core import Next
from VegansDeluxe.core import PostTickGameEvent
from VegansDeluxe.core import RangedAttack
from VegansDeluxe.core.Translator.LocalizedString import ls
from VegansDeluxe.core.Weapons.Weapon import RangedWeapon
from VegansDeluxe.rebuild.States.Injury import Injury


@RegisterWeapon
class Saw(RangedWeapon):
    id = 'saw'
    name = ls("rebuild.weapon.saw.name")
    description = ls("rebuild.weapon.saw.description")

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


@AttachedAction(Saw)
class SawAttack(RangedAttack):
    async def func(self, source, target):
        damage = (await super().attack(source, target)).dealt
        if not damage:
            return damage
        self.session.say(ls("rebuild.weapon.saw.effect")
                         .format(target.name, target.get_state(Injury).injury))

        @Next(self.session.id, PostTickGameEvent)
        async def func(context: EventContext[PostTickGameEvent]):
            injury = target.get_state(Injury)
            injury.injury += 1

        return damage