Question

我需要使用setattr从配置文件中读取，并将类的属性设置为另一个模块中的函数。粗略地说，这是问题所在。

这有效：

class MyClass:
    def __init__(self):
        self.number_cruncher = anothermodule.some_function

# method works like this, of course;
foo = MyClass()
foo.number_cruncher(inputs)

好的，这很简单，但如果我想从配置文件中读取some_function的名称呢？读取文件并使用setattr也很简单：

def read_ini(target):
    # file reading/parsing stuff here, so we effectively do:
    setattr(target, 'number_cruncher', 'anothermodule.some_function')

bar = MyClass()
read_ini(bar)
bar.number_cruncher(inputs)

这给了我一个'str'对象不可调用错误。如果我理解正确，那是因为在第一种情况下我将属性设置为对另一个模块的引用，但在第二种情况下，它只是将属性设置为字符串。

问题：1）我对这个错误的理解是否正确？ 2）如何使用setattr将属性设置为对函数的引用而不仅仅是字符串属性？

我已经查看了其他问题并发现了类似的问题，但似乎没有任何内容可以解决这个问题。

Answer 1

是的，您的理解是正确的。使用setattr时，您将属性设置为当前示例中的字符串。因此，当您再调用bar.number_cruncher(inputs)时，它会正确地告诉您字符串对象不可调用。如果你想让bar.number_chruncher成为函数anothermodule.some_function的可调用等价物，那么你可以导入它并将函数（而不是字符串）设置为属性。

def read_ini(target):
    # file reading/parsing stuff here, so we effectively do:
    __import__("anothermodule.some_function") #Allows you to import based on a string parsed from your configuration file.
    setattr(target, 'number_cruncher', anothermodule.some_function)

Answer 2

最琐碎的方法是：

module_name, function_name = "module.func".rsplit('.', 1)
f = getattr(sys.modules[module_name], function_name)
f()  # f is now callable, analogous to anothermodule.some_function in your code snippets.

显然，很多潜在的问题都没有得到解决。首先，它假定模块已经导入。要解决此问题，您可以参考Dynamic module import in Python并使用__import__的返回值。不用担心，CPython会在内部对其进行优化，同一模块不会被多次解释。

稍微好一点的版本是：

module_name, function_name = "module.func".rsplit('.', 1)
f = getattr(__import__(module_name), function_name)
f()

使用Python setattr设置对方法或函数的属性引用

2 个答案: