当前位置:   article > 正文

Python 装饰器 函数

python装饰器 从键盘输入一串字符串,定义一个输出函数 例如out函数

Python装饰器学习(九步入门):http://www.cnblogs.com/rhcad/archive/2011/12/21/2295507.html
浅谈Python装饰器:https://blog.csdn.net/mdl13412/article/details/22608283
Python装饰器与面向切面编程http://www.cnblogs.com/huxi/archive/2011/03/01/1967600.html

一些更加实用的 Python 装饰器示例https://sikasjc.github.io/2018/09/29/somedecorator/

Python装饰器高级版—Python类内定义装饰器并传递self参数:https://blog.51cto.com/yishi/2354752

Python 装饰器学习(九步入门)

这是在 Python学习小组上介绍的内容,现学现卖、多练习是好的学习方式。

第一步:最简单的函数,准备附加额外功能

示例代码:

  1. # -*- coding:gbk -*-
  2. '''示例1: 最简单的函数,表示调用了两次'''
  3. def deco(func):
  4.     print("before myfunc() called.")
  5.     func()
  6.     print("after myfunc() called.")
  7.     return func
  8. def myfunc():
  9.     print("     myfunc() called.")
  10. print('*****************************')
  11. dec_func = deco(myfunc)
  12. print('*****************************')
  13. dec_func()
  14. """
  15. 结果:
  16. *****************************
  17. before myfunc() called.
  18.      myfunc() called.
  19. after myfunc() called.
  20. *****************************
  21.      myfunc() called.
  22. """

第二步:使用装饰函数在函数执行前和执行后分别附加额外功能

示例代码:

  1. # -*- coding:gbk -*-
  2. '''示例2: 替换函数(装饰)
  3. 装饰函数的参数是被装饰的函数对象,返回原函数对象
  4. 装饰的实质语句: myfunc = deco(myfunc)'''
  5. def deco(func):
  6.     print("before myfunc() called.")
  7.     func()
  8.     print("after myfunc() called.")
  9.     return func
  10. def myfunc():
  11.     print("    myfunc() called.")
  12. dec_func = deco(myfunc)
  13. print('****************************')
  14. dec_func()
  15. dec_func()
  16. """
  17. 结果:
  18. before myfunc() called.
  19.     myfunc() called.
  20. after myfunc() called.
  21. ****************************
  22.     myfunc() called.
  23.     myfunc() called.
  24. """

第三步:使用语法糖@来装饰函数

本例中 deco 没有使用内嵌函数,可以看到第一次执行可以进入到装饰函数,但是第二次不会进入装饰函数

  1. # -*- coding:gbk -*-
  2. '''示例3: 使用语法糖@来装饰函数,相当于“myfunc = deco(myfunc)”
  3. 但发现新函数只在第一次被调用,且原函数多调用了一次'''
  4. def deco(func):
  5.     print("before myfunc() called.")
  6.     func()
  7.     print("after myfunc() called.")
  8.     return func
  9. @deco
  10. def myfunc():
  11.     print("    myfunc() called.")
  12. # 第一次调用后,返回的是 deco 里面的 func 对象,
  13. # 所以第二次调用myfunc() 只输出 myfunc() called.
  14. myfunc()  # 第一次调用
  15. print('************************')
  16. myfunc()  # 第二次调用
  17. """
  18. 结果:
  19. before myfunc() called.
  20.     myfunc() called.
  21. after myfunc() called.
  22.     myfunc() called.
  23. ************************
  24.     myfunc() called.
  25. """

第四步:使用 内嵌包装函数 来确保 每次 新函数 都被调用( 被装饰的函数没有参数 )

装饰函数 deco 返回内嵌包装函数对象 _deco。使用内嵌包装函数来确保每次新函数都被调用

  1. # -*- coding:gbk -*-
  2. '''示例4: 使用内嵌包装函数来确保每次新函数都被调用,
  3. 内嵌包装函数的形参和返回值与原函数相同,装饰函数返回内嵌包装函数对象'''
  4. def deco(func):
  5.     def _deco():
  6.         print("before myfunc() called.")
  7.         func()
  8.         print("after myfunc() called.")
  9.         # 不需要返回func,实际上应返回原函数的返回值
  10.     return _deco
  11. @deco
  12. def myfunc():
  13.     print("    myfunc() called.")
  14.     return 'ok'
  15. myfunc()
  16. print('*************************')
  17. myfunc()
  18. """
  19. 执行结果:
  20. before myfunc() called.
  21.     myfunc() called.
  22. after myfunc() called.
  23. *************************
  24. before myfunc() called.
  25.     myfunc() called.
  26. after myfunc() called.
  27. """

示例代码:无参数的函数( 被装饰的函数没有参数 )

  1. # decorator.py
  2. from time import ctime, sleep
  3. def time_fun(func):
  4. def wrapped_func():
  5. print(f"{func.__name__} called at {ctime()}")
  6. return func()
  7. return wrapped_func
  8. @time_fun
  9. def foo():
  10. pass
  11. foo()
  12. sleep(2)
  13. foo()

第五步:对 带参数的函数 进行装饰( 被装饰的函数带参数  )

示例代码:

  1. # -*- coding:gbk -*-
  2. '''示例5: 对带参数的函数进行装饰,
  3. 内嵌包装函数的形参和返回值与原函数相同,装饰函数返回内嵌包装函数对象'''
  4. def deco(func):
  5. def _deco(a, b):
  6. print("before myfunc() called.")
  7. ret = func(a, b)
  8. print("after myfunc() called. result: %s" % ret)
  9. return ret
  10. return _deco
  11. @deco
  12. def myfunc(a, b):
  13. print(" myfunc(%s,%s) called." % (a, b))
  14. return a + b
  15. myfunc(1, 2)
  16. print('*************************')
  17. myfunc(3, 4)
  18. """
  19. 执行结果:
  20. before myfunc() called.
  21. myfunc(1,2) called.
  22. after myfunc() called. result: 3
  23. *************************
  24. before myfunc() called.
  25. myfunc(3,4) called.
  26. after myfunc() called. result: 7
  27. """

示例代码:被装饰的函数带参数 

  1. # decorator2.py
  2. from time import ctime, sleep
  3. def time_fun(func):
  4. def wrapped_func(a, b):
  5. print(f"{func.__name__} called at {ctime()}")
  6. print(a, b)
  7. return func(a, b)
  8. return wrapped_func
  9. @time_fun
  10. def foo(a, b):
  11. print(a + b)
  12. foo(3, 5)
  13. sleep(2)
  14. foo(2, 4)

第六步:对 参数数量不确定 的函数进行装饰

示例代码:

  1. # -*- coding:gbk -*-
  2. '''示例6: 对参数数量不确定的函数进行装饰,
  3. 参数用(*args, **kwargs),自动适应变参和命名参数'''
  4. def deco(func):
  5.     def _deco(*args, **kwargs):
  6.         print("before %s called." % func.__name__)
  7.         ret = func(*args, **kwargs)
  8.         print("after %s called. result: %s" % (func.__name__, ret))
  9.         return ret
  10.     return _deco
  11. @deco
  12. def myfunc_1(a, b):
  13.     print("    myfunc_1(%s,%s) called." % (a, b))
  14.     return a + b
  15. @deco
  16. def myfunc_2(a, b, c):
  17.     print("    myfunc_2(%s,%s,%s) called." % (a, b, c))
  18.     return a + b + c
  19. print('*' * 30)
  20. myfunc_1(1, 2)
  21. print('*' * 30)
  22. myfunc_1(3, 4)
  23. print('*' * 30)
  24. myfunc_2(1, 2, 3)
  25. print('*' * 30)
  26. myfunc_2(3, 4, 5)
  27. """
  28. 执行结果:
  29. ******************************
  30. before myfunc_1 called.
  31.     myfunc_1(1,2) called.
  32. after myfunc_1 called. result: 3
  33. ******************************
  34. before myfunc_1 called.
  35.     myfunc_1(3,4) called.
  36. after myfunc_1 called. result: 7
  37. ******************************
  38. before myfunc_2 called.
  39.     myfunc_2(1,2,3) called.
  40. after myfunc_2 called. result: 6
  41. ******************************
  42. before myfunc_2 called.
  43.     myfunc_2(3,4,5) called.
  44. after myfunc_2 called. result: 12
  45. """

第七步:让 装饰器 带 参数

示例代码:

  1. # -*- coding:gbk -*-
  2. '''示例7: 在示例4的基础上,让装饰器带参数,
  3. 和上一示例相比在外层多了一层包装。
  4. 装饰函数名实际上应更有意义些'''
  5. def deco(arg):
  6.     def _deco(func):
  7.         def __deco():
  8.             print("before %s called [%s]." % (func.__name__, arg))
  9.             func()
  10.             print("after %s called [%s]." % (func.__name__, arg))
  11.         return __deco
  12.     return _deco
  13. @deco("mymodule")
  14. def myfunc():
  15.     print("    myfunc_1() called.")
  16. @deco("module2")
  17. def myfunc2():
  18.     print("    myfunc_2() called.")
  19. print('************************************')
  20. myfunc()
  21. print('************************************')
  22. myfunc2()
  23. """
  24. 执行结果:
  25. ************************************
  26. before myfunc called [mymodule].
  27.     myfunc_1() called.
  28. after myfunc called [mymodule].
  29. ************************************
  30. before myfunc2 called [module2].
  31.     myfunc_2() called.
  32. after myfunc2 called [module2].
  33. """

装饰器带参数,在原有装饰器的基础上,设置外部变量

  1. from time import ctime, sleep
  2. def time_fun_arg(pre="hello"):
  3. def time_fun(func):
  4. def wrapped_func():
  5. print("%s called at %s %s"%(func.__name__, ctime(), pre))
  6. return func()
  7. return wrapped_func
  8. return time_fun
  9. @time_fun_arg("12345")
  10. def foo():
  11. pass
  12. @time_fun_arg("abcde")
  13. def too():
  14. pass
  15. foo()
  16. sleep(2)
  17. foo()
  18. too()
  19. sleep(2)
  20. too()

第八步:让 装饰器 带 类 参数

示例代码:

  1. # -*- coding:gbk -*-
  2. '''示例8: 装饰器带类参数'''
  3. class Locker:
  4.     def __init__(self):
  5.         print("locker.__init__() should be not called.")
  6.     @staticmethod
  7.     def acquire():
  8.         print("locker.acquire() called.(这是静态方法)")
  9.     @staticmethod
  10.     def release():
  11.         print("locker.release() called.(不需要对象实例)")
  12. def deco(cls):
  13.     '''cls 必须实现acquire和release静态方法'''
  14.     def _deco(func):
  15.         def __deco():
  16.             print("before %s called [%s]." % (func.__name__, cls))
  17.             cls.acquire()
  18.             try:
  19.                 return func()
  20.             finally:
  21.                 cls.release()
  22.         return __deco
  23.     return _deco
  24. @deco(Locker)
  25. def myfunc():
  26.     print("    myfunc() called.")
  27. print('*********************************************')
  28. myfunc()
  29. print('*********************************************')
  30. myfunc()
  31. """
  32. 执行结果:
  33. *********************************************
  34. before myfunc called [<class '__main__.Locker'>].
  35. locker.acquire() called.(这是静态方法)
  36.     myfunc() called.
  37. locker.release() called.(不需要对象实例)
  38. *********************************************
  39. before myfunc called [<class '__main__.Locker'>].
  40. locker.acquire() called.(这是静态方法)
  41.     myfunc() called.
  42. locker.release() called.(不需要对象实例)
  43. """

装饰器 和 闭包 混用:

  1. # coding=utf-8
  2. from time import time
  3. def logged(when):
  4. def log(f, *args, **kargs):
  5. print("fun:%s args:%r kargs:%r" % (f, args, kargs))
  6. # %r字符串的同时,显示原有对象类型
  7. def pre_logged(f):
  8. def wrapper(*args, **kargs):
  9. log(f, *args, **kargs)
  10. return f(*args, **kargs)
  11. return wrapper
  12. def post_logged(f):
  13. def wrapper(*args, **kargs):
  14. now = time()
  15. try:
  16. return f(*args, **kargs)
  17. finally:
  18. log(f, *args, **kargs)
  19. print("time delta: %s" % (time() - now))
  20. return wrapper
  21. try:
  22. return {"pre": pre_logged, "post": post_logged}[when]
  23. except BaseException as e:
  24. print('must be "pre" or "post"')
  25. raise e
  26. @logged("post")
  27. def fun(name):
  28. print("Hello, ", name)
  29. fun("world!")

第九步:装饰器带类参数,并分拆公共类到其他py文件中,

同时 演示了对一个函数应用多个装饰器

mylocker.py

  1. # -*- coding:gbk -*-
  2. class MyLocker:
  3.     def __init__(self):
  4.         print("mylocker.__init__() called.")
  5.     @staticmethod
  6.     def acquire():
  7.         print("mylocker.acquire() called.")
  8.     @staticmethod
  9.     def unlock():
  10.         print("mylocker.unlock() called.")
  11. class LockerEx(MyLocker):
  12.     @staticmethod
  13.     def acquire():
  14.         print("lockerex.acquire() called.")
  15.     @staticmethod
  16.     def unlock():
  17.         print("lockerex.unlock() called.")
  18. def lock_helper(cls):
  19.     """cls 必须实现acquire和release静态方法"""
  20.     def _deco(func):
  21.         def __deco(*args, **kwargs):
  22.             print("before %s called." % func.__name__)
  23.             cls.acquire()
  24.             try:
  25.                 return func(*args, **kwargs)
  26.             finally:
  27.                 cls.unlock()
  28.         return __deco
  29.     return _deco

