python：类与装饰器

超级大猪

发布于 2019-11-21 20:31:27

6060

发布于 2019-11-21 20:31:27

简单装饰

在python中，装饰器可以是一个类。就是这么任性。要把一个类做为装饰器是很简单的。这里借鉴一个很有意思的代码，python管道语法糖的实现，作者JulienPalard 项目

class Pipe:
    def __init__(self, function):
        self.function = function
        functools.update_wrapper(self, function)

    def __ror__(self, other):
        return self.function(other)

    def __call__(self, *args, **kwargs):
        return Pipe(lambda x: self.function(x, *args, **kwargs))

可以这么用：

@Pipe
def average(iterable):
    """
    Build the average for the given iterable, starting with 0.0 as seed
    Will try a division by 0 if the iterable is empty...
    """
    total = 0.0
    qte = 0
    for x in iterable:
        total += x
        qte += 1
    return total / qte

[1, 3, 6] | average

神奇啊，神奇啊。我一看这份代码，就爱上了这个简洁的小项目。真是太有BG了。但是，这个项目有一个小问题。那就是，它不能装饰类方法。

装饰类方法

类的函数和普通函数非常相似，在调用的时候会以self关键字传入当前实例作为参数。这是大家都明白的。但值得一提的是，类函数有比较复杂的调用机制，当执行 cls.method的一瞬间，系统会调用该方法（注意不是类）的__get__(self, instance, cls)函数。instance就是所属类的实例，cls是类的签名。所以当用上面Pipe类直接装饰一个类方法，将会丢失类的self关键字。要实现功能，必须获取运行时的instance实例，要获取instance，就必须hack__get__函数。所以，代码变化成了下面这样：

import functools


class Pipe:
    def __init__(self, function):
        self.function = function
        functools.update_wrapper(self, function)

    def __ror__(self, other):
        return self.function(other)

    def __get__(self, instance, cls):
        return Pipe(lambda x: self.function(instance, x))

要点如下： 1. 在get中是一个经过实例包装的pipe类实例。这个实例包装了instance。 2. 这个成员必须是Pipe类型，否则，普通的Lambda方法或者函数，没有实现xor，自然也不能和|连用。

这个类装饰器，也兼容普通的函数，因为普通函数不会触发__get__方法。所以，行为与原项目一致。

本文参与腾讯云自媒体分享计划，分享自作者个人站点/博客。

原始发表：2017-05-31 ，如有侵权请联系 cloudcommunity@tencent.com 删除

面向对象编程