装饰器的介绍

装饰器的介绍

装饰器的基本使用

import time


def cal_time(fn):
    print('我是外部函数，我被调用了！！！')
    print('fn = {}'.format(fn))

    def inner():
        start = time.time()
        fn()
        end = time.time()
        print('代码耗时', end - start)

    return inner


@cal_time  # 第一件事，调用cal_time；第二件事把被装饰的函数传递给fn
def demo():
    x = 0
    for i in range(1, 100000000):
        x += i
    print(x)


# 第三件事：当再次调用demo函数时，此时的demo已经不再是上面的demo了
print('装饰后的demo ={} '.format(demo))

装饰器详解

import time


def cal_time(fn):
    print('我是外部函数，我被调用了！！！')
    print('fn = {}'.format(fn))

    def inner(x, *args, **kwargs):  # x = 100000000
        start = time.time()
        s = fn(x)  # 100000000
        end = time.time()
        # print('代码耗时', end - start)
        return s, end - start

    return inner


@cal_time  # 第一件事，调用cal_time；第二件事把被装饰的函数传递给fn
def demo(n):
    x = 0
    for i in range(1, n):
        x += i
    return x


m = demo(100000000, 'good', y='hello')
print('m的值是{}'.format(m))

装饰器的使用

def can_play(fn):
    def inner(x, y, *args, **kwargs):
        # print(args)
        # clock = kwargs['clock']
        clock = kwargs.get('clock', 23)
        if clock <= 22:
            fn(x, y)
        else:
            print('太晚了，赶紧睡')

    return inner


@can_play
def play_game(name, game):
    print('{}正在玩{}'.format(name, game))


play_game('张三', '王者荣耀', m='hello', n='good', clock=23)