常见的装饰器

最基本的装饰器

# 不带参数的装饰器
def decorate(func):

    @wraps(func)
    def wrapper(*args, **kwargs):
        # 执行函数前需要做的事
        
        result = func(*args, **kwargs)
        # 执行函数后需要做的事，最后返回结果
        
        return result
    return wrapper

带参数的装饰器

import time
from functools import wraps


# 参数的装饰器
def log_time(output):

    def decorate(func):

        @wraps(func)
        def wrapper(*args, **kwargs):
            begin = time.time()
            result = func(*args, **kwargs)
            output(func.__name__, time.time() - begin)
            return result

        return wrapper

    return decorate


def log_to_console(fname, duration):
    print(f'{fname}执行时间：{duration}秒')


def log_to_file(fname, duration):
    with open('recode.txt', 'a') as file_stream:
        file_stream.write(f'{fname}执行时间：{duration}秒')


@log_time(log_to_console)
def foo():
    print('hello, world!')
    time.sleep(3)


def main():
    print(foo.__name__)
    foo()
    # 取消装饰器
    foo.__wrapped__()


if __name__ == '__main__':
    main()

计算代码运行的时间

import time
from functools import wraps


# 不带参数的装饰器
def decorate(func):

    @wraps(func)
    def wrapper(*args, **kwargs):
        begin = time.time()
        result = func(*args, **kwargs)
        print(f'执行时间：{time.time() - begin}秒')
        return result

    return wrapper

@decorate
def foo():
    print('hello')
    # 取消装饰器
    foo.__wrapped__()

验证是否登录

Try

def is_login(func):
    @wraps(func)
    def check_login(*args, **kwargs):
        try:
            session['user_id']
        except:
            return redirect(url_for('user.login'))
        return func(*args, **kwargs)
    return check_login

If

def is_login(func):
    @wraps(func)
    def check_login(*args, **kwargs):
        if not session.get('user_id'):
            return redirect(url_for('user.login'))
        return func(*args, **kwargs)
    return check_login