dify/api/tests/unit_tests/controllers/rbac_introspection.py

from collections.abc import Callable

from controllers.common.rbac import RBAC_CHECKS_ATTR, RBACCheck


def rbac_checks(view: Callable[..., object]) -> tuple[RBACCheck, ...]:
    current = view
    while RBAC_CHECKS_ATTR not in current.__dict__:
        wrapped = current.__dict__.get("__wrapped__")
        if wrapped is None:
            raise AssertionError(f"no decorator layer of {view!r} carries an rbac_checks bundle")
        current = wrapped
    return current.__dict__[RBAC_CHECKS_ATTR]


def all_rbac_checks(view: Callable[..., object]) -> list[RBACCheck]:
    found: list[RBACCheck] = []
    seen_bundle_ids: set[int] = set()
    current: Callable[..., object] | None = view
    while current is not None:
        bundle = current.__dict__.get(RBAC_CHECKS_ATTR)
        if bundle is not None and id(bundle) not in seen_bundle_ids:
            seen_bundle_ids.add(id(bundle))
            found.extend(bundle)
        current = current.__dict__.get("__wrapped__")
    return found