python 装饰器

python 装饰器

描述

  • python 中一切皆对象,函数也可以当作参数传递
  • 装饰器就是接受一个函数作为参数,添加功能后返回一个新函数的函数或类。
  • python 中使用 @ 来使用装饰器。(@ 只是装饰器的语法糖,可以等价替代为其他)

装饰器经常用到的功能就是在代码中添加 log 日志。

实例

下面简单编写一个记录函数耗时的装饰器。

使用函数编写一个装饰器

import time

def log_time(func):     # 接受一个函数作为参数
    def _log(*args, **kwargs):
        beg = time.time()
        res = func(*args, **kwargs)
        print('use time:{}'.format(time.time() - beg))
        return res
    return _log

@log_time       # 装饰器语法糖
def mysleep():
    time.sleep(1)

mysleep()

""" @log_time 等价于:
    newsleep = log_time(mysleep)
    newsleep()
"""

使用类编写一个装饰器

import time

class Logtime:
    def __call__(self, func):   #魔术方法
        def _log(*args, **kwargs):
            beg = time.time()
            res = func(*args, **kwargs)
            print('use time: {}'.format(time.time() - beg))
            return res
        return _log

@Logtime()      # 注意这里和函数装饰器的区别
def mysleep2():
    time.sleep(1)

mysleep2()

问题:如何给装饰器增加参数

答: 使用类装饰器我们可以比较方便实现装饰器参数 -- 通过定义 __init__ 添加参数。

以上面的代码为例,给装饰器添加一个参数

import time

class Logtime:

    def __init__(self, use_int = False):    # 定义默认参数
        self.use_int = use_int

    def __call__(self, func):   #魔术方法
        def _log(*args, **kwargs):
            beg = time.time()
            res = func(*args, **kwargs)
            if self.use_int:
                print('use time: {}'.format(int(time.time() - beg)))
            else:
                print('use time: {}'.format(time.time() - beg))
            return res
        return _log

@Logtime(True)      # 装饰器参数
def mysleep2():
    time.sleep(1)

mysleep2()

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏量子位

Python官方中文文档上线了:各种教程已汉化,不用苦等野生翻译了

从今往后,不论是版本新变化,入门教程,语法讲解,Python模块安装指南……各种各样的手册,都可以直接看中文了。

50530
来自专栏量子位

用英伟达Jetson Nano运行PyTorch & Fast.ai丨手把手教程+第三方工具

Fast.ai是在PyTorch上非常好用的深度学习库,来自MOOC平台Fast.ai,只要一个API,就包含了所有常见的深度学习应用。

53910
来自专栏linxu shell指南

yum简单修复指南

由于yum是强依赖于python,一旦pyhon环境改变(即默认的python bin程序路径改变),基本yum是跟着殉情的。

30730
来自专栏诸葛青云的专栏

Python入门必学,用Python练习画个美队盾牌

那问题来了,作为一个男人,怎么能忍?虽然是个明星,虽然是假的,虽然已经挽回不了败局了……那我就送个美队的盾牌给你吧……

23920
来自专栏程序员的成长之路

实地探访:月薪 12K 的北京程序员是怎么工作生活的?

在当今社会,有网络的地方就有江湖,有江湖的地方就有程序员。我给大家在这里讲一个生活在北京的程序员,也就是生活中时常被提起的爱受伤的傻小子——白公子。

10050
来自专栏诸葛青云的专栏

Python12306自动抢票下单,五一旅游回家就选Python!

X新闻者从中国铁路上海局集团有限公司获悉,五一的前8天(4月21日至28日)火车票已预售80万张。,东北、西南等多个方向尚有余票,铁路部门将采取动车组重联或增开...

29740
来自专栏Python无止境

视频当道的时代,这些珍藏的优质 Python 播客值得推荐

我国互联网的发展道路与欧美不同,在内容的形式上,我们似乎实现了跨越式的发展——早早进入了移动互联网时代,直播和短视频等形式的内容成为了潮流,而文字形式的博客(b...

12030
来自专栏量子位

Python之父称996惨无人道,系列项目霸榜GitHub,某宝商家开卖996文化衫

五天来,该项目已经获得12万程序员支持,拥有996.XXX系列至少4个衍生项目,持续霸占趋势榜第一名,Google可以搜到超过6万条结果。

9220
来自专栏Python无止境

如何在 Python 里面精确四舍五入?

花下猫语:如何精确地计算浮点数?这是计算机科学的大难题。那 Python 是如何处理浮点数的四舍五入问题的呢?今天分享的文章,对此展开了深入的剖析。

19420
来自专栏量子位

用C++实现强化学习,速度不亚于Python,这里有个框架可用

一位名叫Isaac Poulton的英国小哥,开源了一个名为CppRL的C++强化学习框架。

15520

扫码关注云+社区

领取腾讯云代金券

年度创作总结 领取年终奖励