时间戳

首页 > 解决方案 > 适用于允许您指定应调试哪些方法的类的 Python 装饰器。我该如何实施？

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

问题描述

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

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）中执行适用于类的调试器装饰器。当我调用作为参数传递给装饰器的方法之一（例如：p.goo() 或 p.oops(..)）时，它应该允许我以与调试器装饰器类似的方式调试它们做一个简单的功能。

标签： pythondecoratorpython-decorators

解决方案

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

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

推荐阅读