首先我們看一段代碼

import time
def is_prime(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

def prime_nums():
    t1 = time.time()
    for i in range(2,10000):
        if is_prime(i):
            print(i)
    t2 = time.time()
    print(t2 - t1)

prime_nums()

這段代碼非常容易理解就是找出2-10000中所有的素?cái)?shù)，然后計(jì)算花費(fèi)的時(shí)間，再計(jì)算花費(fèi)時(shí)間。由于不同邏輯混雜在一起，程序的可讀性會大打折扣，于是我們做一些調(diào)整我們把計(jì)算耗時(shí)的功能交給裝飾器來做。

import time
def display_time(func):
    def wrapper():
        t1 = time.time()
        func()
        t2 = time.time()
        print(t2-t1)
    return wrapper


def is_prime(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
def prime_nums():
    for i in range(2,10000):
        if is_prime(i):
            print(i)

prime_nums()

如上面代碼所示 我們寫了一個(gè)display_time()的裝飾器，其中里面的參數(shù)func 就代表我們裝飾的函數(shù)，然后里面還有 一個(gè)函數(shù) wrapper()，用來編寫當(dāng)我們運(yùn)行裝飾器的時(shí)候需要執(zhí)行哪些內(nèi)容。首先運(yùn)行的時(shí)候會先截取一個(gè)時(shí)間，然后運(yùn)行我們裝飾的函數(shù)，最后再截取一個(gè)時(shí)間。這樣一個(gè)簡單的裝飾器就完成了。

另外一種情況就是被修飾的函數(shù)有返回值怎么處理

import time

def display_time(func):
    def wrapper():
        t1 = time.time()
        result = func()
        t2 = time.time()
        print(t2-t1)
        return result
    return wrapper

def is_prime(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
def prime_nums():
    count = 0
    for i in range(2,10000):
        if is_prime(i):
            print(i)
            count+=1
     return count
prime_nums()

上面就是我們實(shí)現(xiàn)的函數(shù)存在返回值 ，裝飾器應(yīng)該如何處理，只需要在裝飾器warpper函數(shù)中，定義一個(gè)值接收代表被裝飾函數(shù)的func返回出來的值，然后再從warpper中返回出來就可以了。

還有一種情況就是，如果被裝飾的函數(shù)存在參數(shù)

import time


def display_time(func):
    def wrapper(*args):
        t1 = time.time()
        result = func(*args)
        t2 = time.time()
        print(t2-t1)
        return result
    return wrapper


def is_prime(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
def prime_nums(maxnum):
    t1 = time.time()
    for i in range(2,maxnum):
        if is_prime(i):
            print(i)
    t2 = time.time()
    print(t2 - t1)
prime_nums(10000)

我們同時(shí)在warpper和func中都傳入了*args，它代表的意思是在不確定被裝飾函數(shù)要傳入的參數(shù)數(shù)量的時(shí)候，用來接收所有被傳入的參數(shù)。