Source code for tuxemon.item.conditions.variable

# SPDX-License-Identifier: GPL-3.0
# Copyright (c) 2014-2023 William Edwards <shadowapex@gmail.com>, Benjamin Bean <superman2k5@gmail.com>
from __future__ import annotations

from dataclasses import dataclass
from typing import TYPE_CHECKING, Union

from tuxemon.item.itemcondition import ItemCondition

if TYPE_CHECKING:
    from tuxemon.monster import Monster


@dataclass
class VariableCondition(ItemCondition):
    """Checks against the variables of the context.
    Accepts two parameters; variable name and expected value.
    """

    name = "variable"
    var_name: str
    expected: Union[str, int, None] = None

    def test(self, target: Monster) -> bool:
        var_name = self.var_name
        expect = self.expected

        if type(expect) is str:
            return bool(self.user.game_variables[var_name] == expect)
        elif type(expect) is int:
            return bool(self.user.game_variables[var_name] >= expect)
        else:
            return not self.user.game_variables[var_name]