Question

我有以下包裹分发：

mymodule/
    data/
        1.txt
    mymodule/
        __init__.py
    tests/
        test_mymodule.py
    setup.py

为了在vitualenv下安装它，我应用此命令：

pip install .

一切都安装得很好，但我的数据文件的路径破了。

>>> from mymodule import get
>>> print(get())
...
FileNotFoundError: [Errno 2] No such file or directory: '/home/alexander/Stuff/pip_example/mymodule_test/venv/lib/python3.5/site-packages/mymodule/../data/1.txt'

我做了一项研究，发现文件夹data是在virtualenv文件夹的根目录中创建的，导致错误。我应该如何改进我的代码以保持我的测试工作并获得数据文件的正确路径？

文件内容：

数据/ 1.txt的

yes

MyModule的/ __初始化__。PY

import os

src = os.path.join(
    os.path.dirname(__file__),
    '../data/1.txt'
)

def get():
    with open(src) as f:
        return f.read().strip()

测试/ test_mymodule.py

import unittest
import mymodule

class MymoduleTest(unittest.TestCase):
    def test_get(self):
        s = mymodule.get()
        self.assertEqual(s, "yes")

setup.py

from distutils.core import setup

data_files = [
    ('data', ['data/1.txt'])
]

setup(
    name='mymodule',
    version='0.0',
    packages=['mymodule'],
    data_files=data_files,
)

我是打包Python模块的新手。请帮我解决这个问题。

=============================================== ======================

我发现我需要使用sys.prefix来访问virtualenv的根目录。换句话说，如果我编辑 mymodule .__ init __。py ，这样一切都会正常工作：

import os
import sys

src = os.path.join(sys.prefix, 'data/1.txt')

def get():
    with open(src) as f:
        return f.read().strip()

但之后测试失败并出现错误：

FileNotFoundError: [Errno 2] No such file or directory: '/usr/data/1.txt'

这是因为sys.prefix是/usr而没有激活virtualenv。我想我需要一种不同的方法来改进包装。

Answer 1

安装软件包时，请检查文件是否正确分发。

sys.prefix找不到“您的”套餐。模块的__file__属性指向__init__.py文件。您可以将其用作基本路径，如：

import os
import mymodule

src = os.path.join(os.dirname(mymodule.__file__), 'data/1.txt')

def get():
    with open(src) as f:
        return f.read().strip()

如何正确分发带有数据文件的pip包？

1 个答案: