#!/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"""
    # 去掉第一個 / 之前的所有內(nèi)容
    if '/' in full_name:
        return full_name.split('/', 1)[-1]
    return full_name
def tag_and_push(image_full):
    """打標(biāo)簽并推送，顯示進(jìn)度條"""
    short = short_image_name(image_full)
    new_tag = f"{DOCKER_USERNAME}/{short}"
    print(f"\n?? 處理鏡像: {image_full}")
    print(f"   → 新標(biāo)簽: {new_tag}")
    # 1. 拉取原始鏡像（確保本地存在）
    try:
        client.images.get(image_full)
    except docker.errors.ImageNotFound:
        print(f"   ?? 本地未找到 {image_full}，嘗試?yán)?..")
        client.images.pull(image_full)
    # 2. 打標(biāo)簽
    img = client.images.get(image_full)
    img.tag(new_tag)
    # 3. 推送（帶進(jìn)度條）
    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
   → 新標(biāo)簽: 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
?? 所有鏡像處理完畢！