我有一个包含一系列包的应用程序。有相当多的模块引用了包层次结构中更高的其他模块。如下例所示,我可以使用相对导入来解决这个问题。但是,直接运行模块进行测试会失败并出现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], '..')))
缺点是我需要到处添加这段代码,而且它不是恒定的:'..'
相对路径会根据层次结构中包的深度而变化。
其他可能的解决方案...
- 我可以使用我的应用程序根目录创建一个 .pth 文件。不幸的是,这必须添加到我的 Python 安装的 site-packages 文件夹中,因此它不是特定于应用程序的。
- 我可以暂时将我的应用程序根目录添加到 PYTHONPATH,并使我的所有模块都从根目录引用包。但是,这消除了通过简单地调用来运行脚本的简单性
python.exe whatever.py
。 - 我可以编写某种模块,当导入时,它会在我的应用程序的根文件夹中查找标记文件。然后可以将这个通用模块安装到我的本地 Python 安装中,并由我的应用程序中的每个模块导入,从而保证根在 PYTHONPATH 中,无论我正在执行哪个模块。