Python 适用于 class 的装饰器，允许您指定应调试哪些方法。我该如何实施？

Question

我正在做一个项目，我需要实现适用于函数和 classes 的调试器装饰器。为了更好地解释，我向您展示了代码：

class debug(object):

  def __init__(self, function):
    self.function = function

  def __call__(self, *args, **kwargs):
    print()
    print(f"Calling function: {self.function}")
    print(f"args: {args}") ; print(f"kwargs: {kwargs}")
    result = self.function(*args, **kwargs)
    print(f"result is: {result}")
    return result



def _debug(function):  
  def wrapper(*args,**kwargs):
    print()
    print(f"Calling function: {function}")
    print(f"ARGS: {args}") ; print(f"KWARGS: {kwargs}")
    result = function(*args, **kwargs)
    print(f"RESULT IS: {result}")
    return result
  return wrapper


@debug  # or @_debug
def foo(x,y):
  return x+y



class cdebug(object):
  pass



def _cdebug(...):
  pass



class pallino(object):
  
  def __init__(self,x,y):
    self.x = x
    self.y = y

  def goo(self):
    return self.x+self.y  



@cdebug('oops', 'goo') # or @_cdebug('oops', 'goo')
class pinco(pallino):

  def __init__(self,x,y,z):
    super().__init__(x, y)
    self.z = z

  def oops(self,a):
    return self.x+self.y+self.z-a

if __name__ == '__main__':

  foo(10,20)

  p = pinco(2,5,7)

  p.oops(4)

  p.goo()

我在实现适用于函数的调试器装饰器时没有遇到重大问题，无论是作为 class(debug) 还是作为函数 (_debug)。但我真的不知道如何在两个版本（cdebug 和 _cdebug）中使用适用于 class 的调试器装饰器。它应该允许我，当我调用已作为参数传递给装饰器的方法之一（例如：p.goo() 或 p.oops(..)）时，以类似的方式调试它们正如调试器装饰器对简单函数所做的那样。

Answer 1

您可以构建子class，并修饰子class 的方法。这样你就不会修改原来的 class ，这可以从装饰者那里得到：

def cdebug(cls):
    class wrapper(cls):
        pass
    for name, method in inspect.getmembers(cls, inspect.isfunction):
        # this one will not decorate special methods
        if not (name.startswith('__') and name.endswith('__')):
            setattr(wrapper,name, _debug(getattr(cls, name)))
    return wrapper

Python 适用于 class 的装饰器，允许您指定应调试哪些方法。我该如何实施？

Python decorator applicable to a class that allows you to specify which methods should be debugged. How can I implement it?

python

decorator

python-decorators