3

我目前有一个TestClass可调用的。如果任何属性等于 ,则可调用对象执行一个引发异常的函数None。将其定义为可调用的目的是,当TestClass实例被传递给另一个函数或被复制时,它将在传递之前检查所有属性是否存在,否则将引发异常。

下面显示此逻辑的行是UsesTestClass(testClass())

理想情况下,我希望能够执行相同的检查而不必“调用”类实例。例如,UsesTestClass(testClass)。是否有一种神奇的方法或其他方式来配置类以在作为参数传递之前执行函数?

class TestClass:
    def __init__(self):
        self.name = None

    def run(self):
        if self.name is None:
            raise Exception("'name' attribute is 'None'")

    def __call__(self):
        self.run()
        return self

def UsesTestClass(testClass):
    print(testClass.name)

testClass = TestClass()
testClass.name = "Hello"
UsesTestClass(testClass())
4

1 回答 1

1

如果您使用集成到 python 中的类型库,您可以这样做。

import types


class TestClass:
    def __init__(self):
        self.name = None

    def __getattribute__(self, attr):
        method = object.__getattribute__(self, attr)
        if not method:
            raise Exception("Attribute %s not implemented" % attr)
        if type(method) == types.MethodType:
            self.run()
        return method

    def run(self):
        if self.name is None:
            raise Exception("'name' attribute is 'None'")

    def __call__(self):
        self.run()
        return self


def UsesTestClass(testClass):
    print(testClass.name)


testClass = TestClass()
testClass.name = "Hello"
UsesTestClass(testClass)
于 2019-03-02T02:22:52.490 回答