如何在运行的Python脚本中终止其所在的Docker容器？

在Docker容器中运行着一个Python脚本，希望在特定时刻通过该脚本终止运行它的父Docker容器。该Docker容器由AWS任务启动，相关Dockerfile和Example.py代码如下：

Dockerfile #

FROM python:3.9.13

# Variable arguments
ENV AM_I_IN_A_DOCKER_CONTAINER True
WORKDIR /code

#Install requirements
COPY ./requirements.txt /code/requirements.txt
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt

#Copy code into the container
COPY ./example/ /code/example/


#Start the app
CMD ["python", "example/example.py"]

Example.py #

import threading
import time 

def start_worker():
    #do some stuff
    time.sleep(10)
    print("WE ARE EXITING")
    print("EXIT")
    exit()
        
ce = threading.Thread(target=start_worker)
ce.start()

要实现Python脚本终止父Docker容器的需求，核心是让容器的主进程退出（Docker容器会随主进程终止而停止），或者直接调用系统命令停止容器，以下是两种可行方案：

方案一：让Python主进程退出 #

当前代码中，exit()仅终止子线程，主进程仍会持续运行，容器不会停止。需要修改代码让主进程也退出：

修改后的Example.py #

import threading
import time 
import os

def start_worker():
    #do some stuff
    time.sleep(10)
    print("WE ARE EXITING")
    print("EXIT")
    # 直接终止当前进程（容器主进程），彻底停止容器
    os._exit(0)
        
ce = threading.Thread(target=start_worker)
ce.start()
# 主线程等待子线程执行完毕，避免提前退出
ce.join()

说明：os._exit(0)会直接终止整个进程，比sys.exit()更彻底，适合需要立即停止容器的场景；如果需要优雅退出，也可以用sys.exit()，但需确保主线程未捕获SystemExit异常。

方案二：直接调用Docker命令停止容器 #

如果需要更直接地控制容器停止，可以在Python脚本中执行Docker命令，但需要容器具备Docker客户端权限（需挂载Docker socket）：

1. 更新Dockerfile安装Docker客户端 #

FROM python:3.9.13

ENV AM_I_IN_A_DOCKER_CONTAINER True
WORKDIR /code

# 安装Docker客户端
RUN apt-get update && apt-get install -y docker.io --no-install-recommends && rm -rf /var/lib/apt/lists/*

COPY ./requirements.txt /code/requirements.txt
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt

COPY ./example/ /code/example/

CMD ["python", "example/example.py"]

2. AWS ECS任务配置挂载Docker socket #

在ECS任务定义中添加卷挂载：

• 类型选择bind mount
• 源路径填写/var/run/docker.sock
• 容器内路径填写/var/run/docker.sock

3. 修改Example.py执行停止命令 #

import threading
import time 
import subprocess
import os

def get_container_id():
    # 从cgroup文件读取当前容器ID
    with open('/proc/self/cgroup', 'r') as f:
        for line in f:
            if 'docker' in line:
                return line.strip().split('/')[-1]
    # 若容器启动时传入了CONTAINER_ID环境变量，直接读取
    return os.environ.get('CONTAINER_ID')

def start_worker():
    time.sleep(10)
    print("WE ARE EXITING")
    container_id = get_container_id()
    if container_id:
        # 执行Docker停止命令
        subprocess.run(['docker', 'stop', container_id], check=True)
    else:
        print("无法获取容器ID，终止进程停止容器")
        os._exit(0)
        
ce = threading.Thread(target=start_worker)
ce.start()
ce.join()

AWS ECS任务补充说明 #

如果是AWS ECS任务，当容器主进程退出后，ECS会根据任务的desired count配置决定是否重启容器。若希望任务彻底终止，需确保任务的desired count设置为0，或者在脚本中通过AWS SDK调用API更新任务的期望计数。

内容的提问来源于stack exchange，提问作者Learning from masters