如果传递类实例,如何执行函数?

问题描述 投票:3回答:1

我目前有一个可以调用的TestClass。如果任何属性等于None,则callable执行一个引发异常的函数。定义它是可调用的目的是当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())
python python-3.x callable callable-object
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)
© www.soinside.com 2019 - 2024. All rights reserved.