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


count = count_prime_nums(10000)
print(count)