Snakemake运行后仅将conda环境导出为yaml文件的实现方案问询

你要的功能可以直接通过Snakemake内置的workflow对象实现，不需要手动匹配.snakemake/conda/下的md5目录名，具体方案如下：

1. 核心实现代码 #

直接在你的Snakefile任意位置添加如下onsuccess配置块即可：

onsuccess:
    import os
    import subprocess

    # 自定义导出的环境文件存储目录
    EXPORT_DIR = "used_conda_envs"
    os.makedirs(EXPORT_DIR, exist_ok=True)

    # 遍历所有工作流用到的conda环境
    # workflow.conda_envs的键是你在rule中定义的conda环境yaml路径，值是对应md5命名的本地环境路径
    for env_def_path, env_installed_path in workflow.conda_envs.items():
        # 从原环境yaml路径提取自定义的环境名，也可以自己修改命名规则
        env_custom_name = os.path.splitext(os.path.basename(env_def_path))[0]
        export_file = os.path.join(EXPORT_DIR, f"{env_custom_name}.yaml")

        # 调用conda命令导出环境，--no-builds参数可移除构建信息缩小文件体积，不需要可删除
        subprocess.run(
            [
                "conda", "env", "export",
                "--prefix", env_installed_path,
                "--file", export_file,
                "--no-builds"
            ],
            check=True,
            capture_output=True
        )
    print(f"所有运行使用的conda环境已导出至 {EXPORT_DIR} 目录")

2. 注意事项 #

• 若多个rule共用同一份conda环境定义，不会重复导出，workflow.conda_envs会自动去重
• 如果你使用mamba作为包管理器，仅需把命令中的conda替换为mamba即可，逻辑完全一致
• 如果需要导出带完整构建信息的环境文件，删除命令中的--no-builds参数即可
• 执行snakemake命令的环境需要能正常调用conda/mamba可执行文件

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