Source code for VegansDeluxe.rebuild.States.Injury

from VegansDeluxe.core import StateContext, EventContext
from VegansDeluxe.core import AttackGameEvent
from VegansDeluxe.core import RegisterState, RegisterEvent

from VegansDeluxe.core import Session
from VegansDeluxe.core import State


class Injury(State):
    id = 'injury'

    def __init__(self):
        super().__init__()
        self.injury = 0


@RegisterState(Injury)
async def register(root_context: StateContext[Injury]):
    session: Session = root_context.session
    source = root_context.entity
    state = root_context.state

    @RegisterEvent(session.id, event=AttackGameEvent)
    async def func(context: EventContext[AttackGameEvent]):
        if not state.injury:
            return
        if context.event.target != source:
            return
        if not context.event.damage:
            return
        context.event.damage += state.injury