装饰器函数与类装饰器的使用

一、装饰器

装饰器：装饰函数和类，作用：扩展增强函数和类的功能

二、装饰器的分类

两大类：装饰器函数和装饰器类

三、装饰器函数定义及应用

函数：封装代码最小单元，提供代码复用性
装饰器函数利用函数的一些特征：
1、函数可以作为参数
3、函数可以作为变量
2、函数也可以返回函数
装饰器函数实质：调用装饰函数的内置方法

四、装饰器函数的使用

import time

def display_time(fun):
    def wrapper():
        t1 = time.time()
        resutl =fun()
        t2 = time.time()
        print("总共耗时{:.4} s".format(t2 - t1))
        return resutl
    return wrapper

def is_price(num):
    if num < 2:
        return False
    elif num == 2:
        return True
    else:
        for i in range(2, num):
            if num % i == 0:
                return False
        return True

'''
相比前面的，多了返回值,
需要在display_time函数中加fun()返回结果
'''
@display_time
def count_prime_number():
    count=0
    for i in range(2, 1000):
        if is_price(i):
            count=count+1
    return count


counts=count_prime_number()
print(counts)

五、类装饰器的使用

import time
# 定义装饰类‐‐‐‐‐》本质类 作用：原来功能+扩展功能
class A:
    # 原来功能 self._func()
    def __init__(self, func):
        self._func = func
    # 扩展功能
    def __call__(self, *args, **kwargs):
        """
        __call__函数：实例化对象()
        :return:
        """
        start = time.time()
        # 原来功能
        result=self._func(*args, **kwargs)  # 原来功能
        end = time.time()

        print("总共耗时{:.4} s".format(end - start))
        return result

@A
def count_prime_number(maxnumber):
    time.sleep(1)
    sum=0
    for i in range(2, maxnumber):
            sum=sum+i
    return sum

sums=count_prime_number(10000)
print(sums)