测试代码:

  1. # -*- coding:gbk -*-
  2. """
  3. 示例 9: 装饰器带类参数,并分拆公共类到其他py文件中
  4. 同时演示了对一个函数应用多个装饰器
  5. """
  6. class Example:
  7. @lock_helper(MyLocker)
  8. def func_1(self):
  9. print("\nfunc_1() called.")
  10. @lock_helper(MyLocker)
  11. @lock_helper(LockerEx)
  12. def func_2(self, a, b):
  13. print("\nfunc_2() called.")
  14. return a + b
  15. if __name__ == "__main__":
  16. a = Example()
  17. a.func_1()
  18. print(a.func_1())
  19. print(a.func_2(1, 2))
  20. print(a.func_2(3, 4))
  21. """
  22. 执行结果:
  23. before func_1 called.
  24. mylocker.acquire() called.
  25. func_1() called.
  26. mylocker.unlock() called.
  27. before func_1 called.
  28. mylocker.acquire() called.
  29. func_1() called.
  30. mylocker.unlock() called.
  31. None
  32. before __deco called.
  33. mylocker.acquire() called.
  34. before func_2 called.
  35. lockerex.acquire() called.
  36. func_2() called.
  37. lockerex.unlock() called.
  38. mylocker.unlock() called.
  39. 3
  40. before __deco called.
  41. mylocker.acquire() called.
  42. before func_2 called.
  43. lockerex.acquire() called.
  44. func_2() called.
  45. lockerex.unlock() called.
  46. mylocker.unlock() called.
  47. 7
  48. """

下面是参考资料,当初有不少地方没看明白,真正练习后才明白些:

1. Python装饰器学习:http://blog.csdn.net/thy38/article/details/4471421

2. Python装饰器与面向切面编程:http://www.cnblogs.com/huxi/archive/2011/03/01/1967600.html

3. Python装饰器的理解:http://apps.hi.baidu.com/share/detail/17572338

Python 装饰器的 4 种类型

  • 函数  装饰  函数
  • 函数  装饰 
  •   装饰  函数
  •   装饰 

@wraps(func) 要使用这个必须导入functools,这个作用是消除(被装饰后的函数名等属性的改变)副作用

参考:https://blog.csdn.net/freeking101/article/details/109662542

  1. from functools import wraps
  2. def decorate(src_func):
  3. @wraps(src_func)
  4. def wrapper():
  5. print("hello")
  6. src_func()
  7. print("world")
  8. return wrapper
  9. @decorate
  10. def func():
  11. print("2019-12-31")
  12. print(func.__name__, func.__doc__)
  13. # 加上 @wraps(src_func) 输出为func None
  14. # 不加 输出为 wrapper None

一:函数 装饰 函数

  1. def wrapFun(func):
  2. def inner(a, b):
  3. print('function name:', func.__name__)
  4. r = func(a, b)
  5. return r
  6. return inner
  7. @wrapFun
  8. def myadd(a, b):
  9. return a + b
  10. print(myadd(2, 3))

装饰不带参数的函数

  1. # -*- coding: utf-8 -*-
  2. def clothes(func):
  3. def wear():
  4. print('Buy clothes!{}'.format(func.__name__))
  5. return func()
  6. return wear
  7. @clothes
  8. def body():
  9. print('The body feels cold!')
  10. body()
  11. #备注:@是语法糖
  12. # 不用语法糖的情况下,使用下面语句也能实现装饰作用:把body再加工,再传给body
  13. # body = clothes(body)

示例 :

  1. def deco(func):
  2. def _deco(*args, **kwargs):
  3. print('call deco')
  4. func(*args, **kwargs)
  5. return _deco
  6. @deco
  7. def test():
  8. print('call test')
  9. # 等同于
  10. # def test():
  11. # print('call test')
  12. # test = deco(func)

装饰带一个参数的函数

  1. # -*- coding: utf-8 -*-
  2. def clothes(func):
  3. def wear(anything): # 实际是定义一个anything参数,对应body函数参数
  4. print('Buy clothes!{}'.format(func.__name__))
  5. return func(anything) # 执行func函数,并传入调用传入的anything参数
  6. # wear = func(anything) # 在这一步,实际上可以看出来,为啥wear必须带参数,因为它就是func(anything)
  7. return wear
  8. # 所以clothes的结果是
  9. # clothes = wear = func(anything)
  10. # 不用语法糖的情况下就是
  11. # body = clothes(body)('hands')
  12. # 进一步证明:print(body.__name__) 显示的是wear函数
  13. @clothes
  14. def body(part):
  15. print('The body feels could!{}'.format(part))
  16. body('hands')

装饰带不定长参数的函数

通常装饰器不只装饰一个函数,每个函数参数的个数也不相同,这个时候使用不定长参数*args,**kwargs

  1. def clothes(func):
  2. def wear(*args, **kwargs):
  3. print('Buy clothes!{}'.format(func.__name__))
  4. return func(*args, **kwargs)
  5. return wear
  6. @clothes
  7. def body(part):
  8. print('The body feels could!{}'.format(part))
  9. @clothes
  10. def head(head_wear, num=2):
  11. print('The head need buy {} {}!'.format(num, head_wear))
  12. body('hands')
  13. head('headdress')

装饰器带参数

  1. # 把装饰器再包装,实现了seasons传递装饰器参数。
  2. def seasons(season_type):
  3. def clothes(func):
  4. def wear(*args, **kwargs):
  5. if season_type == 1:
  6. s = 'spring'
  7. elif season_type == 2:
  8. s = 'summer'
  9. elif season_type == 3:
  10. s = 'autumn'
  11. elif season_type == 4:
  12. s = 'winter'
  13. else:
  14. print('The args is error!')
  15. return func(*args, **kwargs)
  16. print('The season is {}!{}'.format(s, func.__name__))
  17. return func(*args, **kwargs)
  18. return wear
  19. return clothes
  20. @seasons(2)
  21. def children():
  22. print('i am children')

示例:

  1. def deco(*args, **kwargs):
  2. def _deco(func):
  3. print(args, kwargs)
  4. def __deco(*args, **kwargs):
  5. print('call deco')
  6. func(*args, **kwargs)
  7. return __deco
  8. return _deco
  9. @deco('hello', x='ni hao')
  10. def test():
  11. print('call test')
  12. # 等同于
  13. # def test():
  14. # print('call test')
  15. # test = deco('hello', x='ni hao')(test)

二:函数 装饰 类

示例:

  1. def deco(*args, **kwargs):
  2. def _deco(cls):
  3. cls.x = 12
  4. return cls
  5. return _deco
  6. @deco('hello')
  7. class A(object):
  8. pass
  9. >>> A.x
  10. 12
  11. # 等同于
  12. # class A(object):
  13. # pass
  14. # A = deco('hello')(A)

示例:

  1. def wrapper_class(cls):
  2. def inner(a):
  3. print('class name:', cls.__name__)
  4. return cls(a)
  5. return inner
  6. @wrapper_class
  7. class Foo(object):
  8. def __init__(self, a):
  9. self.a = a
  10. def fun(self):
  11. print('self.a =', self.a)
  12. m = Foo('Are you OK!')
  13. m.fun()

示例:

创建单例(Singletons)

单例是一个只有一个实例的类。Python中有几种常用的单例,包括NoneTrueFalse。事实上,None是一个单例,允许你使用is关键字比较None

示例:下面的 @singleton 装饰器将类的第一个实例存储为属性,从而将类转换为单例对象。稍后创建实例的尝试只是返回存储的实例:

  1. import functools
  2. def singleton(cls):
  3. """Make a class a Singleton class (only one instance)"""
  4. @functools.wraps(cls)
  5. def wrapper_singleton(*args, **kwargs):
  6. if not wrapper_singleton.instance:
  7. wrapper_singleton.instance = cls(*args, **kwargs)
  8. return wrapper_singleton.instance
  9. wrapper_singleton.instance = None
  10. return wrapper_singleton
  11. @singleton
  12. class TheOne:
  13. pass

如上所示,这个类装饰器遵循与我们的函数装饰器相同的模板。唯一的区别是,我们使用cls作为参数名来表示它是一个类装饰器,而不是func

运行效果:

  1. >>> first_one = TheOne()
  2. >>> another_one = TheOne()
  3. >>> id(first_one)
  4. 140094218762280
  5. >>> id(another_one)
  6. 140094218762280
  7. >>> first_one is another_one
  8. True

很明显,first_one确实与another_one完全相同。

在Python中,单例类的使用并不像在其他语言中那样频繁。单例的效果通常在模块中作为全局变量更好地实现。

三: 装饰 函数方法

装饰 函数

定义一个类装饰器,装饰函数,默认调用 __call__ 方法

  1. class Decorator(object):
  2. def __init__(self, func): # 传送的是test方法
  3. self.func = func
  4. def __call__(self, *args, **kwargs): # 接受任意参数
  5. print('函数调用CALL')
  6. return self.func(*args, **kwargs) # 适应test的任意参数
  7. @Decorator # 如果带参数,init中的func是此参数。
  8. def test(hh):
  9. print('this is the test of the function !', hh)
  10. test('hh')

示例:

  1. class ShowFunName(object):
  2. def __init__(self, func):
  3. self._func = func
  4. def __call__(self, a):
  5. print('function name:', self._func.__name__)
  6. return self._func(a)
  7. @ShowFunName
  8. def bar(a):
  9. return a
  10. print(bar('Are you OK'))

无参数

  1. class Deco(object):
  2. def __init__(self, func):
  3. self.func = func
  4. def __call__(self, *args, **kwargs):
  5. print('call Deco')
  6. self.func(*args, **kwargs)
  7. @Deco
  8. def test():
  9. print('call test')
  10. # 等同于
  11. test = Deco(test)

有参数

  1. class Deco(object):
  2. def __init__(self, *args, **kwargs):
  3. print(args, kwargs)
  4. def __call__(self, func):
  5. def _deco(*args, **kwargs):
  6. print('call Deco')
  7. func(*args, **kwargs)
  8. return _deco
  9. @Deco('hello')
  10. def test():
  11. print('call test')
  12. # 等同于
  13. # test = Deco('hello')(func)

装饰 方法

    定义一个类装饰器,装饰类中的函数,默认调用__get__方法

    实际上把类方法变成属性了,还记得类属性装饰器吧,@property

    下面自已做一个property

  1. class Decorator(object):
  2. def __init__(self, func):
  3. self.func = func
  4. def __get__(self, instance, owner):
  5. """
  6. instance:代表实例,sum中的self
  7. owner:代表类本身,Test类
  8. """
  9. print('调用的是get函数')
  10. return self.func(instance) # instance就是Test类的self
  11. class Test(object):
  12. def __init__(self):
  13. self.result = 0
  14. @Decorator
  15. def sum(self):
  16. print('There is the Func in the Class !')
  17. t = Test()
  18. print(t.sum) # 众所周知,属性是不加括号的,sum真的变成了属性

示例:

做一个求和属性sum,统计所有输入的数字的和

  1. class Decorator(object):
  2. def __init__(self, func):
  3. self.func = func
  4. def __get__(self, instance, owner):
  5. print('调用的是get函数')
  6. return self.func(instance)
  7. class Test(object):
  8. def __init__(self, *args, **kwargs):
  9. self.value_list = []
  10. if args:
  11. for i in args:
  12. if str(i).isdigit():
  13. self.value_list.append(i)
  14. if kwargs:
  15. for v in kwargs.values():
  16. if str(v).isdigit():
  17. self.value_list.append(v)
  18. @Decorator
  19. def sum(self):
  20. result = 0
  21. print(self.value_list)
  22. for i in self.value_list:
  23. result += i
  24. return result
  25. t = Test(1, 2, 3, 4, 5, 6, 7, 8, i=9, ss=10, strings='lll')
  26. print(t.sum)

无参数

  1. class Deco(object):
  2. def __init__(self, func):
  3. self.func = func
  4. def __get__(self, instance, owner):
  5. def _deco(*args, **kwargs):
  6. print('call Deco')
  7. self.func(instance, *args, **kwargs)
  8. return _deco
  9. class A(object):
  10. @Deco
  11. def test(self):
  12. print('call test')
  13. # 等同于
  14. # class A(object):
  15. # def test(self):
  16. # print('call test')
  17. # test = Deco(test)

有参数

  1. class Deco(object):
  2. def __init__(self, *args, **kwargs):
  3. print(args, kwargs)
  4. def __get__(self, instance, owner):
  5. def _deco(*args, **kwargs):
  6. print('call Deco')
  7. self.func(instance, *args, **kwargs)
  8. return _deco
  9. def __call__(self, func):
  10. self.func = func
  11. return self
  12. class A(object):
  13. @Deco('hello')
  14. def test(self):
  15. print('call test')
  16. # 等同于
  17. # class A(object):
  18. # def test(self):
  19. # print('call test')
  20. # test = Deco('hello')(test)

四:类 装饰 类

  1. class ShowClassName(object):
  2. def __init__(self, cls):
  3. self._cls = cls
  4. def __call__(self, a):
  5. print('class name:', self._cls.__name__)
  6. return self._cls(a)
  7. @ShowClassName
  8. class Foobar(object):
  9. def __init__(self, a):
  10. self.value = a
  11. def fun(self):
  12. print(self.value)
  13. a = Foobar('Are you OK')
  14. a.fun()

函数 装饰 类中的方法

示例代码 1:

From:https://www.cnblogs.com/xieqiankun/p/python_decorate_method.html

  1. import time
  2. import datetime
  3. import requests
  4. def cost_time(file_name, f_or_m=1):
  5. def _deco(origin_func):
  6. if 'f' == f_or_m:
  7. def wrapper(*args, **kwargs):
  8. start_time = datetime.datetime.now()
  9. ret_val = origin_func(*args, **kwargs)
  10. end_time = datetime.datetime.now()
  11. ct = (end_time - start_time).seconds
  12. print(f'[{file_name}][{origin_func.__name__}] cost_time:{ct}')
  13. return ret_val
  14. return wrapper
  15. elif 'm' == f_or_m:
  16. def wrapper(self, *args, **kwargs):
  17. start_time = datetime.datetime.now()
  18. ret_val = origin_func(self, *args, **kwargs)
  19. end_time = datetime.datetime.now()
  20. ct = (end_time - start_time).seconds
  21. print(f'[{file_name}][{origin_func.__name__}] cost_time:{ct}')
  22. return ret_val
  23. return wrapper
  24. return _deco
  25. class Test(object):
  26. def __init__(self):
  27. pass
  28. @cost_time(__file__, 'm')
  29. def test_1(self):
  30. time.sleep(2)
  31. pass
  32. @cost_time(__file__, 'm')
  33. def test_2(self):
  34. r = requests.get('http://www.baidu.com')
  35. if 200 == r.status_code:
  36. print(r.status_code)
  37. else:
  38. print(r.status_code)
  39. if __name__ == '__main__':
  40. t = Test()
  41. t.test_1()
  42. t.test_1()
  43. t.test_2()
  44. pass

