python 装饰器

一、我们在python语言中

 常用@classmethod、@staticmethod这个装饰器,装饰器的作用简单来讲就是我们在很多地方可能都会用到一下相同的功能,这时我们会有两种想法。

1. 将这部分相同的功能封装成函数,然后在使用这个功能的时候程序中调用这个函数。
2.  将这部分相同的功能封装成装饰器,然后在使用这个功能的时候在需要处理的方法前加上对应的装饰器。

 

这两种方式各有好处,我认为
1中的方式更适合新加功能,通过单独新加函数实现功能,然后在需要的位置使用。
2中的方式更适合固定且经常使用的功能,比如说日志记录,执行时间计算等,只需要在需要用到该功能的时候直接在该方法前加装饰器即可。

二、装饰器

下面我们先看个例子:

import time
# Python学习交流群 815624229
def origin_fun(a, b):
    start_time = time.time()
    s = a + b
    exec_time = time.time() - start_time
    print('执行时间: {}'.format(exec_time))
    return s

 

以上是原函数,我们要得到传入两个参数的加和操作的时间消耗,我们如果需要对很多操作进行时间记录,有两个方法

1、拆出函数实现

def func(a, b):
    return a+b

def get_begin_date():
    return time.time()

def final_func(a, b):
    start_time = get_begin_date()
    s = func(a, b)
    exec_time = get_begin_date() - start_time
    print('执行时间: {}'.format(exec_time))
    return s

 

可以发现我们这样是可以实现的, 但是这样再使用的时候还是有点繁琐,接下来我们用装饰器实现

2、装饰器实现

(1)我们先来介绍一下装饰器的一个结构:

简单理解就是将添加了装饰器的函数(func)传入装饰器函数(decorator)中,在装饰器函数中创建内置函数(wrapper),内置函数实现通用的功能并真正执行传入的函数(func)之后将返执行的函数结果在内置函数(wrapper)中返回,在装饰器函数(decorator)中将装饰器的内置函数(wrapper)作为返回对象返回

def decorator(func):
    def wrapper(*args, **kwargs):
        """
        define args and kwargs
        """
        # 通用的功能,这里我们认为输出传入的函数名称就是通用功能
        print(func.__name__)
        # 执行传入的方法
        return func(*args, **kwargs)
    # 返回当前方法的内置函数
    return wrapper

@decorator
def func(a, b):
    return a + b

 

这么一看的话大家应该就会觉得装饰器也很好理解了。

(2)下面我们就将上述的例子通过装饰器实现

a. 无参装饰器

