Python的装饰器

在Python中,函数是一等对象,因此你可以将函数与字符串,整数等对象看成一样。
比如你可以将函数赋值给一个变量:

>>> def square(n):
...     return n * n
>>> square(4)
16
>>> alias = square
>>> alias(4)
16

函数作为一等对象的作用在于,你可以将函数作为参数传递给另外一个函数,或者从另外一个函数中返回。比如Python内建的map函数:

you pass it a function and a list, and map creates a new list by calling your function individually for each item in the list you gave it. Here’s an example that uses our square function from above:

>>> numbers = [1, 2, 3, 4, 5]
>>> map(square, numbers)
[1, 4, 9, 16, 25]

接受函数作为参数或者返回函数的函数被称为:higher-order function。我们可以使用higher-order函数来改变传入的函数的对外行为。

比如:假设我们有一个函数,该函数的会被频繁的调用,因此开销非常大。

>>> def fib(n):
...     "Recursively (i.e., dreadfully) calculate the nth Fibonacci number."
...     return n if n in [0, 1] else fib(n - 2) + fib(n - 1)

最好的方法是保存函数的计算结果,因此如果我们需要计算一个n的函数值,不需要反复计算n之前的值。
有很多种方式来实现保存计算结果,比如可以使用一个dictionary。但是这需要外部来保存这些计算结果,这样会造成一定的紧耦合。因此如果fib函数自身保存计算的结果,外部只需要调用这个函数就可以了。这种技术叫做:memorization

Python采用一种装饰器的方式来实现 memorization,因为我们可以将函数作为参数传递给另外一个函数,因此我们可以写一个通用的外部函数,该函数接收一个函数参数并返回一个带有memorized版本。

def memoize(fn):
    stored_results = {}
    def memoized(*args):
        try:
            # try to get the cached result
            return stored_results[args]
        except KeyError:
            # nothing was cached for those args. let's fix that.
            result = stored_results[args] = fn(*args)
            return result
    return memoized

有了带有memorization版本的函数,可以这样调用:

def fib(n):
   return n if n in [0, 1] else fib(n - 2) + fib(n - 1)
fib = memoize(fib)

而这里还可以采用decorator来简写:

@memoize
def fib(n):
   return n if n in [0, 1] else fib(n - 2) + fib(n - 1)

如果一个函数有多个装饰器,则他们采用从底部到顶部的方式起作用。比如假设我们还有一个higher-order函数来帮助调试。

def make_verbose(fn):
    def verbose(*args):
        # will print (e.g.) fib(5)
        print '%s(%s)' % (fn.__name__, ', '.join(repr(arg) for arg in args))
        return fn(*args) # actually call the decorated function
    return verbose

下面两个形式的代码片段做同样的事情:

@memoize
@make_verbose
def fib(n):
    return n if n in [0, 1] else fib(n - 2) + fib(n - 1)
def fib(n):
    return n if n in [0, 1] else fib(n - 2) + fib(n - 1)
fib = memoize(make_verbose(fib))

更加有趣的是,你可以给装饰器传递额外的参数。比如我们不满足一些简单的memorization,而是想要存储函数结果到memcached中,因此需要给装饰器传入memcached的服务器地址等参数信息。

@memcached('127.0.0.1:11211')
def fib(n):
    return n if n in [0, 1] else fib(n - 2) + fib(n - 1)

上述代码等价于:

fib = memcached('127.0.0.1:11211')(fib)

Python有些内建的装饰器函数,比如:classmethod装饰器,该装饰器下的函数等价于java中的静态函数。

class Foo(object):
    SOME_CLASS_CONSTANT = 42

    @classmethod
    def add_to_my_constant(cls, value):
        # Here, `cls` will just be Foo, but if you called this method on a
        # subclass of Foo, `cls` would be that subclass instead.
        return cls.SOME_CLASS_CONSTANT + value

Foo.add_to_my_constant(10) # => 52

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

推荐阅读更多精彩内容

  • 注意: 如果对闭包不了解的同学请移步到这里先,因为装饰器要通过闭包来实现 前言 刚开始学Python的时候,装饰器...
    DamonYJ阅读 554评论 0 3
  • 1.需要了解一个函数返回的是一个对象。-----在返回值后加 ( ) 就可以执行对象2.是在一个函数中是传入的是一...
    傻傻笨笨宝宝阅读 216评论 0 0
  • http://python.jobbole.com/85231/ 关于专业技能写完项目接着写写一名3年工作经验的J...
    燕京博士阅读 7,539评论 1 118
  • 安阳是一个初二学生,善良大方。她的班上最近转来了一个女孩方惠惠,大方开朗,热情助人。 方惠惠长得还不...
    DYW卫宝阅读 337评论 0 0
  • 谁愿冲锋。 身边有这么位友人,碰面即刻点头哈腰蒙娜丽莎式笑颜,流水线三部曲,巧极。慢动作缱绻眼眸,重拨仪态,一侧而...
    老君眉好吃阅读 616评论 0 1