运行结果:

  1. [test_1] cost_time:3
  2. [test_1] cost_time:3
  3. 200
  4. [test_2] cost_time:0

示例代码 2:

From:https://www.kingname.info/2017/04/17/decorate-for-method/

使用 Python 的装饰器装饰一个类的方法,同时在装饰器函数中调用类里面的其他方法。这里以捕获一个方法的异常为例来进行说明。

有一个类Test, 它的结构如下:

  1. class Test(object):
  2. def __init__(self):
  3. pass
  4. def revive(self):
  5. print('revive from exception.')
  6. # do something to restore
  7. def read_value(self):
  8. print('here I will do something.')
  9. # do something.

在类中有一个方法read_value(),这个方法在多个地方被调用。由于某些原因,方法read_value有可能随机抛出Exception导致程序崩溃。所以需要对整个方法做try ... except处理。最丑陋的做法如下面的代码所示:

  1. class Test(object):
  2. def __init__(self):
  3. pass
  4. def revive(self):
  5. print('revive from exception.')
  6. # do something to restore
  7. def read_value(self):
  8. try:
  9. print('here I will do something.')
  10. # do something.
  11. except Exception as e:
  12. print(f'exception {e} raised, parse exception.')
  13. # do other thing.
  14. self.revive()

这样写虽然可以解决问题,但是代码不Pythonic。

使用装饰器来解决这个问题,装饰器函数应该写在类里面还是类外面呢?答案是,写在类外面。那么既然写在类外面,如何调用这个类的其他方法呢?

首先写出一个最常见的处理异常的装饰器:

  1. def catch_exception(origin_func):
  2. def wrapper(*args, **kwargs):
  3. try:
  4. u = origin_func(*args, **kwargs)
  5. return u
  6. except Exception:
  7. return 'an Exception raised.'
  8. return wrapper
  9. class Test(object):
  10. def __init__(self):
  11. pass
  12. def revive(self):
  13. print('revive from exception.')
  14. # do something to restore
  15. @catch_exception
  16. def read_value(self):
  17. print('here I will do something.')
  18. # do something.

这种写法,确实可以捕获到 origin_func()的异常,但是如果在发生异常的时候,需要调用类里面的另一个方法来处理异常,这又应该怎么办?答案是给 wrapper 增加一个参数:self.

代码变为如下形式:

  1. def catch_exception(origin_func):
  2. def wrapper(self, *args, **kwargs):
  3. try:
  4. u = origin_func(self, *args, **kwargs)
  5. return u
  6. except Exception:
  7. self.revive() #不用顾虑,直接调用原来的类的方法
  8. return 'an Exception raised.'
  9. return wrapper
  10. class Test(object):
  11. def __init__(self):
  12. pass
  13. def revive(self):
  14. print('revive from exception.')
  15. # do something to restore
  16. @catch_exception
  17. def read_value(self):
  18. print('here I will do something.')
  19. # do something.

执行结果:

示例代码 3:

  1. import functools
  2. def auto_retry(src_func):
  3. @functools.wraps(src_func)
  4. def wrapper(*args, **kwargs):
  5. for i in range(3):
  6. try:
  7. return src_func(*args, **kwargs)
  8. except Exception as e:
  9. print(src_func.__name__, e)
  10. return wrapper
  11. class TestClass(object):
  12. def __init__(self):
  13. pass
  14. def __del__(self):
  15. pass
  16. @auto_retry
  17. def crawl(self, url=None):
  18. raise Exception('crawl exception')
  19. pass
  20. TestClass().crawl()

类中的方法 装饰 方法函数

  • 方法:类中的成员函数叫做 方法
  • 函数:不在类中的函数,即普通函数叫做 函数

类中的方法 装饰 函数

在类里面定义个函数,用来装饰其它函数,严格意义上说不属于类装饰器。

  1. class Buy(object):
  2. def __init__(self, func):
  3. self.func = func
  4. # 在类里定义一个函数
  5. def clothes(func): # 这里不能用self,因为接收的是body函数
  6. # 其它都和普通的函数装饰器相同
  7. def ware(*args, **kwargs):
  8. print('This is a decorator!')
  9. return func(*args, **kwargs)
  10. return ware
  11. @Buy.clothes
  12. def body(hh):
  13. print('The body feels could!{}'.format(hh))
  14. body('hh')

类中的方法 装饰 类中的方法

    背景:想要通过装饰器修改类里的self属性值。

  1. class Buy(object):
  2. def __init__(self):
  3. self.reset = True # 定义一个类属性,稍后在装饰器里更改
  4. self.func = True
  5. # 在类里定义一个装饰器
  6. def clothes(func): # func接收body
  7. def ware(self, *args, **kwargs): # self,接收body里的self,也就是类实例
  8. print('This is a decrator!')
  9. if self.reset == True: # 判断类属性
  10. print('Reset is Ture, change Func..')
  11. self.func = False # 修改类属性
  12. else:
  13. print('reset is False.')
  14. return func(self, *args, **kwargs)
  15. return ware
  16. @clothes
  17. def body(self):
  18. print('The body feels could!')
  19. b = Buy() # 实例化类
  20. b.body() # 运行body
  21. print(b.func) # 查看更改后的self.func值,是False,说明修改完成

前置知识

闭 包

javascript 闭包:https://www.runoob.com/js/js-function-closures.html

        闭包是一种保护私有变量的机制,在函数执行时形成私有的作用域,保护里面的私有变量不受外界干扰。直观的说:就是形成一个不销毁的栈环境

        闭包在维基百科上的定义如下: 在计算机科学中,闭包(Closure)是词法闭包(Lexical Closure)的简称,是引用了自由变量的函数。这个被引用的自由变量将和这个函数一同存在,即使已经离开了创造它的环境也不例外。所以,有另一种说法:认为闭包是由函数和与其相关的引用环境组合而成的实体。  

下面给出一个使用闭包实现的 logger factory 的例子:

  1. def logger_factory(prefix="", with_prefix=True):
  2. if with_prefix:
  3. def logger(msg):
  4. print(prefix + msg)
  5. return logger
  6. else:
  7. def logger(msg):
  8. print(msg)
  9. return logger
  10. logger_with_prefix = logger_factory("Prefix: ")
  11. logger_without_prefix = logger_factory(with_prefix=False)
  12. logger_with_prefix("msg")
  13. logger_without_prefix("msg")

运行结果:

  1. Prefix: msg
  2. msg

在上面这个闭包的例子中,prefix 变量时所谓的自由变量,其在 return logger 执行完毕后,便脱离了创建它的环境logger_factory,但因为其被logger_factory 中定义的 logger 函数所引用,其生命周期将至少和 logger 函数相同。这样,在 logger 中就可以引用到logger_factory 作用域内的变量 prefix

将 闭包 与 namespace 结合起来 示例:

  1. var = "var in global"
  2. def fun_outer():
  3. var = "var in fun_outer"
  4. unused_var = "this var is not used in fun_inner"
  5. print("fun_outer: " + var)
  6. print("fun_outer: " + str(locals()))
  7. print("fun_outer: " + str(id(var)))
  8. def fun_inner():
  9. print("fun_inner: " + var)
  10. print("fun_inner: " + str(locals()))
  11. print("fun_inner: " + str(id(var)))
  12. return fun_inner
  13. fun_outer()()

运行结果如下:

  1. fun_outer: var in fun_outer
  2. fun_outer: {'unused_var': 'this var is not used in fun_inner', 'var': 'var in fun_outer'}
  3. fun_outer: 2543733314784
  4. fun_inner: var in fun_outer
  5. fun_inner: {'var': 'var in fun_outer'}
  6. fun_inner: 2543733314784

在这个例子中,当 fun_outer 被定义时,其内部的定义的 fun_inner 函数对 print "fun_inner: " + var 中所引用的 var 变量进行搜索,发现第一个被搜索到的 var 定义在 fun_outer 的 local namespace 中,因此使用此定义,通过 print "fun_outer: " + str(id(var)) 和 print "fun_inner: " + str(id(var)),当var 超出 fun_outer 的作用域后,依然存活,而 fun_outer 中的unused_var 变量由于没有被 fun_inner 所引用,因此会被 GC

什么是闭包?

内部函数对外部函数作用域里变量的引用(非全局变量),则称内部函数为闭包。

简单说,闭包就是根据不同的配置信息得到不同的结果

再来看看专业的解释:闭包(Closure)是词法闭包(Lexical Closure)的简称,是引用了自由变量的函数。这个被引用的自由变量将和这个函数一同存在,即使已经离开了创造它的环境也不例外。所以,有另一种说法认为闭包是由函数和与其相关的引用环境组合而成的实体。

Python 闭包示例

示例 1:

  1. def make_adder(arg_1=None):
  2. def adder(arg_2=None):
  3. return arg_1 + arg_2
  4. return adder
  5. p = make_adder(23)
  6. q = make_adder(44)
  7. print(p(100))
  8. print(q(100))
  9. # 运行结果:
  10. # 123
  11. # 144

分析一下:

我们发现 make_adder 是一个函数,包括一个参数 arg_1 ,比较特殊的地方是这个函数里面又定义了一个新函数,这个新函数里面的一个变量正好是外部 make_adder 的参数。也就是说,外部传递过来的 arg_1 参数已经和 adder 函数绑定到一起了,形成了一个新函数,我们可以把  arg_1 看做新函数的一个配置信息,配置信息不同,函数的功能就不一样了,也就是能得到定制之后的函数.

再看看运行结果,我们发现,虽然 p 和 q 都是 make_adder 生成的,但是因为配置参数不同,后面再执行相同参数的函数后得到了不同的结果。这就是闭包。

示例 2:

  1. def hello_counter(name):
  2. count = [0]
  3. def counter():
  4. count[0] += 1
  5. print('Hello,', name, ',', str(count[0]) + ' access!')
  6. return counter
  7. hello = hello_counter('king')
  8. hello()
  9. hello()
  10. hello()
  11. # 执行结果
  12. # Hello, king , 1 access!
  13. # Hello, king , 2 access!
  14. # Hello, king , 3 access!

分析一下

这个程序比较有趣,我们可以把这个程序看做统计一个函数调用次数的函数。count[0]可以看做一个计数器,每执行一次 hello 函数,count[0] 的值就加 1。也许你会有疑问:为什么不直接写 count 而用一个列表? 这是 python2 的一个bug,如果不用列表的话,会报这样一个错误:UnboundLocalError: local variable 'count' referenced before assignment.

什么意思? 就是说 conut 这个变量你没有定义就直接引用了,我不知道这是个什么东西,程序就崩溃了。于是在 python3 里面引入了一个关键字:nonlocal,这个关键字是干什么的? 就是告诉 python 程序,我的这个 count 变量是在外部定义的,你去外面找吧。然后 python 就去外层函数找,然后就找到了 count=0 这个定义和赋值,程序就能正常执行了。

示例 2 改进:

  1. def hello_counter(name):
  2. count = 0
  3. def counter():
  4. nonlocal count
  5. count += 1
  6. print('Hello,', name, ',', str(count) + ' access!')
  7. return counter
  8. hello = hello_counter('king')
  9. hello()
  10. hello()
  11. hello()

关于这个问题的研究您可以参考:http://linluxiang.iteye.com/blog/789946

示例 3:

  1. def make_bold(fn):
  2. def wrapped():
  3. return "<b>" + fn() + "</b>"
  4. return wrapped
  5. def make_italic(fn):
  6. def wrapped():
  7. return "<i>" + fn() + "</i>"
  8. return wrapped
  9. @make_bold
  10. @make_italic
  11. def hello():
  12. return "hello world"
  13. print(hello())
  14. # 执行结果
  15. # <b><i>hello world</i></b>

简单分析

怎么样? 这个程序熟悉吗? 这不是传说的的装饰器吗? 对,这就是装饰器,其实,装饰器就是一种闭包,

我们再回想一下装饰器的概念:对函数(参数,返回值 等)进行加工处理,生成一个功能增强版的一个函数。

再看看闭包的概念,这个增强版的函数不就是我们配置之后的函数吗 ?

区别在于,装饰器的参数是一个 函数 或 类 ,专门对 类 或 函数 进行加工处理。

Python 里面的好多高级功能,比如装饰器,生成器,列表推到,闭包,匿名函数等。

探索 装饰器

定义

  1. 点击打开装饰器在维基百科上的定义链接:http://en.wikipedia.org/wiki/Python_syntax_and_semantics#Decorators
  2. 如下: A decorator is any callable Python object that is used to modify a function, method or class definition.   

基本语法

语法糖

  1. @bar
  2. def foo():
  3. print("foo")

其等价于:

  1. def foo():
  2. print("foo")
  3. foo = bar(foo)

无参数 装饰器

  1. def foo(func):
  2. print('decorator foo')
  3. return func
  4. @foo
  5. def bar():
  6. print('bar')
  7. bar()

foo 函数被用作装饰器,其本身接收一个函数对象作为参数,然后做一些工作后,返回接收的参数,供外界调用。

注意:时刻牢记 @foo 只是一个语法糖,其本质是 foo = bar(foo)

带参数 的 装饰器

