argment type guard

argguard.py

# -*- coding: utf-8 -*-

import operator

def guard(*guardArgs, **guardKw):
    def _argGuard(f):
        def _inner(*args, **kw):
            argTypes = map(lambda x: type(x), args)
            if not all(map(operator.eq, guardArgs, argTypes)):
                raise Exception("Type Error: not match type {0}, {1}".format(guardArgs, argTypes))
            for key in kw:
                argType = guardKw.get(key, None)
                if argType is None:
                    raise Exception("Type not defined: {0}".format(key))
                if not operator.eq(argType, type(kw[key])):
                    raise Exception("Type Error: not match type {0}".format(key))
            return f(*args, **kw)
        return _inner
    return _argGuard