Parameter Validator with decorator

parameter-validator-with-decorator.py

def accepts(*types):
    def check_accepts(f):
        assert len(types) == f.func_code.co_argcount

        def new_f(*args, **kwds):
            for (a, t) in zip(args, types):
                assert isinstance(a, t), "arg %r does not match %s" % (a, t)
            return f(*args, **kwds)

        new_f.func_name = f.func_name
        return new_f

    return check_accepts


def par_validator(par_values):
    def check_accepts(f):
        def new_f(*args, **kwds):
            for k, v in par_values.iteritems():
                assert kwds[k] in v
            return f(*args, **kwds)

        new_f.func_name = f.func_name
        return new_f

    return check_accepts


@accepts(int, (int, float), str)
@par_validator({'arg2': [2, 3], 'arg3': ['par3_1', 'par3_2']})
def func(arg1, arg2, arg3):
    return


func(arg1=3, arg2=2, arg3='par3_2')