具有可装饰方法或函数的参数的基于Python类的装饰器

Question

我已经看到许多Python装饰器的示例，它们是：

• 函数样式修饰符（包装函数）
• 类风格装饰（实施__init__，__get__和__call__）
• 不带参数的装饰器
• 带参数的装饰器
• “方法友好”的装饰器（即可以装饰类中的方法）
• “功能友好”的装饰器（可以装饰普通功能
• 可以装饰方法和功能的装饰器

但是我从未见过一个可以完成上述所有操作的示例，而且我无法从各种答案到特定问题，这个或这个上看到过，如何结合以上所有内容。

我想要的是一个 _ 基于类的 装饰器，它可以装饰 _ 一个方法或一个函数 ，并且 _ 至少需要一个附加参数_。即，以便以下工作：

class MyDecorator(object):
    def __init__(self, fn, argument):
        self.fn = fn
        self.arg = argument

    def __get__(self, ....):
        # voodoo magic for handling distinction between method and function here

    def __call__(self, *args, *kwargs):
        print "In my decorator before call, with arg %s" % self.arg
        self.fn(*args, **kwargs)
        print "In my decorator after call, with arg %s" % self.arg


class Foo(object):
    @MyDecorator("foo baby!")
    def bar(self):
        print "in bar!"


@MyDecorator("some other func!")
def some_other_function():
    print "in some other function!"

some_other_function()
Foo().bar()

我希望看到：

In my decorator before call, with arg some other func!
in some other function!
In my decorator after call, with arg some other func!
In my decorator before call, with arg foo baby!
in bar!
In my decorator after call, with arg foo baby!

编辑：如果有关系，我正在使用Python 2.7。

Answer 1

您不需要弄乱描述符。在__call__()方法内部创建包装函数并返回它就足够了。根据上下文，标准Python函数始终可以充当方法或函数：

class MyDecorator(object):
    def __init__(self, argument):
        self.arg = argument

    def __call__(self, fn):
        @functools.wraps(fn)
        def decorated(*args, **kwargs):
            print "In my decorator before call, with arg %s" % self.arg
            fn(*args, **kwargs)
            print "In my decorator after call, with arg %s" % self.arg
        return decorated

关于使用这种装饰器时发生的情况的一些解释：

@MyDecorator("some other func!")
def some_other_function():
    print "in some other function!"

第一行创建的实例MyDecorator并将其"some other func!"作为参数传递给__init__()。我们称这个实例my_decorator。接下来，创建未修饰的函数对象（我们称其为“对象”）bare_func并将其传递给装饰器实例，然后my_decorator(bare_func)执行该对象。这将调用MyDecorator.__call__()，这将创建并返回包装器函数。最后，将此包装函数分配给名称some_other_function。