经验首页 前端设计 程序设计 Java相关 移动开发 数据库/运维 软件/图像 大数据/云计算 其他经验
当前位置:技术经验 » 程序设计 » Python » 查看文章
详解Python装饰器的四种定义形式
来源:jb51  时间:2022/11/28 8:55:54  对本文有异议

前言

装饰器(decorator)在Python框架中扮演着重要角色,是Python中实现切面编程(AOP)的重要手段。

aspect-oriented programming (AOP) ,在不改变代码自身的前提下增加程序功能

不改变代码自身,但需要在函数和类头上加一个标注(annotation),这个标注在Python里叫装饰器,在java里叫注解。
在Python里,一共有四种组合形式。下面一一举例。

用函数装饰函数

采用一个函数定义装饰器:

  1. def decorate(f):
  2. def wrapper(*args):
  3. return f(*args)*2
  4. return wrapper

然后作用在一个函数上:

  1. @decorate
  2. def add(a, b):
  3. return a + b

测试一下效果:

  1. def test_decorate():
  2. sum = add(3, 5)
  3. assert sum == 16

用函数装饰一个类

这里通过装饰器实现单例模式:

  1. def singleton(cls):
  2. instances = {}
  3. def wrapper(*args, **kwargs):
  4. if cls not in instances:
  5. instances[cls] = cls(*args, **kwargs)
  6. return instances[cls]
  7. return wrapper

使用该装饰器:

  1. @singleton
  2. class MyClass:
  3. def method(self):
  4. pass

于是,当你定义多个对象时,返回的是同一实例:

  1. obj = MyClass() # creates a new instance
  2. obj2 = MyClass() # returns the same instance
  3. obj3 = MyClass() # returns the same instance
  4. ...

用类定义装饰器,然后装饰一个函数

先采用类定义一个装饰器:

  1. class Star:
  2. def __init__(self, n):
  3. self.n = n
  4.  
  5. def __call__(self, fn):
  6. @wraps(fn)
  7. def wrapper(*args, **kwargs):
  8. result = fn(*args, **kwargs)
  9. return result
  10. return wrapper

再作用在一个函数上:

  1. @Star(5)
  2. def add(a, b):
  3. return a + b

主要是在类中实现__call__方法。上面例子也可以简化:

  1. class MyDecorator:
  2. def __init__(self, function):
  3. self.function = function
  4. def __call__(self, *args, **kwargs):
  5. # We can add some code
  6. # before function call
  7. self.function(*args, **kwargs)
  8. # We can also add some code
  9. # after function call.
  10. # adding class decorator to the function
  11. @MyDecorator
  12. def function(name, message ='Hello'):
  13. print("{}, {}".format(message, name))

用类定义装饰器,然后装饰一个类

先定义装饰器:

  1. class MyClassDecorator(object):
  2. _instances = dict()
  3.  
  4. def __init__(self, name):
  5. pass
  6.  
  7. def __call__(self, cls):
  8. class WrappedClass(cls):
  9. def say_hello(self):
  10. print(f'Hello: {self.username}')
  11. return WrappedClass

该装饰器给被装饰的类上添加了一个方法,名称为say_hello()。使用如下:

  1. @MyClassDecorator('test')
  2. class MyClass():
  3. def __init__(self, username):
  4. self.username = username

然后:

  1. def test_decoratorforclass():
  2. obj = MyClass('user1')
  3. obj.say_hello()

打印出: Hello: user1

小结

学习类装饰,对Python的内部机制会有更多的了解。如__init__, call, __new__等内置方法。

到此这篇关于Python装饰器的四种定义形式的文章就介绍到这了,更多相关Python装饰器内容请搜索w3xue以前的文章或继续浏览下面的相关文章希望大家以后多多支持w3xue!

 友情链接:直通硅谷  点职佳  北美留学生论坛

本站QQ群:前端 618073944 | Java 606181507 | Python 626812652 | C/C++ 612253063 | 微信 634508462 | 苹果 692586424 | C#/.net 182808419 | PHP 305140648 | 运维 608723728

W3xue 的所有内容仅供测试,对任何法律问题及风险不承担任何责任。通过使用本站内容随之而来的风险与本站无关。
关于我们  |  意见建议  |  捐助我们  |  报错有奖  |  广告合作、友情链接(目前9元/月)请联系QQ:27243702 沸活量
皖ICP备17017327号-2 皖公网安备34020702000426号