示例代码:

  1. import time
  2. def function_performance_statistics(trace_this=True):
  3. if trace_this:
  4. def performance_statistics_delegate(func):
  5. def counter(*args, **kwargs):
  6. start = time.perf_counter()
  7. func(*args, **kwargs)
  8. end = time.perf_counter()
  9. print('used time: %d' % (end - start,))
  10. return counter
  11. else:
  12. def performance_statistics_delegate(func):
  13. return func
  14. return performance_statistics_delegate
  15. @function_performance_statistics(True)
  16. def add(x, y):
  17. time.sleep(3)
  18. print('add result: %d' % (x + y,))
  19. @function_performance_statistics(False)
  20. def mul(x, y=1):
  21. print('mul result: %d' % (x * y,))
  22. add(1, 1)
  23. mul(10)

上述代码想要实现一个性能分析器,并接收一个参数,来控制性能分析器是否生效,其运行效果如下所示:

  1. add result: 2
  2. used time: 3
  3. mul result: 10

上述代码中装饰器的调用等价于:

  1. import time
  2. def function_performance_statistics(trace_this=True):
  3. if trace_this:
  4. def performance_statistics_delegate(func):
  5. def counter(*args, **kwargs):
  6. start = time.perf_counter()
  7. func(*args, **kwargs)
  8. end = time.perf_counter()
  9. print('used time: %d' % (end - start,))
  10. return counter
  11. else:
  12. def performance_statistics_delegate(func):
  13. return func
  14. return performance_statistics_delegate
  15. @function_performance_statistics(True)
  16. def add(x, y):
  17. time.sleep(3)
  18. print('add result: %d' % (x + y,))
  19. @function_performance_statistics(False)
  20. def mul(x, y=1):
  21. print('mul result: %d' % (x * y,))
  22. add = function_performance_statistics(True)(add(1, 1))
  23. mul = function_performance_statistics(False)(mul(10))

装饰器

简单示例:

  1. def bar(dummy):
  2. print('bar')
  3. def inject(cls):
  4. cls.bar = bar
  5. return cls
  6. @inject
  7. class Foo(object):
  8. pass
  9. foo = Foo()
  10. foo.bar()

上述代码的 inject 装饰器为类动态的添加一个 bar 方法,因为类在调用非静态方法的时候会传进一个self 指针,因此 bar 的第一个参数我们简单的忽略即可,

运行结果如下:bar

装饰器

类装饰器 相比 函数装饰器,具有灵活度大,高内聚、封装性等优点。其实现起来主要是靠类内部的 __call__ 方法,当使用 @ 形式将装饰器附加到函数上时,就会调用此方法,下面时一个实例:

  1. class Foo(object):
  2. def __init__(self, func):
  3. super(Foo, self).__init__()
  4. self._func = func
  5. def __call__(self):
  6. print('class decorator')
  7. self._func()
  8. @Foo
  9. def bar():
  10. print('bar')
  11. bar()

运行结果如下:

  1. class decorator
  2. bar

内置装饰器

Python 中内置的装饰器有三个: staticmethodclassmethod 和 property

staticmethod 是类静态方法,其跟成员方法的区别是没有 self 指针,并且可以在类不进行实例化的情况下调用,

下面是一个实例,对比静态方法 和 成员方法:

  1. class Foo(object):
  2. def __init__(self):
  3. super(Foo, self).__init__()
  4. self.msg = 'hello word'
  5. @staticmethod
  6. def static_method(msg):
  7. print(msg)
  8. def member_method(self, msg=None):
  9. msg = msg if msg else self.msg
  10. print(msg)
  11. foo = Foo()
  12. foo.member_method('some msg')
  13. foo.static_method('some msg')
  14. Foo.static_method('some msg')

classmethod 与成员方法的区别在于所接收的第一个参数不是 self 类实例的指针,而是当前类的具体类型,

下面是一个实例:

  1. class Foo(object):
  2. @classmethod
  3. def class_method(cls):
  4. print(repr(cls))
  5. def member_method(self):
  6. print(repr(self))
  7. foo = Foo()
  8. foo.class_method()
  9. foo.member_method()

运行结果如下:

  1. <class '__main__.Foo'>
  2. <__main__.Foo object at 0x000002895A412508>

property 是属性的意思,即可以通过通过类实例直接访问的信息,下面是具体的例子:

  1. class Foo(object):
  2. def __init__(self, var):
  3. super(Foo, self).__init__()
  4. self._var = var
  5. @property
  6. def var(self):
  7. return self._var
  8. @var.setter
  9. def var(self, var):
  10. self._var = var
  11. foo = Foo('var 1')
  12. print(foo.var)
  13. foo.var = 'var 2'
  14. print(foo.var)

注意: 如果将上面的 @var.setter 装饰器所装饰的成员函数去掉,则 Foo.var 属性为只读属性,使用 foo.var = 'var 2' 进行赋值时会抛出异常,其运行结果如下:

  1. var 1
  2. var 2

调用顺序

装饰器的调用顺序与使用 @ 语法糖声明的顺序相反,如下所示:

  1. def decorator_a(func):
  2. print("decorator_a")
  3. return func
  4. def decorator_b(func):
  5. print("decorator_b")
  6. return func
  7. @decorator_a
  8. @decorator_b
  9. def foo():
  10. print("foo")
  11. foo()

其等价于:

  1. def decorator_a(func):
  2. print("decorator_a")
  3. return func
  4. def decorator_b(func):
  5. print("decorator_b")
  6. return func
  7. def foo():
  8. print("foo")
  9. foo = decorator_a(decorator_b(foo))
  10. foo()

通过等价的调用形式我们可以看到,按照 python 的函数求值序列,decorator_b(fun) 会首先被求值,然后将其结果作为输入,传递给 decorator_a,因此其调用顺序与声明顺序相反。

其运行结果如下所示:

  1. decorator_b
  2. decorator_a
  3. foo

调用时机

装饰器很好用,那么它什么时候被调用?性能开销怎么样?会不会有副作用?接下来我们就以几个实例来验证我们的猜想。

首先我们验证一下装饰器的性能开销,代码如下所示:

  1. def decorator_a(func):
  2. print("decorator_a")
  3. print('func id: ' + str(id(func)))
  4. return func
  5. def decorator_b(func):
  6. print("decorator_b")
  7. print('func id: ' + str(id(func)))
  8. return func
  9. print('Begin declare foo with decorators')
  10. @decorator_a
  11. @decorator_b
  12. def foo():
  13. print("foo")
  14. print('End declare foo with decorators')
  15. print('First call foo')
  16. foo()
  17. print('Second call foo')
  18. foo()
  19. print('Function infos')
  20. print('decorator_a id: ' + str(id(decorator_a)))
  21. print('decorator_b id: ' + str(id(decorator_b)))
  22. print('foo id : ' + str(id(foo)))

运行结果如下:

  1. Begin declare foo with decorators
  2. decorator_b
  3. func id: 1474741780056
  4. decorator_a
  5. func id: 1474741780056
  6. End declare foo with decorators
  7. First call foo
  8. foo
  9. Second call foo
  10. foo
  11. Function infos
  12. decorator_a id: 1474740396104
  13. decorator_b id: 1474740398552
  14. foo id : 1474741780056

在运行结果中:

  1. Begin declare foo with decorators
  2. decorator_b
  3. func id: 1474741780056
  4. decorator_a
  5. func id: 1474741780056
  6. End declare foo with decorators

证实了装饰器的调用时机为: 被装饰对象定义时

而运行结果中的:

  1. First call foo
  2. foo
  3. Second call foo
  4. foo

证实了在相同 .py 文件中,装饰器对所装饰的函数只进行一次装饰,不会每次调用相应函数时都重新装饰,这个很容易理解,因为其本质等价于下面的函数签名重新绑定:

foo = decorator_a(decorator_b(foo))

对于跨模块的调用,我们编写如下结构的测试代码:

  1. .  
  2. ├── common  
  3. │   ├── decorator.py  
  4. │   ├── __init__.py  
  5. │   ├── mod_a  
  6. │   │   ├── fun_a.py  
  7. │   │   └── __init__.py  
  8. │   └── mod_b  
  9. │       ├── fun_b.py  
  10. │       └── __init__.py  
  11. └── test.py

上述所有模块中的 __init__.py 文件均为: # -*- coding: utf-8 -*-

common/mod_a/fun_a.py

  1. # -*- coding: utf-8 -*-
  2. # common/mod_a/fun_a.py
  3. from common.decorator import foo
  4. def fun_a():
  5. print('in common.mod_a.fun_a.fun_a call foo')
  6. foo()

common/mod_b/fun_b.py  

  1. # -*- coding: utf-8 -*-
  2. # common/mod_b/fun_b.py
  3. from common.decorator import foo
  4. def fun_b():
  5. print('in common.mod_b.fun_b.fun_b call foo')
  6. foo()

common/decorator.py  

  1. # -*- coding: utf-8 -*-
  2. # common/decorator.py
  3. def decorator_a(func):
  4. print('init decorator_a')
  5. return func
  6. @decorator_a
  7. def foo():
  8. print('function foo')

test.py

  1. # -*- coding: utf-8 -*-
  2. # test.py
  3. from common.mod_a.fun_a import fun_a
  4. from common.mod_b.fun_b import fun_b
  5. fun_a()
  6. fun_b()

上述代码通过创建 common.mod_a 和 common.mod_b 两个子模块,并调用 common.decorator 中的 foo 函数,来测试跨模块时装饰器的工作情况,运行 test.py 的结果如下所示:

  1. init decorator_a
  2. in common.mod_a.fun_a.fun_a call foo
  3. function foo
  4. in common.mod_b.fun_b.fun_b call foo
  5. function foo

经过上面的验证,可以看出,对于跨模块的调用,装饰器也只会初始化一次,不过这要归功于 *.pyc,这与本文主题无关,故不详述。

装饰器副作用

关于装饰器副作用的话题比较大,这不仅仅是装饰器本身的问题,更多的时候是我们设计上的问题,

下面给出一个初学装饰器时大家都会遇到的一个问题 —— 丢失函数元信息:

  1. def decorator_a(func):
  2. def inner(*args, **kwargs):
  3. res = func(*args, **kwargs)
  4. return res
  5. return inner
  6. @decorator_a
  7. def foo():
  8. """
  9. foo doc
  10. :return:
  11. """
  12. return 'foo result'
  13. print('foo.__module__: ' + str(foo.__module__))
  14. print('foo.__name__: ' + str(foo.__name__))
  15. print('foo.__doc__: ' + str(foo.__doc__))
  16. print(foo())

运行结果:

  1. foo.__module__: __main__
  2. foo.__name__: inner
  3. foo.__doc__: None
  4. foo result

可以看到,在使用 decorator_a 对 foo 函数进行装饰后,foo 的元信息会丢失,解决方案参见: functools.wraps

多个装饰器运行期行为

前面已经讲解过装饰器的调用顺序和调用时机,但是被多个装饰器装饰的函数,其运行期行为还是有一些细节需要说明的,而且很可能其行为会让你感到惊讶,下面时一个实例:

  1. def tracer(msg):
  2. print("[TRACE] %s" % msg)
  3. def logger(func):
  4. tracer("logger")
  5. def inner(username, password):
  6. tracer("inner")
  7. print("call %s" % func.__name__)
  8. return func(username, password)
  9. return inner
  10. def login_debug_helper(show_debug_info=False):
  11. tracer("login_debug_helper")
  12. def proxy_fun(func):
  13. tracer("proxy_fun")
  14. def delegate_fun(username, password):
  15. tracer("delegate_fun")
  16. if show_debug_info:
  17. print(f"username:{username}\npassword:{password}")
  18. return func(username, password)
  19. return delegate_fun
  20. return proxy_fun
  21. print('Declaring login_a')
  22. @logger
  23. @login_debug_helper(show_debug_info=True)
  24. def login_a(username, password):
  25. tracer("login_a")
  26. print("do some login authentication")
  27. return True
  28. print('Call login_a')
  29. login_a("mdl", "pwd")

大家先来看一下运行结果,看看是不是跟自己想象中的一致:

  1. Declaring login_a
  2. [TRACE] login_debug_helper
  3. [TRACE] proxy_fun
  4. [TRACE] logger
  5. Call login_a
  6. [TRACE] inner
  7. call delegate_fun
  8. [TRACE] delegate_fun
  9. username:mdl
  10. password:pwd
  11. [TRACE] login_a
  12. do some login authentication

首先,装饰器初始化时的调用顺序与我们前面讲解的一致,如下:

  1. Declaring login_a
  2. [TRACE] login_debug_helper
  3. [TRACE] proxy_fun
  4. [TRACE] logger

然而,接下来,来自 logger 装饰器中的 inner 函数首先被执行,然后才是login_debug_helper 返回的 proxy_fun 中的 delegate_fun 函数。各位读者发现了吗,运行期执行login_a 函数的时候,装饰器中返回的函数的执行顺序是相反的,难道是我们前面讲解的例子有错误吗?其实,如果大家的认为运行期调用顺序应该与装饰器初始化阶段的顺序一致的话,那说明大家没有看透这段代码的调用流程,下面我来为大家分析一下。

  1. def login_debug_helper(show_debug_info=False):
  2. tracer("login_debug_helper")
  3. def proxy_fun(func):
  4. tracer("proxy_fun")
  5. def delegate_fun(username, password):
  6. tracer("delegate_fun")
  7. if show_debug_info:
  8. print(f"username:{username}\npassword:{password}")
  9. return func(username, password)
  10. return delegate_fun
  11. return proxy_fun

当装饰器 login_debug_helper 被调用时,其等价于:

gin_debug_helper(show_debug_info=True)(login_a)('mdl', 'pwd')

对于只有 login_debug_helper 的情况,现在就应该是执行完 login_a 输出结果的时刻了,但是如果现在在加上 logger 装饰器的话,那么这个 login_debug_helper(show_debug_info=True)(login_a)('mdl', 'pwd') 就被延迟执行,而将 login_debug_helper(show_debug_info=True)(login_a) 作为参数传递给 logger,我们令 login_tmp = login_debug_helper(show_debug_info=True)(login_a),则调用过程等价于:

  1. login_tmp = login_debug_helper(show_debug_info=True)(login_a)
  2. login_a = logger(login_tmp)
  3. login_a('mdl', 'pwd')

