Python中依赖变量的导入

Question

我正在创建一个需要导入外部类的类，但是要导入的类的选择取决于一个变量，这个变量在类实例化之前是不知道的。模块文件结构是这样的：

root
| -myclass.py
| -pg
| | -tabledata.py
| -pandas
| | -tabledata.py

我的类构建为对以多种格式存储的数据集（例如，Pandas、Postgresql）进行操作，我需要根据使用的后端数据格式导入TableData类。

我想做这样的事

class MyClass:
    def __init__(self,backend):
        if backend == 'pg':
            from .pg.tabledata import TableData
        elif backend == 'pandas':
            from .pandas.tabledata import TableData
    ...

但这当然不允许我在MyClass的其他地方引用TableData类。有没有合适的方法来做这件事？

Answer 1

importlib包有一个很好的import_module方法可以帮助实现这一点。查看https://docs.python.org/3/library/importlib.html#importlib.import_module。

Answer 2

也许您可以同时导入和根据某种外部配置文件或通过运行代码时添加的某种标志，您可以决定调用哪个函数。