我目前有一个可以调用的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中的类型库,你可以这样做。
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)