如果有一个装饰器的参数，如何从装饰器的参数中删除 self 参数

Question

假设我有一个简单的装饰器，以及一个类方法和一个函数，我都用这个装饰器装饰：

import functools

def decorator(func):
    @functools.wraps(func)
    def call(*args):
        print(args)
        func(*args)
    return call

class cls:
    @decorator
    def meth(self, a):
        pass

@decorator
def func(c):
    pass

cls().meth("a")
func("c")

我得到以下输出：

(<__main__.cls object at 0x7f4665c50130>, 'a')
('c',)

但是我想在方法上使用装饰器时去掉self参数，这样输出就变成了：

('a',)
('c',)

但是，如果我只是添加args.pop(0)，我将删除第一个参数，即使它不是self。如果它是self，我怎样才能只删除第一个参数？

注意：我使用inspect 阅读了一些带有长代码的解决方案 - 但在这种伟大且易于使用的编程语言中必须有更短、更简单的方法？

编辑：使用@staticmethod 对我来说不是一个选项，因为我需要方法本身中的self 参数。我只是不希望它被打印。

Answer 1

实现这一点的最简单方法是明确告诉您的装饰器它正在装饰一个类方法。您的装饰器将采用一个关键字参数来表示这一点。

def decorator(is_classmethod=False):
    def wrapper(func):
        @functools.wraps(func)
        def call(*args):
            if is_classmethod:
                print(args[1:])  # omit args[0]
            else:
                print(args)
            return func(*args)      # pass all the args to the function
        return call
    return wrapper

class cls:
    @decorator(is_classmethod=True)
    def meth(self, a):
        pass

@decorator
def func(c):
    pass

cls().meth("a")
func("c")

由于用户已经知道他们在装饰什么，您可以避免装饰器中复杂的自省代码，并让用户可以灵活地处理或不处理第一个参数。

请看看我对这个 SO 问题的回答：Decorators with parameters? 以更简洁（恕我直言）的方式来编写带参数的装饰器。那篇文章中有一些非常好的答案。

Answer 2

您可以通过一些 python 魔法来过滤args 以获取输出。例如：

import functools

def decorator(func):
    @functools.wraps(func)
    def call(*args):
        print(list(filter(lambda x: not hasattr(x, '__dict__'), args)))
        func(*args)
    return call

class cls:
    @decorator
    def meth(self, a):
        pass

@decorator
def func(c):
    pass

cls().meth("a")  # ['a']
func("c")  # ['c']