Source code for ReFreSH.MobileSuit.Decorators.DecoratorUtils

from . import DecoratorNames
from typing import Callable, Optional



[docs]
def get_alias(func: Callable):
    has_attr = hasattr(func, DecoratorNames.suit_alias)
    return getattr(func, DecoratorNames.suit_alias) if has_attr else []




[docs]
def is_ignored(func: Callable):
    check_a = hasattr(func, DecoratorNames.suit_ignored)
    return check_a and getattr(func, DecoratorNames.suit_ignored)




[docs]
def get_info(func: Callable) -> Optional[str]:
    (expr, resObj) = getattr(func, DecoratorNames.suit_info) if hasattr(func, DecoratorNames.suit_info) else (
    None, None)
    if expr is None:
        return None
    return getattr(resObj, expr) if resObj is not None and hasattr(resObj, expr) else expr




[docs]
def get_parser(func: Callable, arg_name: str):
    dic = getattr(func, DecoratorNames.suit_parser) if hasattr(
        func, DecoratorNames.suit_parser) else {}
    return None if arg_name not in dic else dic[arg_name]




[docs]
def get_injected(func: Callable, arg_name: str) -> bool:
    arr = getattr(func, DecoratorNames.suit_injected) if hasattr(
        func, DecoratorNames.suit_injected) else []
    return arg_name in arr