Python装饰器——提高

上一篇文章我们对装饰器有了初步了解,接下来开始是对装饰器的进一步实践

1.少了元信息怎么办

当你写了一个装饰器作用在某个函数上,但是这个函数的重要的元信息比如名字、文档字符串、注解和参数签名都丢失了。这到底是怎样的一种现象呢?直接上代码吧。

# -*- coding:UTF-8 -*- 
def show_me(func):
    def wrapper():
        print "It is in wrapper."
        func()
    return wrapper

@show_me
def func1():
    '''
    this is func1
    '''
    print "running func1"

if __name__ == '__main__':
    func1()
    print func1.__name__
    print func1.__doc__

打印结果:


元数据

我们可以看到func1.__name__输出为wrapperfunc1.__doc__输出为None,哇,全乱套了,怎么办。不要怕,Python大法自有办法。

温馨提醒:任何时候你定义装饰器的时候,都应该使用 functools 库中的 @wraps 装饰器来注解底层包装函数。

也就是说,装饰器函数应该这样做补充:

from functools import wraps ##导入这个包

def show_me(func):
    @wraps(func)
    def wrapper():
        print "It is in wrapper."
        func()
    return wrapper

加上这个后,你可以再运行一下,应该会得到完整的元数据.

2.被装饰函数有参数

我们可以看到,上文的被装饰函数(也就是func函数)是没有参数的,但是在实际应用中,我们大多数的函数都会有参数,这就要用到Python中的可变参数了。那么对于被装饰函数带有参数的装饰器该怎么写呢,可见如下代码:

# -*- coding:UTF-8 -*- 
from functools import wraps

def show_me(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        print "强哥好帅!"
        return func(*args, **kwargs)
    return wrapper

@show_me
def func4(a, b=2):
    print "a = %s"%a
    print "b = %s"%b


if __name__ == '__main__':
    func4(3, 8)

运行结果如下:

>>强哥好帅!
>>a = 3
>>b = 8

可变参数是个好东西,在Python中(星号)*和参数有许多有趣的用法,值得一看。

3.装饰器带有参数

有时候,@语法糖后面还可以带参数。比如某个函数功能只有拥有经理权限的人才能访问。可以编写这样的一个装饰器@has_permission("manager"),下面来举一个简单粗鲁的栗子:

# -*- coding:UTF-8 -*- 
from functools import wraps

def has_permission(position=""):
    def decorator(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            if position == "manager":
                print "我是经理我怕谁!"
                return func(*args, **kwargs)
            else:
                print "You don not have permission to access!"
        return wrapper
    return decorator

@has_permission(position="manager")
def func5(name=""):
    print "My name is : %s"%name

if __name__ == '__main__':
    func5("zhujq")

结果为:

>>我是经理我怕谁!
>>My name is : zhujq

position改为其他时(比如“Boss”),程序便只能输出:

 You don not have permission to access!
4.其他

Python还提供了类装饰器与@staticmethod@classmethod@property和这三个在面向对象编程中常用的装饰器,我们下次再讲吧~~

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,456评论 5 477
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,370评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,337评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,583评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,596评论 5 365
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,572评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,936评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,595评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,850评论 1 297
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,601评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,685评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,371评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,951评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,934评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,167评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 43,636评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,411评论 2 342

推荐阅读更多精彩内容

  • 本文为《爬着学Python》系列第四篇文章。从本篇开始,本专栏在顺序更新的基础上,会有不规则的更新。 在Pytho...
    SyPy阅读 2,489评论 4 11
  • 前言 人生苦多,快来 Kotlin ,快速学习Kotlin! 什么是Kotlin? Kotlin 是种静态类型编程...
    任半生嚣狂阅读 26,139评论 9 118
  • 假以时日,我相信装饰器一定会成为Python这门编程语言一个更加强大的功能。到目前为止,我觉得到我所看到的有关介绍...
    TypingQuietly阅读 5,415评论 2 71
  • 本文为《爬着学Python》系列第十篇文章。 在实际操作中,可能函数是我们几乎唯一的实现操作的方式,这是因为函数能...
    SyPy阅读 5,445评论 0 8
  • 提到中国古代诗歌,大家便会首提唐诗宋词。过去的三天时间里跟大家讲完了唐诗,怎么说也略不过宋词了。 宋词,这个可以与...
    爱在夏末秋初阅读 389评论 0 5