相信大家看过上面的等价变换后,已经明白问题出在哪里了,如果你还没有明白,我强烈建议你把这个例子自己敲一遍,并尝试用自己的方式进行化简,逐步得出结论。

一些实例参考

本文主要讲解原理性的东西,具体的实例可以参考下面的链接:Python装饰器实例:调用参数合法性验证

Python装饰器与面向切面编程

Python装饰器小结

Python tips: 超时装饰器, @timeout decorator

python中判断一个运行时间过长的函数

python利用装饰器和threading实现异步调用

python输出指定函数运行时间的装饰器

python通过装饰器和线程限制函数的执行时间

python装饰器的一个妙用

通过 Python 装饰器实现DRY(不重复代码)原则

参考资料

Understanding Python Decorators in 12 Easy Steps

Decorators and Functional Python

Python Wiki: PythonDecorators

Meta-matters: Using decorators for better Python programming

Python装饰器入门(译)

Python装饰器与面向切面编程

Python 的闭包和装饰器

Python装饰器学习(九步入门)

python 装饰器和 functools 模块

理解 Python 中的装饰器

From:https://www.cnblogs.com/rollenholt/archive/2012/05/02/2479833.html

文章先由 stackoverflow上面的一个问题引起吧,如果使用如下的代码:

  1. @makebold
  2. @makeitalic
  3. def say():
  4. return "Hello"

打印出如下的输出:

<b><i>Hello<i></b>

你会怎么做?最后给出的答案是:

  1. def make_bold(fn):
  2. def wrapped():
  3. return "<b>" + fn() + "</b>"
  4. return wrapped
  5. def make_italic(fn):
  6. def wrapped():
  7. return "<i>" + fn() + "</i>"
  8. return wrapped
  9. @make_bold
  10. @make_italic
  11. def hello():
  12. return "hello world"
  13. print(hello())
  14. # 结果: <b><i>hello world</i></b>

现在我们来看看如何从一些最基础的方式来理解Python的装饰器。英文讨论参考Here

装饰器是一个很著名的设计模式,经常被用于有切面需求的场景,较为经典的有插入日志、性能测试、事务处理等。装饰器是解决这类问题的绝佳设计,有了装饰器,我们就可以抽离出大量函数中与函数功能本身无关的雷同代码并继续重用。概括的讲,装饰器的作用就是为已经存在的对象添加额外的功能。

1.1. 需求是怎么来的 ?

装饰器的定义很是抽象,我们来看一个小例子。

  1. def foo():
  2. print('in foo()')
  3. foo()

这是一个很无聊的函数没错。但是突然有一个更无聊的人,我们称呼他为B君,说我想看看执行这个函数用了多长时间,好吧,那么我们可以这样做:

  1. import time
  2. def foo():
  3. start = time.clock()
  4. print('in foo()')
  5. end = time.clock()
  6. print('used:', end - start)
  7. foo()

很好,功能看起来无懈可击。可是蛋疼的B君此刻突然不想看这个函数了,他对另一个叫foo2的函数产生了更浓厚的兴趣。
怎么办呢?如果把以上新增加的代码复制到foo2里,这就犯了大忌了~复制什么的难道不是最讨厌了么!而且,如果B君继续看了其他的函数呢?

1.2. 以不变应万变,是变也

还记得吗,函数在Python中是一等公民,那么我们可以考虑重新定义一个函数timeit,将foo的引用传递给他,然后在timeit中调用foo并进行计时,这样,我们就达到了不改动foo定义的目的,而且,不论B君看了多少个函数,我们都不用去修改函数定义了!

  1. import time
  2. def foo():
  3. print('in foo()')
  4. def timeit(func):
  5. start = time.clock()
  6. func()
  7. end = time.clock()
  8. print('used:', end - start)
  9. timeit(foo)

看起来逻辑上并没有问题,一切都很美好并且运作正常!……等等,我们似乎修改了调用部分的代码。原本我们是这样调用的:foo(),修改以后变成了:timeit(foo)。这样的话,如果foo在N处都被调用了,你就不得不去修改这N处的代码。或者更极端的,考虑其中某处调用的代码无法修改这个情况,比如:这个函数是你交给别人使用的。

1.3. 最大限度地少改动!

既然如此,我们就来想想办法不修改调用的代码;如果不修改调用代码,也就意味着调用foo()需要产生调用timeit(foo)的效果。我们可以想到将timeit赋值给foo,但是timeit似乎带有一个参数……想办法把参数统一吧!如果timeit(foo)不是直接产生调用效果,而是返回一个与foo参数列表一致的函数的话……就很好办了,将timeit(foo)的返回值赋值给foo,然后,调用foo()的代码完全不用修改!

  1. # -*- coding: UTF-8 -*-
  2. import time
  3. def foo():
  4. print('in foo()')
  5. # 定义一个计时器,传入一个,并返回另一个附加了计时功能的方法
  6. def timeit(func):
  7. # 定义一个内嵌的包装函数,给传入的函数加上计时功能的包装
  8. def wrapper():
  9. start = time.perf_counter()
  10. func()
  11. end = time.perf_counter()
  12. print('used:', end - start)
  13. # 将包装后的函数返回
  14. return wrapper
  15. foo = timeit(foo)
  16. foo()

这样,一个简易的计时器就做好了!我们只需要在定义foo以后调用foo之前,加上foo = timeit(foo),就可以达到计时的目的,这也就是装饰器的概念,看起来像是foo被timeit装饰了。在在这个例子中,函数进入和退出时需要计时,这被称为一个横切面(Aspect),这种编程方式被称为面向切面的编程(Aspect-Oriented Programming)。与传统编程习惯的从上往下执行方式相比较而言,像是在函数执行的流程中横向地插入了一段逻辑。在特定的业务领域里,能减少大量重复代码。面向切面编程还有相当多的术语,这里就不多做介绍,感兴趣的话可以去找找相关的资料。
这个例子仅用于演示,并没有考虑foo带有参数和有返回值的情况,完善它的重任就交给你了 :)
上面这段代码看起来似乎已经不能再精简了,Python于是提供了一个语法糖来降低字符输入量。

  1. import time
  2. def timeit(func):
  3. def wrapper():
  4. start = time.perf_counter()
  5. func()
  6. end = time.perf_counter()
  7. print('used:', end - start)
  8. return wrapper
  9. @timeit
  10. def foo():
  11. print('in foo()')
  12. foo()

重点关注第11行的@timeit,在定义上加上这一行与另外写foo = timeit(foo)完全等价

千万不要以为@有另外的魔力。除了字符输入少了一些,还有一个额外的好处:这样看上去更有装饰器的感觉。

-------------------

Python中函数也是被视为对象(Python 中一切 皆 对象)

要理解python的装饰器,我们首先必须明白在Python中函数也是被视为对象。这一点很重要。先看一个例子:

  1. def shout(word="yes"):
  2. return word.capitalize() + " !"
  3. print(shout())
  4. # 输出 : 'Yes !'
  5. # 作为一个对象,你可以把函数赋给任何其他对象变量
  6. scream = shout
  7. # 注意我们没有使用圆括号,因为我们不是在调用函数
  8. # 我们把函数shout赋给scream,也就是说你可以通过scream调用shout
  9. print(scream())
  10. # 输出 : 'Yes !'
  11. # 还有,你可以删除旧的名字shout,但是你仍然可以通过scream来访问该函数
  12. del shout
  13. try:
  14. print(shout())
  15. except BaseException as e:
  16. print(e)
  17. # 输出 : "name 'shout' is not defined"
  18. print(scream())
  19. # 输出 : 'Yes !'

我们暂且把这个话题放旁边,我们先看看 python 另外一个很有意思的属性:可以在函数中定义函数:

  1. def talk():
  2. # 你可以在talk中定义另外一个函数
  3. def whisper(word="yes"):
  4. return word.lower() + "..."
  5. # ... 并且立马使用它
  6. print(whisper())
  7. # 你每次调用'talk',定义在talk里面的whisper同样也会被调用
  8. talk()
  9. # 输出 :
  10. # yes...
  11. # 但是"whisper" 不会单独存在:
  12. try:
  13. print(whisper())
  14. except BaseException as e:
  15. print(e)
  16. # 输出 : "name 'whisper' is not defined"*

函数引用

从以上两个例子我们可以得出,函数既然作为一个对象,因此:

  • 1. 其可以被赋给其他变量
  • 2. 其可以被定义在另外一个函数内

这也就是说,函数可以返回一个函数,看下面的例子:

  1. def get_talk(type="shout"):
  2. # 我们定义另外一个函数
  3. def shout(word="yes"):
  4. return word.capitalize() + " !"
  5. def whisper(word="yes"):
  6. return word.lower() + "..."
  7. # 然后我们返回其中一个
  8. if type == "shout":
  9. # 我们没有使用(),因为我们不是在调用该函数
  10. # 我们是在返回该函数
  11. return shout
  12. else:
  13. return whisper
  14. # 然后怎么使用呢 ?
  15. # 把该函数赋予某个变量
  16. talk = get_talk()
  17. # 这里你可以看到talk其实是一个函数对象:
  18. print(talk)
  19. # 输出 : <function shout at 0xb7ea817c>
  20. # 该对象由函数返回的其中一个对象:
  21. print(talk())
  22. # 或者你可以直接如下调用 :
  23. print(get_talk("whisper")())
  24. # 输出 : yes...

还有,既然可以返回一个函数,我们可以把它作为参数传递给函数:

  1. def do_something_before(func):
  2. print("I do something before then I call the function you gave me")
  3. print(func())
  4. do_something_before(scream)
  5. # 输出 :
  6. # I do something before then I call the function you gave me
  7. # Yes !

这里你已经足够能理解装饰器了,其他它可被视为封装器。也就是说,它能够让你在装饰前后执行代码而无须改变函数本身内容。

手工装饰

那么如何进行手动装饰呢?

  1. # 装饰器是一个函数,而其参数为另外一个函数
  2. def my_shiny_new_decorator(a_function_to_decorate):
  3. # 在内部定义了另外一个函数:一个封装器。
  4. # 这个函数将原始函数进行封装,所以你可以在它之前或者之后执行一些代码
  5. def the_wrapper_around_the_original_function():
  6. # 放一些你希望在真正函数执行前的一些代码
  7. print("Before the function runs")
  8. # 执行原始函数
  9. a_function_to_decorate()
  10. # 放一些你希望在原始函数执行后的一些代码
  11. print("After the function runs")
  12. # 在此刻,"a_function_to_decrorate"还没有被执行,我们返回了创建的封装函数
  13. # 封装器包含了函数以及其前后执行的代码,其已经准备完毕
  14. return the_wrapper_around_the_original_function
  15. # 现在想象下,你创建了一个你永远也不远再次接触的函数
  16. def a_stand_alone_function():
  17. print("I am a stand alone function, don't you dare modify me")
  18. a_stand_alone_function()
  19. # 输出: I am a stand alone function, don't you dare modify me
  20. # 好了,你可以封装它实现行为的扩展。可以简单的把它丢给装饰器
  21. # 装饰器将动态地把它和你要的代码封装起来,并且返回一个新的可用的函数。
  22. a_stand_alone_function_decorated = my_shiny_new_decorator(a_stand_alone_function)
  23. a_stand_alone_function_decorated()
  24. # 输出 :
  25. # Before the function runs
  26. # I am a stand alone function, don't you dare modify me
  27. # After the function runs
  28. # 现在你也许要求当每次调用a_stand_alone_function时,
  29. # 实际调用却是a_stand_alone_function_decorated。
  30. # 实现也很简单,可以用my_shiny_new_decorator来给a_stand_alone_function重新赋值。
  31. a_stand_alone_function = my_shiny_new_decorator(a_stand_alone_function)
  32. a_stand_alone_function()
  33. # 输出 :
  34. # Before the function runs
  35. # I am a stand alone function, don't you dare modify me
  36. # After the function runs
  37. # And guess what, that's EXACTLY what decorators do !

装饰器揭秘

前面的例子,我们可以使用装饰器的语法:

  1. @my_shiny_new_decorator
  2. def another_stand_alone_function():
  3. print("Leave me alone")
  4. another_stand_alone_function()
  5. # 输出 :
  6. # Before the function runs
  7. # Leave me alone
  8. # After the function runs

当然你也可以累积装饰:

  1. def bread(func):
  2. def wrapper():
  3. print(r"</''''''\>")
  4. func()
  5. print(r"<\______/>")
  6. return wrapper
  7. def ingredients(func):
  8. def wrapper():
  9. print("#tomatoes#")
  10. func()
  11. print("~salad~")
  12. return wrapper
  13. def sandwich(food="--ham--"):
  14. print(food)
  15. sandwich()
  16. # 输出 : --ham--
  17. sandwich = bread(ingredients(sandwich))
  18. sandwich()
  19. # outputs :
  20. # </''''''\>
  21. # #tomatoes#
  22. # --ham--
  23. # ~salad~
  24. # <\______/>

使用 python 装饰器语法:

  1. @bread
  2. @ingredients
  3. def sandwich(food="--ham--"):
  4. print(food)
  5. sandwich()
  6. # 输出 :
  7. # </''''''\>
  8. # #tomatoes#
  9. # --ham--
  10. # ~salad~
  11. # <\______/>

装饰器的顺序很重要,需要注意:

  1. @ingredients
  2. @bread
  3. def strange_sandwich(food="--ham--"):
  4. print(food)
  5. strange_sandwich()
  6. # 输出 :
  7. # tomatoes#
  8. # </''''''\>
  9. # --ham--
  10. # <\______/>
  11. # ~salad~

