python 启动时自动加载模块

Question

我希望 IPython 或 Python 解释器在启动时自动加载模块。

是否可以？

例如，当我启动 IPython 时：

$ ipython

...

>>> from __future__ import division
>>> from mymodule import *

In [1]:

在教程页面。

Answer 1

在您的主目录中有一个 .pythonstartup 并在其中加载模块，并将 PYTHONSTARTUP env 指向该文件。

在以交互模式显示第一个提示之前，将执行该文件中的 Python 命令。

• http://docs.python.org/using/cmdline.html

我用它来在 python 解释器 shell 中启用命令行补全

Answer 2

除非将 -S 选项传递给 python 二进制文件，否则会使用特殊的 site 模块。除此之外，该模块还会查找 *.pth 文件。在每一行中，*.pth 文件应包含要包含到 sys.path 中的路径或要执行的命令。该模块还导入 sitecustomize 和 usercustomize （它可以包含任意代码，如果他们碰巧引发错误，这是让你的同事发疯的好方法）（如果它们存在于某处）在sys.path中。

但问题是，当导入 site 模块时，当前目录不在 sys.path 中，也就是说很难配置您的特定脚本。

我有时会在脚本的开头添加以下行，以便脚本首先在当前目录中搜索 .pth 文件并将缺少的路径添加到 sys.path：

# search for *.pth files in the current directory
import site; site.addsitedir('')

Answer 3

检查文件 ~/.ipython/ipythonrc - 您可以列出要在启动时加载的所有模块。

Answer 4

另一种可能的解决方案是使用 参数 -i 来自 python 解释器，在执行脚本后启动交互模式。

例如，您可以使用：

• python -i your_module.py
• python -i /path/to/your/module 如果您定义了 __main__.py< /code>
• 甚至 python -i -m your.module

Answer 5

要在使用时自动延迟导入所有顶级可导入模块，请在 PYTHONSTARTUP 文件中定义：

import pkgutil
from importlib import import_module

class LazyModule:
    def __init__(self, alias, path):
        self._alias = alias
        self._path = path
        globals()[self._alias] = self

    def __getattr__(self, attr):
        module = import_module(self._path)
        globals()[self._alias] = module
        return getattr(module, attr)

# All top-level modules.
modules = [x.name for x in pkgutil.iter_modules()]

for module in modules:
    LazyModule(alias=module, path=module)

# Also include any other custom aliases.
LazyModule("mpl", "matplotlib")
LazyModule("plt", "matplotlib.pyplot")
LazyModule("pd", "pandas")
LazyModule("sns", "seaborn")
LazyModule("tf", "tensorflow")

现在您可以访问模块，而无需手动导入它们：

>>> math.sqrt(0)
0