如何使用Python SDK实现docker compose自动化并生成compose配置文件

完全可以通过Python代码实现docker-compose配置的自动化生成，不需要手动在不同项目中重复编写相似配置，以下是具体实现方案：

基础实现：用Python YAML库直接生成 #

这是最灵活、学习成本最低的方案，核心逻辑是把compose配置作为Python字典处理，差异化字段动态赋值后导出为YAML文件即可，示例代码如下：

import yaml
import subprocess

# 可复用的通用服务配置模板
common_service_base = {
    "restart": "unless-stopped",
    "logging": {
        "driver": "json-file",
        "options": {
            "max-size": "50m",
            "max-file": "3"
        }
    }
}

def generate_compose(project_name: str, web_port: int, enable_mysql: bool = False):
    compose_data = {
        "version": "3.8",
        "services": {
            "web": {
                **common_service_base,
                "image": "nginx:alpine",
                "container_name": f"{project_name}_web",
                "ports": [f"{web_port}:80"],
                "volumes": [f"./{project_name}/html:/usr/share/nginx/html"]
            }
        }
    }
    # 根据参数动态添加服务
    if enable_mysql:
        compose_data["services"]["mysql"] = {
            **common_service_base,
            "image": "mysql:8.0",
            "container_name": f"{project_name}_mysql",
            "environment": {
                "MYSQL_ROOT_PASSWORD": "123456"
            },
            "volumes": [f"./{project_name}/mysql_data:/var/lib/mysql"]
        }
    # 写入配置文件
    with open("docker-compose.yml", "w", encoding="utf-8") as f:
        yaml.safe_dump(compose_data, f, sort_keys=False, indent=2)
    # 自动校验配置合法性
    subprocess.run(["docker-compose", "config", "-q"], check=True)
    print("配置文件生成完成，格式合法")

# 调用示例：生成名为test_project的项目配置，web端口8080，开启mysql服务
if __name__ == "__main__":
    generate_compose("test_project", 8080, enable_mysql=True)

生成的YAML文件完全符合docker-compose语法规范，可以直接使用。

进阶优化建议 #

• 把通用服务模板封装成独立的Python模块，所有项目直接导入使用，仅传入镜像版本、端口、挂载路径、环境变量等差异化参数即可
• 可以结合配置中心、环境变量实现多环境配置一键生成，不用单独维护开发、测试、生产三套compose文件
• 如果配置结构固定、仅参数变化，可以结合jinja2模板引擎实现，把compose模板存为独立文件，Python代码仅负责填充变量

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