python 装饰器的使用与要点
一、装饰器使用场景
经常用于有切面需求的场景,比如:插入日志、性能测试、事务处理、缓存、权限校验等场景。装饰器是解决这类问题的绝佳设计,有了装饰器,我们就可以抽离出大量与函数功能本身无关的雷同代码并继续重用。
概括的讲,装饰器的作用就是为已经存在的对象添加额外的功能。
二、为什么需要装饰器
1、先来看一个简单例子:
def foo():
print('i am foo')
2、增加需求
现在有一个新的需求,希望可以记录下函数的执行日志,于是在代码中添加日志代码:
def foo():
print('i am foo')
print("foo is running")
3、又有需求
假设现在有100个函数需要增加这个需求,并且后续可能还要对这一百个函数都增加执行前打印日志的需求,怎么办?还一个个改吗?
当然不了,这样会造成大量雷同的代码,为了减少重复写代码,我们可以这样做,重新定义一个函数:专门处理日志 ,日志处理完之后再执行真正的业务代码。
def use_logging(func):
print("%s is running" % func.__name__)
func()
def bar():
print('i am bar')
use_logging(bar)
运行结果:
#bar is running
#i am bar
函数use_logging就是装饰器,它把执行真正业务方法的func包裹在函数里面,看起来像bar被use_logging装饰了。在这个例子中,函数进入和退出时 ,被称为一个横切面(Aspect),这种编程方式被称为面向切面的编程(Aspect-Oriented Programming)。
通过以上use_logging函数我们增加了日志功能,不管以后有多少函数需要增加日志或者修改日志的格式我们只需要修改use_logging函数,并执行use_logging(被装饰的函数)就达到了我们想要的效果。
def use_logging(func):
print("%s is running" % func.__name__)
return func
@use_logging
def bar():
print('i am bar')
bar()
三、基础装饰器入门
1、装饰器语法糖
python提供了@符号作为装饰器的语法糖,使我们更方便的应用装饰函数;但使用语法糖要求装饰函数必须return一个函数对象。因此我们将上面的func函数使用内嵌函数包裹并return。
装饰器相当于执行了装饰函数use_loggin后又返回被装饰函数bar,因此bar()被调用的时候相当于执行了两个函数。等价于use_logging(bar)()
def use_logging(func):
def _deco():
print("%s is running" % func.__name__)
func()
return _deco
@use_logging
def bar():
print('i am bar')
bar()
2、对带参数的函数进行装饰
现在我们的参数需要传入两个参数并计算值,因此我们需要对内层函数进行改动传入我们的两个参数a和b,等价于use_logging(bar)(1,2)
def use_logging(func):
def _deco(a,b):
print("%s is running" % func.__name__)
func(a,b)
return _deco
@use_logging
def bar(a,b):
print('i am bar:%s'%(a+b))
bar(1,2)
我们装饰的函数可能参数的个数和类型都不一样,每一次我们都需要对装饰器做修改吗?这样做当然是不科学的,因此我们使用python的变长参数*args和**kwargs来解决我们的参数问题。
3、函数参数数量不确定
不带参数装饰器版本,这个格式适用于不带参数的装饰器。
经过以下修改,我们已经适应了各种长度和类型的参数。这个版本的装饰器可以装饰任意类型的无参数函数。
def use_logging(func):
def _deco(*args,**kwargs):
print("%s is running" % func.__name__)
func(*args,**kwargs)
return _deco
@use_logging
def bar(a,b):
print('i am bar:%s'%(a+b))
@use_logging
def foo(a,b,c):
print('i am bar:%s'%(a+b+c))
bar(1,2)
foo(1,2,3)
4、装饰器带参数
带参数的装饰器,这个格式适用于带参数的装饰器。
某些情况我们需要让装饰器带上参数,那就需要编写一个返回一个装饰器的高阶函数,写出来会更复杂。比如:
#! /usr/bin/env python
# -*- coding:utf-8 -*-
# __author__ = "TKQ"
def use_logging(level):
def _deco(func):
def __deco(*args, **kwargs):
if level == "warn":
print "%s is running" % func.__name__
return func(*args, **kwargs)
return __deco
return _deco
@use_logging(level="warn")
def bar(a,b):
print('i am bar:%s'%(a+b))
bar(1,3)
# 等价于use_logging(level="warn")(bar)(1,3)
5、functools.wraps
使用装饰器极大地复用了代码,但是他有一个缺点就是原函数的元信息不见了,比如函数的docstring、__name__、参数列表,先看例子:
def use_logging(func):
def _deco(*args,**kwargs):
print("%s is running" % func.__name__)
func(*args,**kwargs)
return _deco
@use_logging
def bar():
print('i am bar')
print(bar.__name__)
bar()
#bar is running
#i am bar
#_deco
#函数名变为_deco而不是bar,这个情况在使用反射的特性的时候就会造成问题。因此引入了functools.wraps解决这个问题。
使用functools.wraps:
import functools
def use_logging(func):
@functools.wraps(func)
def _deco(*args,**kwargs):
print("%s is running" % func.__name__)
func(*args,**kwargs)
return _deco
@use_logging
def bar():
print('i am bar')
print(bar.__name__)
bar()
#result:
#bar is running
#i am bar
#bar ,这个结果是我们想要的。OK啦!
6、实现带参数和不带参数的装饰器自适应
import functools
def use_logging(arg):
if callable(arg):#判断参入的参数是否是函数,不带参数的装饰器调用这个分支
@functools.wraps(arg)
def _deco(*args,**kwargs):
print("%s is running" % arg.__name__)
arg(*args,**kwargs)
return _deco
else:#带参数的装饰器调用这个分支
def _deco(func):
@functools.wraps(func)
def __deco(*args, **kwargs):
if arg == "warn":
print "warn%s is running" % func.__name__
return func(*args, **kwargs)
return __deco
return _deco
@use_logging("warn")
# @use_logging
def bar():
print('i am bar')
print(bar.__name__)
bar()
三、类装饰器
使用类装饰器可以实现带参数装饰器的效果,但实现的更加优雅简洁,而且可以通过继承来灵活的扩展.
1、类装饰器
class loging(object):
def __init__(self,level="warn"):
self.level = level
def __call__(self,func):
@functools.wraps(func)
def _deco(*args, **kwargs):
if self.level == "warn":
self.notify(func)
return func(*args, **kwargs)
return _deco
def notify(self,func):
# logit只打日志,不做别的
print "%s is running" % func.__name__
@loging(level="warn")#执行__call__方法
def bar(a,b):
print('i am bar:%s'%(a+b))
bar(1,3)
2、继承扩展类装饰器
class email_loging(Loging):
'''
一个loging的实现版本,可以在函数调用时发送email给管理员
'''
def __init__(self, email='admin@myproject.com', *args, **kwargs):
self.email = email
super(email_loging, self).__init__(*args, **kwargs)
def notify(self,func):
# 发送一封email到self.email
print "%s is running" % func.__name__
print "sending email to %s" %self.email
@email_loging(level="warn")
def bar(a,b):
print('i am bar:%s'%(a+b))
bar(1,3)
(资源库 www.zyku.net)
原文链接:https://www.cnblogs.com/hls-code/p/14780807.html
上一篇:Pytorch测试神经网络时出现 RuntimeError:的解决方案
栏 目:Python教程
下一篇:Python趣味挑战之turtle库绘画飘落的银杏树
本文标题:python 装饰器的使用与要点
本文地址:https://www.zyku.net/python/9904.html
您可能感兴趣的文章
- 03-01小米11ultra背屏副屏预览开启方法
- 09-14华为手表怎么配对小米手机
- 01-12PicsArt-PicsArt应用软件功能介绍
- 01-11短腿熊旅行-短腿熊旅行应用软件功能介
- 12-28三河质控-三河质控应用软件功能介绍
- 11-29一行代码实现网页直接跳转抖音APP
- 01-11旭日OA-旭日OA应用软件功能介绍
- 07-22阿里云CentOS报错Redirecting to /bin
- 10-21vivox70pro+开启双击熄屏教程一览
- 01-15小米12如何设置息屏样式
- 01-19魔法蛋糕店-魔法蛋糕店应用软件功能介
- 03-24小米11相机连拍方法
- 02-18小米智能门锁设置密码教程
- 07-26微信程序开发之-WeixinJSBridge调用
- 01-25华为mate40pro设置注视屏幕不熄屏教程
- 02-10详解Python魔法方法之描述符类
- 09-14SCP选择二进制,二进制传输与文本传输
- 01-28华为手机息屏关闭网络使用方法
- 04-26小米11自定义锁屏签名教程
- 11-16钉钉撤销请假申请方法一览
最近更新
阅读排行
猜你喜欢
- 10-29苹果手机ios15系统在哪更新
- 07-12mysql中格式化数字详解
- 02-12小米11深色模式设置教程
- 01-17PP抓娃娃-PP抓娃娃应用软件功能介绍
- 02-27帝国CMS多图上传插件去掉图片链接的方
- 01-08GT次元壁纸-GT次元壁纸应用软件功能介
- 01-26oppo微信分身使用方法
- 02-15ios14设置自动调节屏幕亮度方法
- 01-09爱清理优化助手-爱清理优化助手应用软
- 03-20帝国CMS首页调用最新会员、活跃会员与