最后回答前面提到的问题:

  1. # 装饰器 make_bold 用于转换为粗体
  2. def make_bold(fn):
  3. # 结果返回该函数
  4. def wrapper():
  5. # 插入一些执行前后的代码
  6. return "<b>" + fn() + "</b>"
  7. return wrapper
  8. # 装饰器 make_italic 用于转换为斜体
  9. def make_italic(fn):
  10. # 结果返回该函数
  11. def wrapper():
  12. # 插入一些执行前后的代码
  13. return "<i>" + fn() + "</i>"
  14. return wrapper
  15. @make_bold
  16. @make_italic
  17. def say():
  18. return "hello"
  19. print(say())
  20. # 输出: <b><i>hello</i></b>
  21. # 等同于
  22. def say():
  23. return "hello"
  24. say = make_bold(make_italic(say))
  25. print(say())
  26. # 输出: <b><i>hello</i></b>

内置的装饰器

内置的装饰器有三个,分别是staticmethod、classmethod和property,作用分别是把类中定义的实例方法变成静态方法、类方法和类属性。由于模块里可以定义函数,所以静态方法和类方法的用处并不是太多,除非你想要完全的面向对象编程。而属性也不是不可或缺的,Java没有属性也一样活得很滋润。从我个人的Python经验来看,我没有使用过property,使用staticmethod和classmethod的频率也非常低。

  1. class Rabbit(object):
  2. def __init__(self, name):
  3. self._name = name
  4. @staticmethod
  5. def newRabbit(name):
  6. return Rabbit(name)
  7. @classmethod
  8. def newRabbit2(cls):
  9. return Rabbit('')
  10. @property
  11. def name(self):
  12. return self._name

这里定义的属性是一个只读属性,如果需要可写,则需要再定义一个setter:
@name.setter
def name(self, name):
    self._name = name
functools模块
functools模块提供了两个装饰器。这个模块是Python 2.5后新增的,一般来说大家用的应该都高于这个版本。但我平时的工作环境是2.4 T-T
2.3.1. wraps(wrapped[, assigned][, updated]): 
这是一个很有用的装饰器。看过前一篇反射的朋友应该知道,函数是有几个特殊属性比如函数名,在被装饰后,上例中的函数名foo会变成包装函数的名字wrapper,如果你希望使用反射,可能会导致意外的结果。这个装饰器可以解决这个问题,它能将装饰过的函数的特殊属性保留。

  1. import time
  2. import functools
  3. def timeit(func):
  4. @functools.wraps(func)
  5. def wrapper():
  6. start = time.clock()
  7. func()
  8. end =time.clock()
  9. print 'used:', end - start
  10. return wrapper
  11. @timeit
  12. def foo():
  13. print 'in foo()'
  14. foo()
  15. print foo.__name__

首先注意第5行,如果注释这一行,foo.__name__将是'wrapper'。另外相信你也注意到了,这个装饰器竟然带有一个参数。实际上,他还有另外两个可选的参数,assigned中的属性名将使用赋值的方式替换,而updated中的属性名将使用update的方式合并,你可以通过查看functools的源代码获得它们的默认值。对于这个装饰器,相当于wrapper = functools.wraps(func)(wrapper)。

深入浅出 Python 装饰器:16 步轻松搞定 Python 装饰器

而Python使用了一种相对于Decorator Pattern和Annotation来说非常优雅的方法,这种方法不需要你去掌握什么复杂的OO模型或是Annotation的各种类库规定,完全就是语言层面的玩法:一种函数式编程的技巧。如果你看过本站的《函数式编程》,你一定会为函数式编程的那种“描述你想干什么,而不是描述你要怎么去实现”的编程方式感到畅快。(如果你不了解函数式编程,那在读本文之前,还请你移步去看看《函数式编程》) 好了。

1. 函数

在python中,函数通过def关键字、函数名和可选的参数列表定义。通过return关键字返回值。我们举例来说明如何定义和调用一个简单的函数:

  1. def foo():
  2. return 1
  3. foo()
  4. 1

方法体(当然多行也是一样的)是必须的,通过缩进来表示,在方法名的后面加上双括号()就能够调用函数

2. 作用域

python中,函数会创建一个新的作用域。python开发者可能会说函数有自己的命名空间,差不多一个意思。这意味着在函数内部碰到一个变量的时候函数会优先在自己的命名空间里面去寻找。让我们写一个简单的函数看一下 本地作用域 和 全局作用域有什么不同:

  1. a_string = "This is a global variable"
  2. def foo():
  3. print locals()
  4. print globals()
  5. {..., 'a_string': 'This is a global variable'}
  6. foo() # 2
  7. {}

内置的函数globals返回一个包含所有python解释器知道的变量名称的字典(为了干净和洗的白白的,我省略了python自行创建的一些变量)。在#2我调用了函数 foo 把函数内部本地作用域里面的内容打印出来。我们能够看到,函数foo有自己独立的命名空间,虽然暂时命名空间里面什么都还没有。

3. 变量解析规则

当然这并不是说我们在函数里面就不能访问外面的全局变量。在python的作用域规则里面,创建变量一定会在当前作用域里创建一个变量,但是访问或者修改变量时会先在当前作用域查找变量,没有找到匹配变量的话会依次向上在闭合的作用域里面进行查看找。所以如果我们修改函数foo的实现让它打印全局的作用域里的变量也是可以的:

  1. a_string = "This is a global variable"
  2. def foo():
  3. print a_string # 1
  4. foo()
  5. This is a global variable

在#1处,python解释器会尝试查找变量a_string,当然在函数的本地作用域里面找不到,所以接着会去上层的作用域里面去查找。

但是另一方面,假如我们在函数内部给全局变量赋值,结果却和我们想的不一样:

  1. a_string = "This is a global variable"
  2. def foo():
  3. a_string = "test" # 1
  4. print locals()
  5. foo()
  6. {'a_string': 'test'}
  7. a_string # 2
  8. 'This is a global variable'

我们能够看到,全局变量能够被访问到(如果是可变数据类型(像list,dict这些)甚至能够被更改)但是赋值不行。在函数内部的#1处,我们实际上新创建了一个局部变量,隐藏全局作用域中的同名变量。我们可以通过打印出局部命名空间中的内容得出这个结论。我们也能看到在#2处打印出来的变量a_string的值并没有改变。

4. 变量生存周期

值得注意的一个点是,变量不仅是生存在一个个的命名空间内,他们都有自己的生存周期,请看下面这个例子:

  1. def foo():
  2. x = 1
  3. foo()
  4. print x # 1
  5. #Traceback (most recent call last):
  6. #NameError: name 'x' is not defined

#1处发生的错误不仅仅是因为作用域规则导致的(尽管这是抛出了NameError的错误的原因)它还和python以及其它很多编程语言中函数调用实现的机制有关。在这个地方这个执行时间点并没有什么有效的语法让我们能够获取变量x的值,因为它这个时候压根不存在!函数foo的命名空间随着函数调用开始而开始,结束而销毁。

5. 函数参数

python允许我们向函数传递参数,参数会变成本地变量存在于函数内部。

  1. def foo(x):
  2. print locals()
  3. foo(1)
  4. {'x': 1}

在Python里有很多的方式来定义和传递参数,完整版可以查看 python官方文档。我们这里简略的说明一下:函数的参数可以是必须的位置参数或者是可选的命名,默认参数。

  1. def foo(x, y=0): # 1
  2. return x - y
  3. foo(3, 1) # 2
  4. 2
  5. foo(3) # 3
  6. 3
  7. foo() # 4
  8. #Traceback (most recent call last):
  9. #TypeError: foo() takes at least 1 argument (0 given)
  10. foo(y=1, x=3) # 5
  11. 2

在#1处我们定义了函数foo,它有一个位置参数x和一个命名参数y。在#2处我们能够通过常规的方式来调用函数,尽管有一个命名参数,但参数依然可以通过位置传递给函数。在调用函数的时候,对于命名参数y我们也可以完全不管就像#3处所示的一样。如果命名参数没有接收到任何值的话,python会自动使用声明的默认值也就是0。需要注意的是我们不能省略第一个位置参数x, 否则的话就会像#4处所示发生错误。

目前还算简洁清晰吧, 但是接下来可能会有点令人困惑。python支持函数调用时的命名参数(个人觉得应该是命名实参)。看看#5处的函数调用,我们传递的是两个命名实参,这个时候因为有名称标识,参数传递的顺序也就不用在意了。

当然相反的情况也是正确的:函数的第二个形参是y,但是我们通过位置的方式传递值给它。在#2处的函数调用foo(3,1),我们把3传递给了第一个参数,把1传递给了第二个参数,尽管第二个参数是一个命名参数。

一个简单的概念:函数的参数可以有名称和位置。这意味着在函数的定义和调用的时候会稍稍在理解上有点儿不同。我们可以给只定义了位置参数的函数传递命名参数(实参),反之亦然!如果觉得不够可以查看官方文档

6. 嵌套函数

Python允许创建嵌套函数。这意味着我们可以在函数里面定义函数而且现有的作用域和变量生存周期依旧适用。

  1. def outer():
  2. x = 1
  3. def inner():
  4. print x # 1
  5. return inner() # 2
  6. outer()
  7. 1

这个例子有一点儿复杂,但是看起来也还行。想一想在#1发生了什么:python解释器需找一个叫x的本地变量,查找失败之后会继续在上层的作用域里面寻找,这个上层的作用域定义在另外一个函数里面。对函数outer来说,变量x是一个本地变量,但是如先前提到的一样,函数inner可以访问封闭的作用域(至少可以读和修改)。在#2处,我们调用函数inner,非常重要的一点是,inner也仅仅是一个遵循python变量解析规则的变量名,python解释器会优先在outer的作用域里面对变量名inner查找匹配的变量.

7. 函数是 python 世界里的一级类对象

显而易见,在python里函数和其他东西一样都是对象。(此处应该大声歌唱)啊!包含变量的函数,你也并不是那么特殊!

  1. issubclass(int, object) # all objects in Python inherit from a common baseclass
  2. #True
  3. def foo():
  4. pass
  5. foo.__class__ # 1
  6. #<type 'function'>
  7. issubclass(foo.__class__, object)
  8. #True

你也许从没有想过,你定义的函数居然会有属性。没办法,函数在python里面就是对象(Python一切皆对象),和其他的东西一样,也许这样描述会太学院派太官方了点:在python里,函数只是一些普通的值而已和其他的值一毛一样。这就是说你可以把函数像参数一样传递给其他的函数或者说从函数里面返回函数!如果你从来没有这么想过,那看看下面这个例子:

  1. def add(x, y):
  2. return x + y
  3. def sub(x, y):
  4. return x - y
  5. def apply(func, x, y): # 1
  6. return func(x, y) # 2
  7. apply(add, 2, 1) # 3
  8. 3
  9. apply(sub, 2, 1)
  10. 1

这个例子对你来说应该不会很奇怪。add和sub是非常普通的两个python函数,接受两个值,返回一个计算后的结果值。在#1处你们能看到准备接收一个函数的变量只是一个普通的变量而已,和其他变量一样。在#2处我们调用传进来的函数:“()代表着调用的操作并且调用变量包含的值。在#3处,你们也能看到传递函数并没有什么特殊的语法。” 函数的名称只是很其他变量一样的表标识符而已。

你们也许看到过这样的行为:“python把频繁要用的操作变成函数作为参数进行使用,像通过传递一个函数给内置排序函数的key参数从而来自定义排序规则。那把函数当做返回值回事这样的情况呢:

  1. def outer():
  2. def inner():
  3. print "Inside inner"
  4. return inner # 1
  5. foo = outer() #2
  6. foo
  7. #<function inner at 0x...>
  8. foo()
  9. #Inside inner

这个例子看起来也许会更加的奇怪。在#1处我把恰好是函数标识符的变量inner作为返回值返回出来。这并没有什么特殊的语法:”把函数inner返回出来,否则它根本不可能会被调用到。“还记得变量的生存周期吗?每次函数outer被调用的时候,函数inner都会被重新定义,如果它不被当做变量返回的话,每次执行过后它将不复存在。

在#2处我们捕获住返回值 – 函数inner,将它存在一个新变量foo里。我们能够看到,当对变量foo进行求值,它确实包含函数inner,而且我们能够对他进行调用。初次看起来可能会觉得有点奇怪,但是理解起来并不困难是吧。坚持住,因为奇怪的转折马上就要来了

8. 闭包

我们先不急着定义什么是闭包,先来看看一段代码,仅仅是把上一个例子简单的调整了一下:

  1. def outer():
  2. x = 1
  3. def inner():
  4. print x # 1
  5. return inner
  6. foo = outer()
  7. foo.func_closure
  8. #(<cell at 0x...: int object at 0x...>,)

在上一个例子中我们了解到,inner作为一个函数被outer返回,保存在一个变量foo,并且我们能够对它进行调用foo()。不过它会正常的运行吗?我们先来看看作用域规则。

所有的东西都在python的作用域规则下进行工作:“x是函数outer里的一个局部变量。当函数inner在#1处打印x的时候,python解释器会在inner内部查找相应的变量,当然会找不到,所以接着会到封闭作用域里面查找,并且会找到匹配。

但是从变量的生存周期来看,该怎么理解呢?我们的变量x是函数outer的一个本地变量,这意味着只有当函数outer正在运行的时候才会存在。根据我们已知的python运行模式,我们没法在函数outer返回之后继续调用函数inner,在函数inner被调用的时候,变量x早已不复存在,可能会发生一个运行时错误。

万万没想到,返回的函数inner居然能够正常工作。Python支持一个叫做函数闭包的特性,用人话来讲就是,嵌套定义在非全局作用域里面的函数能够记住它在被定义的时候它所处的封闭命名空间。这能够通过查看函数的func_closure属性得出结论,这个属性里面包含封闭作用域里面的值(只会包含被捕捉到的值,比如x,如果在outer里面还定义了其他的值,封闭作用域里面是不会有的)

记住,每次函数outer被调用的时候,函数inner都会被重新定义。现在变量x的值不会变化,所以每次返回的函数inner会是同样的逻辑,假如我们稍微改动一下呢?

  1. def outer(x):
  2. def inner():
  3. print x # 1
  4. return inner
  5. print1 = outer(1)
  6. print2 = outer(2)
  7. print1()
  8. 1
  9. print2()
  10. 2

