如何为基本包设置配置 __main__.py、__init__.py 和 __setup__.py？

Question

背景：

我有一个这样的目录结构：

Package/
    setup.py
    src/
        __init__.py
        __main__.py 
        code.py

我希望能够以多种不同的方式运行代码。

1. pip install Package 然后python 然后from Package import *

2. python -m Package 应该做__main__.py 中的事情

3. python __main__.py 也应该在__main__.py 中执行此操作，但是这一次，我们假设您已下载源而不是pip installing。

现在我已经完成了前两个工作，但是设置很混乱：

setup.py:

setup(
    name='Package',
    packages=['Package'],
    package_dir={'Package': 'src'},
    ...
    entry_points={ 'console_scripts': ['Package = src.__main__:main' ] }

__init__.py：

from Package.code import .......

__main__.py：

from . import .......

对我来说更有意义的是在这两种情况下都写

from code import ........

但这给了我导入错误。

问题：

我拥有它的方式真的是唯一的方式吗？

最重要的是，我如何支持第三个用例？现在，python __main__.py 抛出

File "__main__.py", line 10, in <module>
    from . import code
ImportError: cannot import name 'class defined in code.py'

注意事项：

我读过

• https://chriswarrick.com/blog/2014/09/15/python-apps-the-right-way-entry_points-and-scripts/
• http://setuptools.readthedocs.io/en/latest/setuptools.html
• 这里的许多问题看起来像这个，但没有回答我上面的问题。

Answer 1

您几乎拥有所需的一切（甚至更多）！我会采用以下设置：

code.py：

foo = 1

__init__.py：

from .code import foo

在这里进行相对导入，因为在导入整个包时将使用__init__.py。请注意，我们使用 .-syntax 将导入显式标记为相对，因为这是 Python 3 所必需的（如果您使用了from __future__ import absolute_import，则在 Python 2 中）。

__main__.py：

from Package import foo

print('foo = ', foo)

这是包的主脚本，所以我们使用绝对的import 语句。通过这样做，我们假设该软件包已安装（或至少已放在路径上）；这就是处理包裹的方式！您可能认为这与您的第三个用例冲突，但实际上在处理包时没有理由不 使用pip install。这真的没什么大不了的（尤其是在使用virtualenv 时）！

如果您关心的是修改源文件并通过运行__main__.py 文件轻松观察更改，那么您只需使用-e（“可编辑”）开关安装包：pip install -e .（假设您位于目录Package)。但是，对于您当前的目录结构，这将不起作用，因为-e 开关会将egg-link 放置到包含setup.py 文件的目录中；这个目录不包含一个名为Package的包，而是src（我有a question about that）。

相反，如果您按照约定在包本身之后命名包源的根目录（即在您的示例中为 Package），那么使用 -e 安装不是问题：Python 确实找到对应目录下需要的包Package：

$ tree Package/
Package/
├── setup.py
└── Package   <-- Renamed "src" to "Package" because that's the package's name.
    ├── code.py
    ├── __init__.py
    └── __main__.py

这也可以让您省略package_dir={'Package': 'src'} 在setup.py 中的额外定义。

关于setup.py的说明：对于您指定的三个用例，无需定义入口点。也就是说，您可以跳过entry_points={ 'console_scripts': ['Package = src.__main__:main' ] } 行。通过发送__main__.py 模块python -m Package 将很容易执行此模块中的代码。您还可以添加一个额外的 if 子句：

def main():
    print('foo = ', foo)

if __name__ == '__main__':
    main()

另一方面，入口点允许您直接从 CLI 执行 __main__.main 中的代码；即运行$ Package会执行相应的代码。

回顾

底线是我在处理包裹时总是使用pip install。为什么不呢，特别是如果您已经创建了一个setup.py 文件？如果要“实时”应用对包的更改，则可以使用-e 开关进行安装（这可能需要重命名src 文件夹，见上文）。因此，您的第三个用例将显示为“下载源代码和 pip install (-e) Package（在 virtualenv 中）；然后您可以运行 python __main__.py”。

---

编辑

在没有 pip install 的情况下运行 __main__.py

如果您不想通过 pip 安装软件包但仍然能够运行 __main__.py 脚本，我仍然会使用上述设置。然后我们需要确保from Package import ... 语句仍然成功，这可以通过扩展导入路径来实现（注意这需要将src 目录重命名为包的名称！）。

修改PYTHONPATH

对于 Linux bash，您可以按如下方式设置 Pythonpath：

export PYTHONPATH=$PYTHONPATH:/path/to/Package

或者如果你和__main__.py在同一个目录下：

export PYTHONPATH=$PYTHONPATH:`cd ..; pwd`

当然，不同的操作系统有不同的方法。

扩展__main__.py中的路径

您（或者更确切地说是您的同事）可以将以下行添加到脚本顶部（在 from Package import ... 语句之前）：

import sys
sys.path.append('/path/to/Package')

扩展sitecustomize.py中的路径

您可以在 Python 安装的 lib/python3.5/site-packages/ 目录中放置一个名为 sitecustomize.py 的模块，该目录包含以下几行：

import sys
sys.path.append('/path/to/Package')

创建一个单独的顶级main.py 脚本

所以你会有以下布局：

$ tree Package/
Package/
├── main.py   <-- Add this file.
├── setup.py
└── src
    ├── code.py
    ├── __init__.py
    └── __main__.py

main.py 包含在哪里

import src.__main__

现在__main__.py 被视为src 包的一部分，并且相对导入将起作用。 您现在可以运行python main.py，而不是运行python src/__main__.py。

Answer 2

我经常使用这种设置，因为它更适合 python setup.py develop

Package_root/
    setup.py
    src/
        Package/
            __init__.py
            __main__.py 
            code.py

这可能不是（还）您期望的详细答案，但我认为这三个用例值得一试。

setup( ...
    package_dir = {'': 'src'},
    entry_points = {'console_scripts': ['Package = Package.__main__:main'],},
    packages = find_packages(exclude=["Package.egg_info",]),
...)

Answer 3

from code import ......... 失败，因为您的系统上没有安装名为 code 的 Python 包。您的系统上有一个名为 code 的 Python 模块，但在您的导入语句中，您没有指定可以在其中找到您的 code 模块的包。

src/ 中的 __init__.py 文件的用途告诉 Python，src/ 目录应该被视为 Python 包，其内容作为包中的模块。由于code.py 与您的__init__.py 文件一起位于src/ 中，因此您的code 模块位于您的src 包中。

现在您知道您的 code 模块可以在哪个包中找到，您可以从其中导入内容：

from src.code import .........

另外，作为旁注：__init__.py 只是通过出现在您的src/ 目录中来完成它的工作，因此它甚至不需要包含任何代码。因此，将__init__.py 文件留空通常是个好主意。