3

我有一个库中的类，可以说MyClass是一个方法my_method。

from MyLibrary import MyClass
obj = MyClass()
result = obj.my_method('arg')

我无法理解原始实现，但我想装饰my_method，特别是我想检查返回值是否为not None，以防引发异常，如：

result = obj.my_method('arg')
if result is None: raise ValueError()

这很无聊，因为我需要做很多次。特别是我不想在调用时更改代码obj.my_method('arg')。我在尝试着：

def safe_method(self, *args):
  result = self.my_method(*args)
  if result is None: raise ValueError()
  return result

MyClass.my_method = safe_method

这不起作用：

RuntimeError: maximum recursion depth exceeded while calling a Python object

我明白为什么，但我无法解决问题

4

2 回答 2

2

你可以这样装饰它：

def safe_decorator(func):
    def wrapper(self, *args):
        result = func(self, *args)
        if result is None:
            raise ValueError()
        return result
    return wrapper

MyClass.my_method = safe_decorator(MyClass.my_method)

然后，您可以从任何类定义中将这个装饰器重用于您想要的任何方法。这就是@符号或多或少的作用:)

希望这可以帮助！

于 2013-11-06T18:45:34.927 回答
1 
def safe_method(self, *args):
  result = self.my_method(*args)
  if result is None: raise ValueError()
  return result

MyClass.my_method = safe_method

上面发生的事情是您将 MyClass.my_method 替换为另一个调用 my_method 的方法。因此，您会遇到递归错误。

尝试这个：

MyClass._my_method = MyClass.my_method

def safe_method(self, *args):
    result = self._my_method(*args)
    if result is None:
        raise ValueError()
    return result

MyClass.my_method = safe_method
于 2013-11-06T18:29:34.190 回答