从这个例子中你能够看到闭包 – 被函数记住的封闭作用域 – 能够被用来创建自定义的函数,本质上来说是一个硬编码的参数。事实上我们并不是传递参数1或者2给函数inner,我们实际上是创建了能够打印各种数字的各种自定义版本。

闭包单独拿出来就是一个非常强大的功能, 在某些方面,你也许会把它当做一个类似于面向对象的技术:outer像是给inner服务的构造器,x像一个私有变量。使用闭包的方式也有很多:你如果熟悉python内置排序方法的参数key,你说不定已经写过一个lambda方法在排序一个列表的列表的时候基于第二个元素而不是第一个。现在你说不定也可以写一个itemgetter方法,接收一个索引值来返回一个完美的函数,传递给排序函数的参数key。

不过,我们现在不会用闭包做这么low的事(⊙o⊙)…!相反,让我们再爽一次,写一个高大上的装饰器!

9. 装饰器

装饰器其实就是一个闭包,把一个函数当做参数然后返回一个替代版函数。我们一步步从简到繁来瞅瞅:

  1. def outer(some_func):
  2. def inner():
  3. print "before some_func"
  4. ret = some_func() # 1
  5. return ret + 1
  6. return inner
  7. def foo():
  8. return 1
  9. decorated = outer(foo) # 2
  10. decorated()
  11. #before some_func
  12. #2

仔细看看上面这个装饰器的例子。们定义了一个函数outer,它只有一个some_func的参数,在他里面我们定义了一个嵌套的函数inner。inner会打印一串字符串,然后调用some_func,在#1处得到它的返回值。在outer每次调用的时候some_func的值可能会不一样,但是不管some_func的之如何,我们都会调用它。最后,inner返回some_func() + 1的值 – 我们通过调用在#2处存储在变量decorated里面的函数能够看到被打印出来的字符串以及返回值2,而不是期望中调用函数foo得到的返回值1。

我们可以认为变量decorated是函数foo的一个装饰版本,一个加强版本。事实上如果打算写一个有用的装饰器的话,我们可能会想愿意用装饰版本完全取代原先的函数foo,这样我们总是会得到我们的”加强版“foo。想要达到这个效果,完全不需要学习新的语法,简单地赋值给变量foo就行了:

  1. foo = outer(foo)
  2. foo # doctest: +ELLIPSIS
  3. #<function inner at 0x...>

现在,任何怎么调用都不会牵扯到原先的函数foo,都会得到新的装饰版本的foo。

假设有如下函数:

  1. def now():
  2. print '2013-12-25'
  3. f = now
  4. f()
  5. #2013-12-25

现在假设我们要增强now()函数的功能,比如,在函数调用前后自动打印日志,但又不希望修改now()函数的定义,这种在代码运行期间动态增加功能的方式,称之为“装饰器”(Decorator)。

本质上,decorator就是一个返回函数的高阶函数。所以,我们要定义一个能打印日志的decorator,可以定义如下:

  1. def log(func):
  2. def wrapper(*args, **kw):
  3. print 'call %s():' % func.__name__
  4. return func(*args, **kw)
  5. return wrapper

观察上面的log,因为它是一个decorator,所以接受一个函数作为参数,并返回一个函数。

10. 使用 @ 标识符将装饰器应用到函数

Python2.4支持使用标识符@将装饰器应用在函数上,只需要在函数的定义前加上@和装饰器的名称。在上一节的例子里我们是将原本的方法用装饰后的方法代替:

add = wrapper(add)

这种方式能够在任何时候对任意方法进行包装。但是如果我们自定义一个方法,我们可以使用@进行装饰:

  1. @wrapper
  2. def add(a, b):
  3. return Coordinate(a.x + b.x, a.y + b.y)

需要明白的是,这样的做法和先前简单的用包装方法替代原有方法是一模一样的, python只是加了一些语法糖让装饰的行为更加的直接明确和优雅一点。

多个decorator

  1. @decorator_one
  2. @decorator_two
  3. def func():
  4. pass

相当于:

func = decorator_one(decorator_two(func))

比如:带参数的decorator:

  1. @decorator(arg1, arg2)
  2. def func():
  3. pass

相当于:

func = decorator(arg1,arg2)(func)

这意味着decorator(arg1, arg2)这个函数需要返回一个“真正的decorator”。

11. *args and **kwargs

我们已经完成了一个有用的装饰器,但是由于硬编码的原因它只能应用在一类具体的方法上,这类方法接收两个参数,传递给闭包捕获的函数。如果我们想实现一个能够应用在任何方法上的装饰器要怎么做呢?再比如,如果我们要实现一个能应用在任何方法上的类似于计数器的装饰器,不需要改变原有方法的任何逻辑。这意味着装饰器能够接受拥有任何签名的函数作为自己的被装饰方法,同时能够用传递给它的参数对被装饰的方法进行调用。

非常巧合的是Python正好有支持这个特性的语法。可以阅读 Python Tutorial 获取更多的细节。当定义函数的时候使用了*,意味着那些通过位置传递的参数将会被放在带有*前缀的变量中, 所以:

  1. def one(*args):
  2. print args # 1
  3. one()
  4. #()
  5. one(1, 2, 3)
  6. #(1, 2, 3)
  7. def two(x, y, *args): # 2
  8. print x, y, args
  9. two('a', 'b', 'c')
  10. #a b ('c',)

第一个函数one只是简单地讲任何传递过来的位置参数全部打印出来而已,你们能够看到,在代码#1处我们只是引用了函数内的变量args, *args仅仅只是用在函数定义的时候用来表示位置参数应该存储在变量args里面。Python允许我们制定一些参数并且通过args捕获其他所有剩余的未被捕捉的位置参数,就像#2处所示的那样。
*操作符在函数被调用的时候也能使用。意义基本是一样的。当调用一个函数的时候,一个用*标志的变量意思是变量里面的内容需要被提取出来然后当做位置参数被使用。同样的,来看个例子:

  1. def add(x, y):
  2. return x + y
  3. lst = [1,2]
  4. add(lst[0], lst[1]) # 1
  5. 3
  6. add(*lst) # 2
  7. 3

#1处的代码和#2处的代码所做的事情其实是一样的,在#2处,python为我们所做的事其实也可以手动完成。这也不是什么坏事,*args要么是表示调用方法大的时候额外的参数可以从一个可迭代列表中取得,要么就是定义方法的时候标志这个方法能够接受任意的位置参数。
接下来提到的**会稍多更复杂一点,**代表着键值对的餐宿字典,和*所代表的意义相差无几,也很简单对不对:

  1. def foo(**kwargs):
  2. print kwargs
  3. foo()
  4. #{}
  5. foo(x=1, y=2)
  6. #{'y': 2, 'x': 1}

当我们定义一个函数的时候,我们能够用**kwargs来表明,所有未被捕获的关键字参数都应该存储在kwargs的字典中。如前所诉,args、kwargs并不是python语法的一部分,但在定义函数的时候,使用这样的变量名算是一个不成文的约定。和*一样,我们同样可以在定义或者调用函数的时候使用**。

  1. dct = {'x': 1, 'y': 2}
  2. def bar(x, y):
  3. return x + y
  4. bar(**dct)
  5. #3

12. 更通用的装饰器

有了这招新的技能,我们随随便便就可以写一个能够记录下传递给函数参数的装饰器了。先来个简单地把日志输出到界面的例子:

  1. def logger(func):
  2. def inner(*args, **kwargs): #1
  3. print "Arguments were: %s, %s" % (args, kwargs)
  4. return func(*args, **kwargs) #2
  5. return inner

请注意函数inner,它能够接受任意数量和类型的参数并把它们传递给被包装的方法,这让我们能够用这个装饰器来装饰任何方法。

  1. @logger
  2. def foo1(x, y=1):
  3. return x * y
  4. @logger
  5. def foo2():
  6. return 2
  7. foo1(5, 4)
  8. #Arguments were: (5, 4), {}
  9. #20
  10. foo1(1)
  11. #Arguments were: (1,), {}
  12. #1
  13. foo2()
  14. #Arguments were: (), {}
  15. #2

随便调用我们定义的哪个方法,相应的日志也会打印到输出窗口,和我们预期的一样。

13. 带参数的装饰器:

如果decorator本身需要传入参数,那就需要编写一个返回decorator的高阶函数,写出来会更复杂。比如,要自定义log的文本:

  1. def log(text):
  2. def decorator(func):
  3. def wrapper(*args, **kw):
  4. print '%s %s():' % (text, func.__name__)
  5. return func(*args, **kw)
  6. return wrapper
  7. return decorator

这个3层嵌套的decorator用法如下:

  1. @log('execute')
  2. def now():
  3. print '2013-12-25'

执行结果如下:

  1. >>> now()
  2. execute now():
  3. 2013-12-25

和两层嵌套的decorator相比,3层嵌套的效果是这样的:

now = log('execute')(now)

我们来剖析上面的语句,首先执行log('execute'),返回的是decorator函数,再调用返回的函数,参数是now函数,返回值最终是wrapper函数。

14. 装饰器的副作用

以上两种decorator的定义都没有问题,但还差最后一步。因为我们讲了函数也是对象,它有__name__等属性,但你去看经过decorator装饰之后的函数,它们的__name__已经从原来的'now'变成了'wrapper'

  1. >>> now.__name__
  2. 'wrapper'

因为返回的那个wrapper()函数名字就是'wrapper',所以,需要把原始函数的__name__等属性复制到wrapper()函数中,否则,有些依赖函数签名的代码执行就会出错。

不需要编写wrapper.__name__ = func.__name__这样的代码,Python内置的functools.wraps就是干这个事的,所以,一个完整的decorator的写法如下:

  1. import functools
  2. def log(func):
  3. @functools.wraps(func)
  4. def wrapper(*args, **kw):
  5. print 'call %s():' % func.__name__
  6. return func(*args, **kw)
  7. return wrapper

或者针对带参数的decorator:

  1. import functools
  2. def log(text):
  3. def decorator(func):
  4. @functools.wraps(func)
  5. def wrapper(*args, **kw):
  6. print '%s %s():' % (text, func.__name__)
  7. return func(*args, **kw)
  8. return wrapper
  9. return decorator

import functools是导入functools模块。模块的概念稍候讲解。现在,只需记住在定义wrapper()的前面加上@functools.wraps(func)即可。

当然,即使是你用了functools的wraps,也不能完全消除这样的副作用。你会发现,即使是你你用了functools的wraps,你在用getargspec时,参数也不见了。要修正这一问题,我们还得用Python的反射来解决,当然,我相信大多数人的程序都不会去getargspec。所以,用functools的wraps应该够用了。

15. class式的 Decorator

首先,先得说一下,decorator的class方式,还是看个示例:

  1. class myDecorator(object):
  2. def __init__(self, fn):
  3. print "inside myDecorator.__init__()"
  4. self.fn = fn
  5. def __call__(self):
  6. self.fn()
  7. print "inside myDecorator.__call__()"
  8. @myDecorator
  9. def aFunction():
  10. print "inside aFunction()"
  11. print "Finished decorating aFunction()"
  12. aFunction()
  13. # 输出:
  14. # inside myDecorator.__init__()
  15. # Finished decorating aFunction()
  16. # inside aFunction()
  17. # inside myDecorator.__call__()

1)一个是__init__(),这个方法是在我们给某个函数decorator时被调用,所以,需要有一个fn的参数,也就是被decorator的函数。
2)一个是__call__(),这个方法是在我们调用被decorator函数时被调用的。
上面输出可以看到整个程序的执行顺序。

这看上去要比“函数式”的方式更易读一些。

上面这段代码中,我们需要注意这几点:

1)如果decorator有参数的话,__init__() 成员就不能传入fn了,而fn是在__call__的时候传入的。

16. 一些decorator的示例

好了,现在我们来看一下各种decorator的例子:

16.1 给函数调用做缓存

这个例实在是太经典了,整个网上都用这个例子做decorator的经典范例,因为太经典了,所以,我这篇文章也不能免俗。

  1. from functools import wraps
  2. def memo(fn):
  3. cache = {}
  4. miss = object()
  5. @wraps(fn)
  6. def wrapper(*args):
  7. result = cache.get(args, miss)
  8. if result is miss:
  9. result = fn(*args)
  10. cache[args] = result
  11. return result
  12. return wrapper
  13. @memo
  14. def fib(n):
  15. if n < 2:
  16. return n
  17. return fib(n - 1) + fib(n - 2)

上面这个例子中,是一个斐波拉契数例的递归算法。我们知道,这个递归是相当没有效率的,因为会重复调用。比如:我们要计算fib(5),于是其分解成fib(4) + fib(3),而fib(4)分解成fib(3)+fib(2),fib(3)又分解成fib(2)+fib(1)…… 你可看到,基本上来说,fib(3), fib(2), fib(1)在整个递归过程中被调用了两次。

而我们用decorator,在调用函数前查询一下缓存,如果没有才调用了,有了就从缓存中返回值。一下子,这个递归从二叉树式的递归成了线性的递归。

16.2 Profiler的例子

这个例子没什么高深的,就是实用一些。

  1. import cProfile, pstats, StringIO
  2. def profiler(func):
  3. def wrapper(*args, **kwargs):
  4. datafn = func.__name__ + ".profile" # Name the data file
  5. prof = cProfile.Profile()
  6. retval = prof.runcall(func, *args, **kwargs)
  7. #prof.dump_stats(datafn)
  8. s = StringIO.StringIO()
  9. sortby = 'cumulative'
  10. ps = pstats.Stats(prof, stream=s).sort_stats(sortby)
  11. ps.print_stats()
  12. print s.getvalue()
  13. return retval
  14. return wrapper

16.3 注册回调函数

