python -- 装饰器

干嘛非要把一个简单的语法搞得这么复杂呢？看的教学是好多函数里面套函数，还加一些参数改变，还解析的那么“详细”，好像很厉害的样子。

嘻嘻，我也学习一下：

import time

def display_time(func):
    def wrapper(*args):
        t1 = time.time()
        result = func(*args)
        t2 = time.time()
        print('Total time: {:.4} s'.format(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 count_prime_nums(maxnum):
    count = 0
    for i in range(2, maxnum):
        if is_prime(i):
            count = count + 1
    return count

count = count_prime_nums(10000)
print(count)