Python decorator适用于允许您指定应该调试哪些方法的类。我如何实现它？

Question

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

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()

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

Answer 1

您可以构建一个子类，并修饰该子类的方法。这样，您就不会修改装饰器期望的原始类：

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