如何在Python中限制Docker容器运行时长并处理异常？

要给Docker容器设置5分钟（300秒）的运行时长限制，同时处理容器运行失败的异常，可以通过Docker Python SDK的wait()方法配合超时参数实现，同时完善异常处理逻辑避免未定义变量等问题。

完整代码 #

import docker
from docker.errors import APIError, NotFound

client = docker.from_env()
container = None

try:
    # 启动后台运行的Python容器
    container = client.containers.run("python", detach=True, name="python-c")
    # 等待容器运行结束，设置超时时间为300秒（5分钟）
    container.wait(timeout=300)
except APIError as err:
    print(f"容器运行出错: {str(err)}")
except TimeoutError:
    print("容器运行超时，强制终止")
    if container:
        container.stop()
finally:
    # 无论成功或失败，尝试清理容器
    try:
        if container:
            container.remove()
        else:
            # 处理容器已存在但未被成功赋值的情况
            existing_cont = client.containers.get("python-c")
            existing_cont.remove()
    except NotFound:
        # 容器已不存在，无需操作
        pass

关键说明 #

• 超时控制：使用container.wait(timeout=300)等待容器运行，超时后会抛出TimeoutError，此时捕获异常并调用container.stop()强制终止容器。
• 异常处理：捕获APIError处理容器启动失败的情况（比如容器名冲突），同时捕获NotFound避免清理时因容器不存在报错。
• 资源清理：通过finally块确保无论容器运行状态如何，都能尝试移除容器，避免残留无效容器。
• 变量初始化：提前将container设为None，避免异常场景下变量未定义导致后续代码报错。

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