装饰器--decorator2

装饰器加参数

---

import time

def timer(func):  # timer(test2) func = test2

	def deco():

		start_time = time.time()

		func()   # run test1()

		stop_time = time.time()

		print("the action time of the program is {}".format(stop_time-start_time))

	return deco   # 返回了deco的内存地址

@timer         # test2 = timer(test2)=deco

def test2(name):

	time.sleep(2)

	print("the name is {}".format(name))

test2("bigberg") # test2("bigberg") = deco("bigberg")

#输出

Traceback (most recent call last):

  File "G:/python/untitled/study3/decoration4.py", line 17, in <module>

    test2("bigberg")

TypeError: deco() takes 0 positional arguments but 1 was given

1. 我们重新定义一个test2函数，并传入一个参数name，而函数运行报错 deco()没有传入参数。
2. @timer 其实相当于 test2 = timer(test2) = deco
3. test2("bigberg") = deco("bigberg")　

所以我们需要在嵌套函数 deco()中传入一个参数，才能确保程序正确

import time

def timer(func):

	def deco(arg1):

		start_time = time.time()

		func(arg1)

		stop_time = time.time()

		print("the action time of the program is {}".format(stop_time-start_time))

	return deco   # 返回了deco的内存地址

@timer

def test2(name):

	time.sleep(2)

	print("the name is {}".format(name))

test2("bigberg")

#输出

the name is bigberg

the action time of the program is 2.0001132488250732

　　到这里传参已经实现了，但是如果参数个数不固定呢？我们还有非固定参数：

import time

def timer(func):  # timer(test1) func = test1

	def deco(*args,**kwargs):

		start_time = time.time()

		func(*args,**kwargs)   # run test1()

		stop_time = time.time()

		print("the action time of the program is {}".format(stop_time-start_time))

	return deco   # 返回了deco的内存地址

@timer

def test1():

	time.sleep(2)

	print("in the test1.")

@timer

def test2(name,age):

	time.sleep(2)

	print("the name is {} and age is {}".format(name,age))

test1()

test2("bigberg",18)

#输出

in the test1.

the action time of the program is 2.0002853870391846

the name is bigberg and age is 18

the action time of the program is 2.000582218170166

　　由此可见使用非固定参数后，被修饰函数有没有参数都可以正常运行了。