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

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' object is not callable 错误。如果我理解正确，那是因为在第一种情况下，我将属性设置为对另一个模块的引用，但在第二种情况下，它只是将属性设置为字符串。

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

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

Answer 1

最简单的方法是这样的：

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()

Answer 2

是的，你的理解是正确的。当您使用 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)