Source code for python2verilog.utils.decorator

"""
Special decorator primatives
"""

from functools import wraps
from types import FunctionType




def decorator_with_args(func):
    """
    a decorator decorator, allowing the decorator to be used as:
    @decorator(with, arguments, and=kwargs)
    or
    @decorator

    Note: can't distinguish between a function as a parameter vs
    a function to-be decorated
    """

    @wraps(func)
    def new_dec(*args, **kwargs):
        if len(args) == 1 and len(kwargs) == 0 and isinstance(args[0], FunctionType):
            # actual decorated function
            return func(args[0])
        # decorator arguments
        return lambda real_func: func(real_func, *args, **kwargs)

    return new_dec