utils.py

from inspect import getsource
from typing import Any, Callable, Union


def serialize_callable(
    value: Union[Callable, str], keep_callable=False
) -> Union[Callable, str]:
    """Serializes a given function or string.

    If 'keep_callable' is True, the original callable is returned.
    Otherwise, attempts to return the source code of the callable using 'getsource'.
    If serialization fails, returns the string representation.
    """
    if keep_callable:
        return value
    else:
        try:
            return getsource(value)
        except (TypeError, OSError):
            return str(value)


def maybe_serialize(
    val: Union[Callable, Any], keep_callable=False
) -> Union[Callable, Any]:
    """Conditionally serializes a value if it is callable."""

    return (
        serialize_callable(val, keep_callable=keep_callable) if callable(val) else val
    )