Function _get_decorator_bool_argument

mypy/plugins/common.py:55–63

Return the bool argument for the decorator. This handles both @decorator(...) and @decorator.

(ctx: ClassDefContext, name: str, default: bool)

def _get_decorator_bool_argument(ctx: ClassDefContext, name: str, default: bool) -> bool:
    """Return the bool argument for the decorator.

    This handles both @decorator(...) and @decorator.
    """
    if isinstance(ctx.reason, CallExpr):
        return _get_bool_argument(ctx, ctx.reason, name, default)
    else:
        return default


def _get_bool_argument(ctx: ClassDefContext, expr: CallExpr, name: str, default: bool) -> bool: