作为脚本的Python模块[副本]

Question

program.py
modules/__init__.py
modules/mod1.py
modules/mod2.py
modules/utils.py

program.py包括modules/目录中的所有模块，而modules/中的模块通过从.utils import*包括modules/utils.py

在运行program.py时，上面的操作很好。但是，mod1.py和mod2.py也作为它们自己的脚本工作，但是运行Python mod1.py会出现错误

from .utils import *
ModuleNotFoundError: No module named '__main__.utils'; '__main__' is not a package

从导入中删除.可以使各个脚本正常工作，但主program.py脚本找不到utils。

我如何使导入对主程序和单个模块都起作用？

Answer 1

您可能会发现有一个单一的点访问您的程序更好。如果这是一个命令行实用程序，您可以使用命令行解析器（如内置argparse）或外部库（如click）为所有功能创建一个包装器。

使用click，您可以添加一个名为CLI.PY的文件，并将其作为您的单一访问点，然后使用标志来访问程序的不同功能。并使用setuptools,您可以为程序创建一个干净的入口点。

因此您的文件结构变为：

|-__init__.py
|-program.py
|-cli.py
|-setuptools.py
|-modules
  |-mod1.py
  |-mod2.py
  |-utils.py

cli.py：

import click
import program

@click.group()
def cli():
    pass

@cli.command()
def program():
    program.run_the_program()

@cli.command()
def mod1():
    program.mod1.run_the_program()

@cli.command()
def mod2():
    program.mod2.run_the_program()

if __name__ == '__main__':
    cli()

和您的设置.py：

from setuptools import setup

setup(
    name="your-program",
    version="1.0",
    py_modules=["program"],
    include_package_data=True,
    install_requires=["click"],
    entry_points="""
        [console_scripts]
        program=program:cli
    """,
)

在您的目录中创建virtualenv，激活它，运行setuptoolspip install--editable，然后您应该能够使用程序程序、程序mod1等从命令行运行您的程序或mod1和mod2子程序。