Python 스크립트를 명령줄 프로그램으로 변환

现在创建一个 ​​Git​​​ 标记（例如 ​​v0.2​​​），此工具会将其识别为可安装版本。在提交更改之前，浏览一下自动生成的 ​​setup.cfg​​​ 并根据需要编辑它。对于此示例，你可以修改 ​​LICENSE​​ 和项目描述，将这些更改添加到 Git 的暂存区，我必须禁用预提交钩子，然后提交它们。否则，我会遇到错误，因为 Python 风格检查器 flake8 会抱怨糟糕的格式。

$ PRE_COMMIT_ALLOW_NO_CONFIG=1 git commit

如果这个脚本有一个入口点，用户可以从命令行调用，那就更好了。现在，你只能通过找 ​​.py​​​ 文件并手动执行它来运行。幸运的是，Python 的打包基础设施有一个很好的“罐装”方式，可以轻松地进行配置更改。将以下内容添加到 ​​setup.cfg​​​ 的 ​​options.entry_points​​ 部分：

console_scripts =

roto = rotoscope.rotoscope:rotoscope

这个更改会创建一个名为 ​​roto​​​ 的 shell 命令，你可以使用它来调用 rotoscope 脚本，使用 ​​pip​​​ 安装 rotoscope 后，可以使用 ​​roto​​ 命令。

就是这样，你可以从 Pyscaffold 免费获得所有打包、测试和文档设置。你还获得了一个预提交钩子来保证（大部分情况下）你按照设定规则提交。

CLI 工具化

现在，一些值会硬编码到脚本中，它们作为命令 ​​参数​​​ 会更方便。例如，将 ​​INCOMING​​ 常量作为命令行参数会更好。

首先，导入 ​​click​​​ 库，使用 Click 提供的命令装饰器对 ​​rotoscope()​​​ 方法进行装饰，并添加一个 Click 传递给 ​​rotoscope​​ 函数的参数。Click 提供了一组验证器，因此要向参数添加一个路径验证器。Click 还方便地使用函数的内嵌字符串作为命令行文档的一部分。所以你最终会得到以下方法签名：

@click.command()

@click.argument('incoming', type=click.Path(exists=True))

def rotoscope(incoming):

"""

Rotoscope 0.4 - Bleep, blooop.

Simple sample that move files.

"""

主函数会调用 ​​rotoscope()​​，它现在是一个 Click 命令，不需要传递任何参数。

选项也可以使用 ​​环境变量​​​ 自动填充。例如，将 ​​ARCHIVE​​ 常量改为一个选项：

@click.option('archive', '--archive', default='/Users/mark/archive', envvar='ROTO_ARCHIVE', type=click.Path())

使用相同的路径验证器。这一次，让 Click 填充环境变量，如果环境变量没有提供任何内容，则默认为旧常量的值。

Click 可以做更多的事情，它有彩色的控制台输出、提示和子命令，可以让你构建复杂的 CLI 工具。浏览 Click 文档会发现它的更多功能。

现在添加一些测试。

测试

Click 对使用 CLI 运行器 ​​运行端到端测试​​​ 提供了一些建议。你可以用它来实现一个完整的测试（在 ​​示例项目​​​ 中，测试在 ​​tests​​ 文件夹中。）

测试位于测试类的一个方法中。大多数约定与我在其他 Python 项目中使用的非常接近，但有一些细节，因为 rotoscope 使用 ​​click​​​。在 ​​test​​​ 方法中，我创建了一个 ​​CliRunner​​​。测试使用它在一个隔离的文件系统中运行此命令。然后测试在隔离的文件系统中创建 ​​incoming​​​ 和 ​​archive​​​ 目录和一个虚拟的 ​​incoming/test.txt​​​ 文件，然后它调用 CliRunner，就像你调用命令行应用程序一样。运行完成后，测试会检查隔离的文件系统，并验证 ​​incoming​​​ 为空，并且 ​​archive​​ 包含两个文件（最新链接和存档文件）。

from os import listdir, mkdir

from click.testing import CliRunner

from rotoscope.rotoscope import rotoscope


class TestRotoscope:

def test_roto_good(self, tmp_path):

runner = CliRunner()


with runner.isolated_filesystem(temp_dir=tmp_path) as td:

mkdir("incoming")

mkdir("archive")

with open("incoming/test.txt", "w") as f:

f.write("hello")


result = runner.invoke(rotoscope, ["incoming", "--archive", "archive"])

assert result.exit_code == 0


print(td)

incoming_f = listdir("incoming")

archive_f = listdir("archive")

assert len(incoming_f) == 0

assert len(archive_f) == 2

要在控制台上执行这些测试，在项目的根目录中运行 ​​tox​​。

在执行测试期间，我在代码中发现了一个错误。当我进行 Click 转换时，​​rotoscope​​ 只是取消了最新文件的链接，无论它是否存在。测试从一个新的文件系统（不是我的主文件夹）开始，很快就失败了。我可以通过在一个很好的隔离和自动化测试环境中运行来防止这种错误。这将避免很多“它在我的机器上正常工作”的问题。

搭建骨架脚本和模块

本文到此结束，我们可以使用 ​​scaffold​​​ 和 ​​click​​ 完成一些高级操作。有很多方法可以升级一个普通的 Python 脚本，甚至可以将你的简单实用程序变成成熟的 CLI 工具。

위 내용은 Python 스크립트를 명령줄 프로그램으로 변환의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!