使用 Bash 函数在 Python 脚本运行前自动格式化代码-Python教程-PHP中文网

使用 Bash 函数在 Python 脚本运行前自动格式化代码

花韻仙語

发布： 2025-08-22 17:24:17

原创

161人浏览过

使用 bash 函数在 python 脚本运行前自动格式化代码

本文介绍如何通过编写一个简单的 Bash 函数，实现在每次运行 Python 脚本之前自动使用 Black 进行代码格式化。这种方法能够帮助开发者在脚本执行前及时发现并修正代码风格问题，从而提高代码质量，减少潜在的错误。该方案轻量级，易于配置，适用于快速本地测试和开发环境。

为了确保每次运行 Python 脚本前都执行 Black 格式化，我们可以创建一个 Bash 函数。这个函数首先会运行 Black，如果 Black 成功执行，没有发现错误，则继续执行 Python 脚本。

Bash 函数定义：

runpy() {
    if ! black "$1"; then
        echo "Black encountered an error. Aborting."
        return 1
    fi
    python3 "$@"
}

登录后复制

函数解释：

立即学习“Python免费学习笔记（深入）”；

runpy()：定义了一个名为 runpy 的 Bash 函数。
if ! black "$1"; then：首先，尝试使用 Black 格式化脚本。$1 代表传递给函数的第一个参数，也就是 Python 脚本的文件名。! 符号表示如果 Black 命令执行失败（返回非零退出码），则执行 then 后面的语句。
echo "Black encountered an error. Aborting."：如果 Black 遇到错误，则输出错误信息。
return 1：如果 Black 遇到错误，则函数返回 1，表示执行失败。
python3 "$@"：如果 Black 成功执行，则使用 python3 命令执行脚本。$@ 代表传递给函数的所有参数，包括脚本文件名和任何其他参数。

使用方法：

将上述代码添加到你的 Bash 配置文件（例如 .bashrc 或 .zshrc）中，然后重新加载配置文件 (例如 source ~/.bashrc)。之后，你就可以像下面这样使用这个函数：

runpy my_python_file.py argument1 --option2

登录后复制

示例：

假设你有一个名为 my_script.py 的 Python 脚本，并且你想传递一些参数给它，你可以这样运行：

runpy my_script.py --verbose --output=result.txt

登录后复制

这个命令会先使用 Black 格式化 my_script.py，如果 Black 成功执行，则使用 python3 运行 my_script.py，并将 --verbose 和 --output=result.txt 作为参数传递给脚本。

Black 的安装： 确保你已经安装了 Black。你可以使用 pip 安装：pip install black
Python 版本： 确保你使用的是正确的 Python 版本。在函数中使用 python3 确保使用 Python 3。如果你的系统默认 Python 版本是 Python 2，则需要显式指定 python3。
错误处理： 函数中的错误处理机制会在 Black 遇到错误时停止脚本的执行。这可以防止在代码风格不符合规范的情况下运行脚本。
配置文件： 确保将函数添加到你的 Bash 配置文件中，以便每次启动终端时都可以使用它。