个人中心
gpt4 book ai didi

python 装饰器的使用与要点

转载 作者：qq735679552 更新时间：2022-09-27 22:32:09 26 4
gpt4 key购买 nike

CFSDN坚持开源创造价值，我们致力于搭建一个资源共享平台，让每一个IT人在这里找到属于你的精彩世界.

这篇CFSDN的博客文章python 装饰器的使用与要点由作者收集整理，如果你对这篇文章有兴趣，记得点赞哟.

1、装饰器使用场景

经常用于有切面需求的场景，比如：插入日志、性能测试、事务处理、缓存、权限校验等场景。装饰器是解决这类问题的绝佳设计，有了装饰器，我们就可以抽离出大量与函数功能本身无关的雷同代码并继续重用.

概括的讲，装饰器的作用就是为已经存在的对象添加额外的功能.

2、为什么需要装饰器

1、先来看一个简单例子：

?
1
2
def foo():
print ( 'i am foo' )

2、增加需求

现在有一个新的需求，希望可以记录下函数的执行日志，于是在代码中添加日志代码:

?
1
2
3
def foo():
     print ( 'i am foo' )
     print ( "foo is running" )

3、又有需求

假设现在有100个函数需要增加这个需求，并且后续可能还要对这一百个函数都增加执行前打印日志的需求，怎么办？还一个个改吗?

当然不了，这样会造成大量雷同的代码，为了减少重复写代码，我们可以这样做，重新定义一个函数：专门处理日志 ，日志处理完之后再执行真正的业务代码.

?
1
2
3
4
5
6
7
8
9
10
11
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(被装饰的函数)就达到了我们想要的效果.

?
1
2
3
4
5
6
7
8
9
def use_logging(func):
     print ( "%s is running" % func.__name__)
     return func
 
@use_logging
def bar():
     print ( 'i am bar' )
 
bar()

3、基础装饰器入门

1、装饰器语法糖

python提供了@符号作为装饰器的语法糖，使我们更方便的应用装饰函数；但使用语法糖要求装饰函数必须return一个函数对象。因此我们将上面的func函数使用内嵌函数包裹并return.

装饰器相当于执行了装饰函数use_loggin后又返回被装饰函数bar,因此bar()被调用的时候相当于执行了两个函数。等价于use_logging(bar)() 。

?
1
2
3
4
5
6
7
8
9
10
11
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) 。

?
1
2
3
4
5
6
7
8
9
10
11
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、函数参数数量不确定

不带参数装饰器版本，这个格式适用于不带参数的装饰器.

经过以下修改，我们已经适应了各种长度和类型的参数。这个版本的装饰器可以装饰任意类型的无参数函数.

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
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、装饰器带参数

带参数的装饰器，这个格式适用于带参数的装饰器.

某些情况我们需要让装饰器带上参数，那就需要编写一个返回一个装饰器的高阶函数，写出来会更复杂。比如:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
#! /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__、参数列表，先看例子:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
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

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
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、实现带参数和不带参数的装饰器自适应

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
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()

3、类装饰器

使用类装饰器可以实现带参数装饰器的效果，但实现的更加优雅简洁,而且可以通过继承来灵活的扩展. 。

1、类装饰器

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
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、继承扩展类装饰器

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
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 )

以上就是python 装饰器的使用与要点的详细内容，更多关于python 装饰器的资料请关注我其它相关文章！ 。

原文链接：https://www.cnblogs.com/hls-code/p/14780807.html 。

最后此篇关于python 装饰器的使用与要点的文章就讲到这里了,如果你想了解更多关于python 装饰器的使用与要点的内容请搜索CFSDN的文章或继续浏览相关文章，希望大家以后支持我的博客！ 。

26 4 0
文章推荐： C语言实现数字游戏
文章推荐： Java中File类中常用方法详解
文章推荐： C字符串函数对应的C++ string操作详解
文章推荐： SpringCloud实现Redis在各个微服务的Session共享问题
qq735679552
个人简介

我是一名优秀的程序员,十分优秀！

滴滴打车优惠券免费领取
滴滴打车优惠券
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com