def time_local(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        f = func(*args, **kwargs)
        exec_time = time.time() - start_time
        print('执行时间: {}'.format(exec_time))
        return f
    return wrapper
@time_local
def func1(a, b):
    return a+b
    
@time_local
def func2(a, b):
    return a*b

 

上述代码块就是我们的装饰器实现,可以发现我们在计算其他操作的消耗的时间的时候只需要在函数前添加装饰器就可以了,这样就代码的改动就会尽可能少的改动,这就是简单的装饰器的实现方式和应用方式。

b. 有参装饰器
高级一点的我们可能需要对装饰器进行传参,这样的装饰器就是在无参装饰器外再包一层函数,如下

def time_local_args(level):
    def time_local(func):
        def wrapper(*args, **kwargs):
            if level:
                start_time = time.time()
                f = func(*args, **kwargs)
                exec_time = time.time() - start_time
                print('执行时间: {}'.format(exec_time))
            else:
                f = func(*args, **kwargs)
            return f
        return wrapper
    return time_local

@time_local_args(level=True)
def func(a, b):
    return a+b

 

这样一看的话就很清晰了吧,装饰器就是一个函数内嵌套一个内置函数,在内置函数中实现通用功能以及执行被装饰的函数,嵌套函数的返回值为被装饰的函数,装饰器函数返回内置函数对象。

c. 再优雅一些的写法, 传参并封装成类且有参数的装饰器

class TimeLocal:
    def __init__(self, level):
        self._level = level

    def __call__(self, func, *args, **kwargs):
        def wrapper(*args, **kwargs):
            if self._level:
                start_time = time.time()
                f = func(*args, **kwargs)
                exec_time = time.time() - start_time
                print('执行时间: {}'.format(exec_time))
            else:
                f = func(*args, **kwargs)
            return f
        return wrapper
        
@TimeLocal(level=True)
def func4(a, b):
    return b / a

 

d.偏函数与类结合构建装饰器

事实上,Python 对某个对象是否能通过装饰器( @decorator)形式使用只有一个要求:decorator 必须是一个“可被调用(callable)的对象。

对于这个 callable 对象,我们最熟悉的就是函数了。

除函数之外,类也可以是 callable 对象,只要实现了call 函数(上面几个例子已经接触过了)。

还有容易被人忽略的偏函数其实也是 callable 对象。

接下来就来说说,如何使用 类和偏函数结合实现一个与众不同的装饰器。

如下所示,DelayFunc 是一个实现了 call 的类,delay 返回一个偏函数,在这里 delay 就可以做为一个装饰器。(以下代码摘自 Python工匠:使用装饰器的小技巧)

import time
import functools

class DelayFunc:
    def __init__(self,  duration, func):
        self.duration = duration
        self.func = func

    def __call__(self, *args, **kwargs):
        print(f'Wait for {self.duration} seconds...')
        time.sleep(self.duration)
        return self.func(*args, **kwargs)

    def eager_call(self, *args, **kwargs):
        print('Call without delay')
        return self.func(*args, **kwargs)

def delay(duration):
    """
    装饰器:推迟某个函数的执行。
    同时提供 .eager_call 方法立即执行
    """
    # 此处为了避免定义额外函数,
    # 直接使用 functools.partial 帮助构造 DelayFunc 实例
    return functools.partial(DelayFunc, duration)
   
@delay(duration=2)
def add(a, b):
    return a+b

 

以上就是我们装饰器的理解以及几张实现的方式,最后我们看一下用在类上的装饰器的代码

instances = {}

def singleton(cls):
    def get_instance(*args, **kw):
        cls_name = cls.__name__
        print('===== 1 ====')
        if not cls_name in instances:
            print('===== 2 ====')
            instance = cls(*args, **kw)
            instances[cls_name] = instance
        return instances[cls_name]
    return get_instance

@singleton
class User:
    _instance = None

    def __init__(self, name):
        print('===== 3 ====')
        self.name = name
print(User('sdma'))

>>> ===== 1 ====
>>> ===== 2 ====
>>> ===== 3 ====
>>> <__main__.User object at 0x000001EAC67AEC48>

 

本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:python 装饰器 - Python技术站

(0)
上一篇 2023年4月2日 下午5:49
下一篇 2023年4月2日 下午5:49

相关文章

  • 【Python】Flask + MySQL 实现用户注册,登录和登出

    今天用Flask + MySQL 实现用户注册,登录和登出。 一、实战场景 Flask 框架实现用户的注册,登录和登出。 二、主要知识点 flask_login 插件使用 SQLAlchemy 基础操作 用户基础类设计 Flask 读取配置文件 蓝图注册与使用 wtforms 表单提交数据 wtforms 表单验证 Bootstrap 集成 Jinjia2 …

    Python开发 2023年4月2日
    00
  • Python之字典查找元素(3种方式)

    本文描述的是查找字典的某一个元素(字典遍历元素请点击->这里) 上下文代码 smart_girl = {“name”:”yuan wai”, “age”: 25,”sex”:”女”}   第一种方式:[] 注意:这种方式,如果找不到对应的key,会报一个KeyError错误 smart_girl[“name”] # []传入key   第二种方式:ge…

    Python开发 2023年3月31日
    00
  • 8条非常实用的python代码案例,初学者必备知识点!

    给大家推荐八个非常实用的Python案例,希望大家看过后能够有所收获! 1、合并两个字典 Python3.5之后,合并字典变得容易起来,我们可以通过**符号解压字典,并将多个字典传入{}中,实现合并。 def Merge(dict1,dict2): res = {**dict1,**dict2} return res # 两个字典 # Python学习交流1…

    Python开发 2023年4月2日
    00
  • 这将使你见过最全面的Python制作GUI学生管理系统教程

    表弟大学快毕业了,学了一个学期Python居然还不会写学生管理系统,真的给我丢脸啊,教他又不肯学,还让我直接给他写,我真想两巴掌上去,最终还是写了给他,谁让他是我表弟呢,关键时候还是得帮他一把!       写完了放在那也是放着,所以今天分享给大家吧! 话不多说,咱们直接开始吧!     代码解析 一、登录页面 1、定义登录类及初始化对象 首先导入咱们需要用…

    2023年4月2日
    00
  • 用Python自动实现图表可视化操作,提高工作效率,又能有更多的时间摸鱼了~

    在数据分析过程中,一般提取数据库里面的数据时候,拿着表格数据反复思索,希望能够根据自己所想立马生成一张数据可视化的图表来更直观的呈现数据。 但想要进行数据可视化的时候,往往需要调用很多的库与函数,还需要数据转换以及大量的代码处理编写。这都是十分繁琐的工作,确实只为了数据可视化我们不需要实现数据可视化的工程编程,这都是数据分析师以及拥有专业的报表工具来做的事情…

    Python开发 2023年4月2日
    00
  • 20行Python代码实现爬取起点小说

    相信在座各位应该没有几个不看小说的吧,嘿嘿~一般来说咱们书荒的时候怎么办?自然是去起某点排行榜先找到小说名字,然后再找度娘一搜,哎 ,笔趣阁就出来答案了,美滋滋~ 但是那多麻烦,咱们直接用python,直接全部下载下来慢慢看不就好了~小孩子才做选择,成年人选择都要… 好了,不啰嗦了,等下大家要骂我了~ 准备工作 环境模块 咱们没装软件的装一下软件,没装模块的…

    Python开发 2023年4月2日
    00
  • 重温Python基础——列表

    哈喽,兄弟们,今天让我们一起来重温一下Python基础中的列表,看看你还记得多少呢? 列表是什么 列表有一系列特定顺序排列的元素组成。可以包含字母表中的所有字母,数字等。其中的元素之间可以没有任何联系,列表通常包含多种元素。 在Python中用方括号([])表示列表,用逗号分隔其中的元素。 a=[‘wo’,’ai’,’xue’,’xi’,13,14]   访…

    Python开发 2023年4月2日
    00
  • Python基于PC版微信实现机器人

    闲着没事总想搞点事情,这不是web版的微信有专门的实时收发消息api,可以用Python直接调用,但是现在基本上登录不了web版微信,所以就用不了了。 但这不是还有windows版的微信吗,于是在基于前辈写好的第三方库,今天我们来实现一下实时收发信息,实现自动聊天机器人。 效果展示 微信实时收发消息 1、第三方库 基于第三方pc-wechat-hook-ht…

    Python开发 2023年4月2日
    00
合作推广
合作推广
分享本页
返回顶部