下面这个示例展示了通过URL的路由来调用相关注册的函数示例:

  1. class MyApp():
  2. def __init__(self):
  3. self.func_map = {}
  4. def register(self, name):
  5. def func_wrapper(func):
  6. self.func_map[name] = func
  7. return func
  8. return func_wrapper
  9. def call_method(self, name=None):
  10. func = self.func_map.get(name, None)
  11. if func is None:
  12. raise Exception("No function registered against - " + str(name))
  13. return func()
  14. app = MyApp()
  15. @app.register('/')
  16. def main_page_func():
  17. return "This is the main page."
  18. @app.register('/next_page')
  19. def next_page_func():
  20. return "This is the next page."
  21. print app.call_method('/')
  22. print app.call_method('/next_page')

注意:
1)上面这个示例中,用类的实例来做decorator。
2)decorator类中没有__call__(),但是wrapper返回了原函数。所以,原函数没有发生任何变化。

16.4 给函数打日志

下面这个示例演示了一个logger的decorator,这个decorator输出了函数名,参数,返回值,和运行时间。

  1. from functools import wraps
  2. def logger(fn):
  3. @wraps(fn)
  4. def wrapper(*args, **kwargs):
  5. ts = time.time()
  6. result = fn(*args, **kwargs)
  7. te = time.time()
  8. print "function = {0}".format(fn.__name__)
  9. print " arguments = {0} {1}".format(args, kwargs)
  10. print " return = {0}".format(result)
  11. print " time = %.6f sec" % (te-ts)
  12. return result
  13. return wrapper
  14. @logger
  15. def multipy(x, y):
  16. return x * y
  17. @logger
  18. def sum_num(n):
  19. s = 0
  20. for i in xrange(n+1):
  21. s += i
  22. return s
  23. print multipy(2, 10)
  24. print sum_num(100)
  25. print sum_num(10000000)

上面那个打日志还是有点粗糙,让我们看一个更好一点的(带log level参数的):

  1. import inspect
  2. def get_line_number():
  3. return inspect.currentframe().f_back.f_back.f_lineno
  4. def logger(loglevel):
  5. def log_decorator(fn):
  6. @wraps(fn)
  7. def wrapper(*args, **kwargs):
  8. ts = time.time()
  9. result = fn(*args, **kwargs)
  10. te = time.time()
  11. print "function = " + fn.__name__,
  12. print " arguments = {0} {1}".format(args, kwargs)
  13. print " return = {0}".format(result)
  14. print " time = %.6f sec" % (te-ts)
  15. if (loglevel == 'debug'):
  16. print " called_from_line : " + str(get_line_number())
  17. return result
  18. return wrapper
  19. return log_decorator

但是,上面这个带log level参数的有两具不好的地方,
1) loglevel不是debug的时候,还是要计算函数调用的时间。
2) 不同level的要写在一起,不易读。

我们再接着改进:

  1. import inspect
  2. def advance_logger(loglevel):
  3. def get_line_number():
  4. return inspect.currentframe().f_back.f_back.f_lineno
  5. def _basic_log(fn, result, *args, **kwargs):
  6. print "function = " + fn.__name__,
  7. print " arguments = {0} {1}".format(args, kwargs)
  8. print " return = {0}".format(result)
  9. def info_log_decorator(fn):
  10. @wraps(fn)
  11. def wrapper(*args, **kwargs):
  12. result = fn(*args, **kwargs)
  13. _basic_log(fn, result, args, kwargs)
  14. return wrapper
  15. def debug_log_decorator(fn):
  16. @wraps(fn)
  17. def wrapper(*args, **kwargs):
  18. ts = time.time()
  19. result = fn(*args, **kwargs)
  20. te = time.time()
  21. _basic_log(fn, result, args, kwargs)
  22. print " time = %.6f sec" % (te-ts)
  23. print " called_from_line : " + str(get_line_number())
  24. return wrapper
  25. if loglevel is "debug":
  26. return debug_log_decorator
  27. else:
  28. return info_log_decorator

你可以看到两点,
1)我们分了两个log level,一个是info的,一个是debug的,然后我们在外尾根据不同的参数返回不同的decorator。
2)我们把info和debug中的相同的代码抽到了一个叫_basic_log的函数里,DRY原则。

16.5 一个MySQL的Decorator

下面这个decorator是我在工作中用到的代码,我简化了一下,把DB连接池的代码去掉了,这样能简单点,方便阅读。

  1. import umysql
  2. from functools import wraps
  3. class Configuraion:
  4. def __init__(self, env):
  5. if env == "Prod":
  6. self.host = "coolshell.cn"
  7. self.port = 3306
  8. self.db = "coolshell"
  9. self.user = "coolshell"
  10. self.passwd = "fuckgfw"
  11. elif env == "Test":
  12. self.host = 'localhost'
  13. self.port = 3300
  14. self.user = 'coolshell'
  15. self.db = 'coolshell'
  16. self.passwd = 'fuckgfw'
  17. def mysql(sql):
  18. _conf = Configuraion(env="Prod")
  19. def on_sql_error(err):
  20. print err
  21. sys.exit(-1)
  22. def handle_sql_result(rs):
  23. if rs.rows > 0:
  24. fieldnames = [f[0] for f in rs.fields]
  25. return [dict(zip(fieldnames, r)) for r in rs.rows]
  26. else:
  27. return []
  28. def decorator(fn):
  29. @wraps(fn)
  30. def wrapper(*args, **kwargs):
  31. mysqlconn = umysql.Connection()
  32. mysqlconn.settimeout(5)
  33. mysqlconn.connect(_conf.host, _conf.port, _conf.user, \
  34. _conf.passwd, _conf.db, True, 'utf8')
  35. try:
  36. rs = mysqlconn.query(sql, {})
  37. except umysql.Error as e:
  38. on_sql_error(e)
  39. data = handle_sql_result(rs)
  40. kwargs["data"] = data
  41. result = fn(*args, **kwargs)
  42. mysqlconn.close()
  43. return result
  44. return wrapper
  45. return decorator
  46. @mysql(sql = "select * from coolshell" )
  47. def get_coolshell(data):
  48. ... ...
  49. ... ..

16.6 线程异步

下面量个非常简单的异步执行的decorator,注意,异步处理并不简单,下面只是一个示例。

  1. from threading import Thread
  2. from functools import wraps
  3. def async(func):
  4. @wraps(func)
  5. def async_func(*args, **kwargs):
  6. func_hl = Thread(target = func, args = args, kwargs = kwargs)
  7. func_hl.start()
  8. return func_hl
  9. return async_func
  10. if __name__ == '__main__':
  11. from time import sleep
  12. @async
  13. def print_somedata():
  14. print 'starting print_somedata'
  15. sleep(2)
  16. print 'print_somedata: 2 sec passed'
  17. sleep(2)
  18. print 'print_somedata: 2 sec passed'
  19. sleep(2)
  20. print 'finished print_somedata'
  21. def main():
  22. print_somedata()
  23. print 'back in main'
  24. print_somedata()
  25. print 'back in main'
  26. main()

16.7 超时函数

这个函数的作用在于可以给任意可能会hang住的函数添加超时功能,这个功能在编写外部API调用 、网络爬虫、数据库查询的时候特别有用。

timeout装饰器的代码如下:

  1. # coding=utf-8
  2. # 测试utf-8编码
  3. import sys
  4. reload(sys)
  5. sys.setdefaultencoding('utf-8')
  6. import signal, functools
  7. class TimeoutError(Exception): pass
  8. def timeout(seconds, error_message="Timeout Error: the cmd 30s have not finished."):
  9. def decorated(func):
  10. result = ""
  11. def _handle_timeout(signum, frame):
  12. global result
  13. result = error_message
  14. raise TimeoutError(error_message)
  15. def wrapper(*args, **kwargs):
  16. global result
  17. signal.signal(signal.SIGALRM, _handle_timeout)
  18. signal.alarm(seconds)
  19. try:
  20. result = func(*args, **kwargs)
  21. finally:
  22. signal.alarm(0)
  23. return result
  24. return result
  25. return functools.wraps(func)(wrapper)
  26. return decorated
  27. @timeout(2) # 限定下面的slowfunc函数如果在5s内不返回就强制抛TimeoutError Exception结束
  28. def slowfunc(sleep_time):
  29. a = 1
  30. import time
  31. time.sleep(sleep_time)
  32. return a
  33. # slowfunc(3) #sleep 3秒,正常返回 没有异常
  34. print slowfunc(11) # 被终止

16.8 Trace函数

有时候出于演示目的或者调试目的,我们需要程序运行的时候打印出每一步的运行顺序 和调用逻辑。类似写bash的时候的bash -x调试功能,然后Python解释器并没有 内置这个时分有用的功能,那么我们就“自己动手,丰衣足食”。

Trace装饰器的代码如下:

  1. # coding=utf-8
  2. # 测试utf-8编码
  3. import sys
  4. reload(sys)
  5. sys.setdefaultencoding('utf-8')
  6. import sys,os,linecache
  7. def trace(f):
  8. def globaltrace(frame, why, arg):
  9. if why == "call": return localtrace
  10. return None
  11. def localtrace(frame=1, why=2, arg=4):
  12. if why == "line":
  13. # record the file name and line number of every trace
  14. filename = frame.f_code.co_filename
  15. lineno = frame.f_lineno
  16. bname = os.path.basename(filename)
  17. print "{}({}): {}".format( bname,
  18. lineno,
  19. linecache.getline(filename, lineno)),
  20. return localtrace
  21. def _f(*args, **kwds):
  22. sys.settrace(globaltrace)
  23. result = f(*args, **kwds)
  24. sys.settrace(None)
  25. return result
  26. return _f
  27. @trace
  28. def xxx():
  29. a=1
  30. print a
  31. print 22
  32. print 333
  33. xxx() #调用
  34. #######################################
  35. C:\Python27\python.exe F:/sourceDemo/flask/study/com.test.bj/t2.py
  36. t2.py(31): a=1
  37. t2.py(32): print a
  38. 1
  39. t2.py(33): print 22
  40. 22
  41. t2.py(34): print 333
  42. 333
  43. Process finished with exit code 0

16.9 单例模式

示例代码:

  1. # coding=utf-8
  2. # 测试utf-8编码
  3. # 单例装饰器
  4. import sys
  5. reload(sys)
  6. sys.setdefaultencoding('utf-8')
  7. # 使用装饰器实现简单的单例模式
  8. def singleton(cls):
  9. instances = dict() # 初始为空
  10. def _singleton(*args, **kwargs):
  11. if cls not in instances: #如果不存在, 则创建并放入字典
  12. instances[cls] = cls(*args, **kwargs)
  13. return instances[cls]
  14. return _singleton
  15. @singleton
  16. class Test(object):
  17. pass
  18. if __name__ == '__main__':
  19. t1 = Test()
  20. t2 = Test()
  21. # 两者具有相同的地址
  22. print t1
  23. print t2

16.10 LRUCache

下面要分享的这个LRUCache不是我做的,是github上的一个库,我们在实际环境中有用到。

先来说下这个概念,cache的意思就是缓存,LRU就是Least Recently Used,即最近最少使用,是一种内存管理算法。总结来说这就是一种缓存方法,基于时间和容量。

一般在简单的python程序中,遇到需要处理缓存的情况时最简单的方式,声明一个全局的dict就能解决(在python中应尽量避免使用全局变量)。但是只是简单情况,这种情况会带来的问题就是内存泄漏,因为可能会出现一直不命中的情况。

由此导致的一个需求就是,你要设定这个dict的最大容量,防止发生泄漏。但仅仅是设定最大容量是不够的,设想当你的dict变量已被占满,还是没有命中,该如何处理。

这时就需要加一个失效时间了。如果在指定失效期内没有使用到该缓存,则删除。

综述上面的需求和功能就是LRUCache干的事了。不过这份代码做了更进一层的封装,可以让你直接把缓存功能做为一个装饰器来用。具体实现可以去参考代码,别人实现之后看起来并不复杂 :)

  1. from lru import lru_cache_function
  2. @lru_cache_function(max_size=1024, expiration=15*60)
  3. def f(x):
  4. print "Calling f(" + str(x) + ")"
  5. return x
  6. f(3) # This will print "Calling f(3)", will return 3
  7. f(3) # This will not print anything, but will return 3 (unless 15 minutes have passed between the first and second function call).

代码: https://github.com/the5fire/Python-LRU-cache/blob/master/lru.py

从python3.2开始内置在functools了lru_cache的功能,说明这个需求是很普遍的。

17. 小结

在面向对象(OOP)的设计模式中,decorator被称为装饰模式。OOP的装饰模式需要通过继承和组合来实现,而Python除了能支持OOP的decorator外,直接从语法层次支持decorator。Python的decorator可以用函数实现,也可以用类实现。

decorator可以增强函数的功能,定义起来虽然有点复杂,但使用起来非常灵活和方便。

最后留个小作业:

请编写一个decorator,能在函数调用的前后打印出'begin call''end call'的日志。

再思考一下能否写出一个@log的decorator,使它既支持:

  1. @log
  2. def f():
  3. pass

又支持:

  1. @log('execute')
  2. def f():
  3. pass

18. Refer:

[1] 12步轻松搞定python装饰器

http://python.jobbole.com/81683/

[2] 装饰器

liaoxuefeng.com/wiki/001374738125095c955c1e6d8bb493182103fac9270762a000/001386819879946007bbf6ad052463ab18034f0254bf355000

[3] Python修饰器的函数式编程

http://coolshell.cn/articles/11265.html

[4] Python Decorator Library

https://wiki.python.org/moin/PythonDecoratorLibrary

[5] Python装饰器实例:调用参数合法性验证

http://python.jobbole.com/82114/

[6] Python 装饰器

http://python.jobbole.com/82344/

[7] 两个实用的Python的装饰器

http://blog.51reboot.com/%E4%B8%A4%E4%B8%AA%E5%AE%9E%E7%94%A8%E7%9A%84python%E7%9A%84%E8%A3%85%E9%A5%B0%E5%99%A8/

[8] Python 中的闭包总结

http://dwz.cn/2CiO78

[9] Python 的闭包和装饰器

https://segmentfault.com/a/1190000004461404

[10] Python修饰器的问题

https://segmentfault.com/q/1010000004595899

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/AllinToyou/article/detail/140763
推荐阅读
相关标签
  

闽ICP备14008679号