用 Python 实现 Docker 镜像批量推送(带进度条)

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import docker
import sys
from tqdm import tqdm
# ========== 配置 ==========
docker_username = "rockmelodies"   # 你的 docker hub 用户名
# =========================
client = docker.from_env()
def get_unique_images_from_all_containers():
    """获取所有容器（包括停止的）使用的镜像名（唯一）"""
    containers = client.containers.list(all=true)
    images = set()
    for c in containers:
        # 镜像名格式：repository:tag
        image_full = c.image.tags[0] if c.image.tags else f"{c.image.id[:12]}"
        images.add(image_full)
    return images
def short_image_name(full_name):
    """从 vulhub/nginx:1.11.13 提取 nginx:1.11.13"""
    # 去掉第一个 / 之前的所有内容
    if '/' in full_name:
        return full_name.split('/', 1)[-1]
    return full_name
def tag_and_push(image_full):
    """打标签并推送，显示进度条"""
    short = short_image_name(image_full)
    new_tag = f"{docker_username}/{short}"
    print(f"\n🚀 处理镜像: {image_full}")
    print(f"   → 新标签: {new_tag}")
    # 1. 拉取原始镜像（确保本地存在）
    try:
        client.images.get(image_full)
    except docker.errors.imagenotfound:
        print(f"   ⚠️ 本地未找到 {image_full}，尝试拉取...")
        client.images.pull(image_full)
    # 2. 打标签
    img = client.images.get(image_full)
    img.tag(new_tag)
    # 3. 推送（带进度条）
    print("   ⬆️ 开始推送...")
    layers_progress = {}
    for line in client.images.push(new_tag, stream=true, decode=true):
        if 'id' in line and 'progressdetail' in line:
            layer_id = line['id']
            progress = line['progressdetail']
            total = progress.get('total')
            current = progress.get('current')
            if total and current:
                if layer_id not in layers_progress:
                    layers_progress[layer_id] = tqdm(
                        total=total, unit='b', unit_scale=true,
                        desc=f"    层 {layer_id[:12]}", leave=false
                    )
                layers_progress[layer_id].update(current - layers_progress[layer_id].n)
                if current >= total:
                    layers_progress[layer_id].close()
                    del layers_progress[layer_id]
        elif 'status' in line:
            status = line['status']
            if 'already exists' in status.lower():
                tqdm.write(f"   ℹ️ {status}")
    print(f"   ✅ 推送完成: {new_tag}\n")
def main():
    print("🔍 获取所有容器使用的镜像...")
    images = get_unique_images_from_all_containers()
    if not images:
        print("❌ 没有找到任何容器镜像。")
        sys.exit(1)
    print(f"📋 找到 {len(images)} 个唯一镜像:")
    for img in images:
        print(f"   - {img}")
    for img in images:
        try:
            tag_and_push(img)
        except exception as e:
            print(f"❌ 推送失败 {img}: {e}")
    print("\n🎉 所有镜像处理完毕！")
if __name__ == "__main__":
    main()

pip install docker tqdm

docker login

python3 push_images_with_progress.py

🔍 获取所有容器使用的镜像...
📋 找到 11 个唯一镜像:
   - vulhub/comfyui:3.39.1-with-manager
   - vulhub/webmin:1.910
   ...
🚀 处理镜像: vulhub/comfyui:3.39.1-with-manager
   → 新标签: rockmelodies/comfyui:3.39.1-with-manager
   ⬆️ 开始推送...
    层 a1b2c3d4e5f6: 45%|████▌     | 45.2m/100m [00:05<00:06, 8.2mb/s]
   ℹ️ layer already exists
   ✅ 推送完成: rockmelodies/comfyui:3.39.1-with-manager
🎉 所有镜像处理完毕！