MegEngine/imperative/python/megengine/utils/deprecation.py

import importlib
import warnings
from functools import wraps

from deprecated.sphinx import deprecated

warnings.filterwarnings(action="default", module="megengine")


def deprecated_func(version, origin, name, tbd):
    r"""

    Args:
        version: version to deprecate this function
        origin: origin module path
        name: function name
        tbd: to be discussed, if true, ignore warnings
    """
    should_warning = not tbd
    module = importlib.import_module(origin)
    func = module.__getattribute__(name)

    @wraps(func)
    def wrapper(*args, **kwargs):
        nonlocal should_warning
        if should_warning:
            warnings.warn(
                "Call to deprecated function {}. (use {}.{} instead) -- Deprecated since version {}.".format(
                    name, origin, name, version
                ),
                category=DeprecationWarning,
                stacklevel=2,
            )
        return func(*args, **kwargs)

    wrapper.__deprecated__ = True
    return wrapper


def deprecated_kwargs_default(version, kwargs_name, kwargs_pos):
    r"""
    Args:
        version: version to deprecate this default
        kwargs_name: kwargs name
        kwargs_pos: kwargs position
    """

    def deprecated(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            if len(args) < kwargs_pos and kwargs_name not in kwargs:
                warnings.warn(
                    "the default behavior for {} will be changed in version {}, please use it in keyword parameter way".format(
                        kwargs_name, version
                    ),
                    category=PendingDeprecationWarning,
                    stacklevel=2,
                )
            return func(*args, **kwargs)

        wrapper.__deprecated__ = True
        return wrapper

    return deprecated