python 装饰器（带参数）

装饰器

装饰器的作用就是不改变原函数的功能的基础上，增加新的功能。多个地方都要验证，就要装饰器。单个就用if。

#!/usr/bin/env python3.6
# -*- coding: utf-8 -*-
# @Time    : 2018/5/9 16:44
# @Author  : ysj
from functools import wraps


def check_login(strict_check=True):
    """strict_check 为真，则检查一个参数>=10,不满足则返回0"""
    def inner(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            if strict_check:
                a, _ = args
                if a >= 10:
                    """balabala 可在返回前任意合适位置加入个性化判断"""
                    return func(*args, **kwargs)
                else:
                    return 0
            else:
                return func(*args, **kwargs)
        return wrapper
    return inner


@check_login()
def add(a, b):
    return a + b


@check_login(False)
def add2(a, b):
    return a + b


assert add(10, 20) == 30
assert add(5, 10) == 0
assert add2(10, 20) == 30
assert add2(5, 10) == 15