5

我有一个包含一系列包的应用程序。有相当多的模块引用了包层次结构中更高的其他模块。如下例所示,我可以使用相对导入来解决这个问题。但是,直接运行模块进行测试会失败并出现Attempted relative import in non-package异常。

有没有更好的方法来组织我的应用程序或导入语句,以便模块可以单独执行以进行测试并从其他模块导入?

例子

\spam
    \morespam
        child.py
    base.py
\eggs
    user.py

基础.py

class Base(object):
    def hello(self):
        print 'Um, wot?'

孩子.py

from ..base import Base       # references the parent package correctly, 
                              # but fails when this module is executed individually

class Child(Base):
    def hello(self):
        print 'Hello, world!'

if __name__ == '__main__':
    import unittest
    # ... unit test code ...

用户.py

from spam.morespam.child import Child
print Child().hello()

现有解决方案

我发现我可以将以下标头添加到需要引用层次结构中更高模块的模块顶部:

if __name__ == '__main__':
    import sys, os
    sys.path.append(os.path.abspath(os.path.join(sys.path[0], '..')))

缺点是我需要到处添加这段代码,而且它不是恒定的:'..'相对路径会根据层次结构中包的深度而变化。

其他可能的解决方案...

  1. 我可以使用我的应用程序根目录创建一个 .pth 文件。不幸的是,这必须添加到我的 Python 安装的 site-packages 文件夹中,因此它不是特定于应用程序的。
  2. 我可以暂时将我的应用程序根目录添加到 PYTHONPATH,并使我的所有模块都从根目录引用包。但是,这消除了通过简单地调用来运行脚本的简单性python.exe whatever.py
  3. 我可以编写某种模块,当导入时,它会在我的应用程序的根文件夹中查找标记文件。然后可以将这个通用模块安装到我的本地 Python 安装中,并由我的应用程序中的每个模块导入,从而保证根在 PYTHONPATH 中,无论我正在执行哪个模块。
4

2 回答 2

1

您需要__init__.py在要从中导入的文件夹中添加空文件。这样做会导致 python 将目录视为包,允许您在导入语句中使用它们:

\spam
    \morespam
        __init__.py
        child.py
    __init__.py
    base.py
\eggs
    __init__.py
    user.py

一旦你这样做并建立你的 PYTHONPATH 到这个目录的基础,你就可以进行导入:

基础.py

from morespam.child import Child
from ..eggs import user

虽然__init__.py只需要存在一个文件,但您还可以__ALL__将此文件中的变量定义为该目录中的模块列表,如果脚本尝试使用import *.

于 2011-08-19T13:59:57.377 回答
1

我用 PyCharm 解决了这个问题,它每次运行我的应用程序时都会自动将我的源根目录添加到 PYTHONPATH。据我所知,PyCharm 基本上为我做了选项#2。

于 2011-10-31T21:05:44.893 回答