增加装饰器 以实现下面的效果

def foo(var):
    print(var)
    return var

# 增加装饰器 以实现下面的效果：
# def foo(var):
#     if not var:
#         return
#     print(var)
#     return var

def null_var_decorator(func):
    def wrapper(*args, **kwargs):
        if not kwargs.get('var'):
            return
        return func(*args, **kwargs)
    return wrapper

@null_var_decorator
def foo(var):
    print(var)
    return var

foo()