如何从Python脚本内部终止Bash的for循环？

问题背景 #

我有一个Bash的for循环，会迭代运行Python脚本。当Python脚本内的条件满足时，需要终止当前Python脚本，并直接停止Bash的循环。

原脚本示例 #

Bash脚本：

for i in {1..15}; do
    echo "Pre Outside ${i}"
    python minimal_test.py $i
    echo "Post Outside ${i}"
done

Python脚本：

import sys
import subprocess
k = int(sys.argv[1])
end_bash = False

if k == 7:
    end_bash = True
print("Inside " + str(k))
if end_bash:
    bashCommand = "break"
    process = subprocess.Popen(bashCommand, stdout=subprocess.PIPE)

预期输出（i=6和7时） #

Pre Outside 6
Inside 6
Post Outside 6    
Pre Outside 7
Inside 7 # 程序需在此处终止

遇到的错误 #

File "/software/software/Python/3.9.5-GCCcore-10.3.0/lib/python3.9/subprocess.py", line 1821, in _execute_child
raise child_exception_type(errno_num, err_msg, err_filename)
FileNotFoundError: [Errno 2] No such file or directory: 'break'

解决方案 #

核心原因 #

break是Bash的内置命令，不是独立的可执行程序，因此subprocess.Popen无法找到它。此外，Python启动的子进程与父Bash进程是独立的，子进程的操作无法直接控制父进程的循环。正确的做法是通过进程间通信机制让Python通知Bash终止循环，最常用的是退出码或输出标记。

方法1：通过Python返回退出码（推荐） #

让Python在满足条件时返回非0的退出码，Bash捕获该退出码后执行break终止循环。

修改后的Bash脚本

for i in {1..15}; do
    echo "Pre Outside ${i}"
    python minimal_test.py $i
    # 检查上一个命令的退出码，非0则终止循环
    if [ $? -ne 0 ]; then
        break
    fi
    echo "Post Outside ${i}"
done

修改后的Python脚本

import sys
k = int(sys.argv[1])

print("Inside " + str(k))
# 满足条件时返回非0退出码
if k == 7:
    sys.exit(1)

方法2：通过输出特定标记判断 #

让Python在满足条件时打印特定标记字符串，Bash捕获输出后判断是否终止循环。这种方法适合需要同时传递其他信息的场景，但不如退出码可靠。

修改后的Bash脚本

for i in {1..15}; do
    echo "Pre Outside ${i}"
    # 捕获Python脚本的所有输出
    output=$(python minimal_test.py $i)
    echo "$output"
    # 检查输出是否包含终止标记
    if echo "$output" | grep -q "TERMINATE_BASH"; then
        break
    fi
    echo "Post Outside ${i}"
done

修改后的Python脚本

import sys
k = int(sys.argv[1])

print("Inside " + str(k))
if k == 7:
    # 打印终止标记
    print("TERMINATE_BASH")
    sys.exit()

内容的提问来源于stack exchange，提问作者je2018