组织大量的 python 导入

Question

我的代码大约 25% 依赖于以下模块：Traits、tvtk...这些模块导入起来相当繁重。在我的机器上通常需要 2 秒的时间（在其他机器上则需要更多时间）。

我的模块组织如下

mainmodule
|--submodule1
|--submodule2
   |--subsubmodule1
   |--subsubmodule2
|--submodule3
|--submodule4
   |--subsubmodule1
   |--subsubmodule2

在这些模块中，子模块 1 和子模块 2 使用 Traits。这意味着 75% 的情况下，如果我调用 import mainmodule，我将不得不等待重型模块被导入，但随后它们将不会被使用。

如何组织导入以缩短导入时间？

也许有一种方法可以做类似的事情：

import mainmodule

并且

mainmodule
|--submodule3
|--submodule4
   |--subsubmodule1
   |--subsubmodule2

只有调用：

import mainmodule.heavy

拥有一切

Answer 1

听起来您想要的是一种方法，以便导入 mainmodule 不会自动导入 submodule1 和 submodule2，这需要很长时间才能加载。

实际上，这很容易。您只能在需要它们的函数中导入 submodule1 和 submodule2。或者将这些函数移至名为 mainmodule_heavy.py 的单独模块中。

（或者您可以破解Python模块系统以延迟加载模块。但这种黑客往往会引起问题，而且对于您的情况来说这听起来没有必要。）

Answer 2

您可以将这样的代码放入函数/模块中：-

def heavy():
    global x
    global y
    import x, y

def mainmodule():
    if heavy not in globals():
        import heavy

实际上，这在同一个程序中不起作用，因为无法导入函数。另外，您需要检查全局变量中的字符串，而不是模块本身。所以，相反：-

def heavy():
    global x
    global y
    import x, y

def mainmodule():
    if 'x' not in globals() or 'y' not in globals():
        heavy()