python - 在 setup.py 脚本中安装 nltk 数据依赖

Question

我在我的项目中使用 NLTK 和 wordnet。我在我的 PC 上使用 pip 手动安装:pip3 install nltk --user 在终端中，然后 nltk.download() 在 python shell 中下载 wordnet。

我想用 setup.py 文件自动化这些，但我不知道安装 wordnet 的好方法。

目前，我在调用 setup 之后有这段代码("nltk" 在 install_requires 列表中调用 setup):

import sys
if 'install' in sys.argv:
    import nltk
    nltk.download("wordnet")

有更好的方法吗？

Answer 1

我通过用我自己的 Install 类覆盖 cmdclass 设法在 setup.py 中安装了 NLTK 数据:

from setuptools import setup, find_packages
from setuptools.command.install import install as _install


class Install(_install):
    def run(self):
        _install.do_egg_install(self)
        import nltk
        nltk.download("popular")

setup(...
    cmdclass={'install': Install},
    ...
    install_requires=[
      'nltk',
      ],
    setup_requires=['nltk']
    ...
   )

重要的是在 run() 方法中使用方法 do_egg_install() 以确保在 import nltk 之前安装了 nltk被称为(另见此处 python setuptools install_requires is ignored when overriding cmdclass )。也不要忘记将 nltk 添加到 setup_requires。