增加下载进度显示，优化用户体验

Author: topcss

README.md

@@ -7,11 +7,11 @@ Docker Image Puller 是一个方便的工具，用于从 Docker 仓库拉取镜
 ## 特点
 
 - **无需安装 Docker 或 Python 环境**：直接使用单文件 EXE 或 Python 脚本，开箱即用。
+- **无依赖 EXE 执行**：编译为独立 EXE 文件，无需安装 Python 环境，无需安装 Docker 环境，直接在 Releases 下载就能直接使用。
 - **国内镜像源加速**：通过配置国内镜像源，大幅提高镜像下载速度，解决国内无法直接下载的问题。
 - **多架构支持**：支持多种架构（如 `amd64`、`arm64`），满足不同环境需求，尤其是 arm64 内网服务器。
 - **兼容最新 Docker Hub API**：确保与 Docker Hub 的最新接口兼容，获取最新的镜像信息。
 - **单文件 Python 脚本**：便于携带和使用，无需复杂安装。
-- **无依赖 EXE 执行**：编译为独立 EXE 文件，无需安装 Python 环境，无法安装 Docker 环境，直接在 Releases 下载就能直接使用。
 - **用户友好**：提供交互式输入，简化操作流程。
 - **优化性能**：提高下载速度和可靠性。
 

clear.bat

@@ -1,5 +1,5 @@
 @REM 删除多余文件
-@REM rmdir dist /s /q
+rmdir dist /s /q
 rmdir build /s /q
 del DockerPull.exe.spec
 del Pipfile*

dev.md

@@ -17,7 +17,7 @@ pipenv shell
 
 ``` bat
 @REM 安装依赖，在虚拟环境中
-pip install pyinstaller requests urllib3 -i https://pypi.tuna.tsinghua.edu.cn/simple/
+pip install pyinstaller requests urllib3 tqdm -i https://pypi.tuna.tsinghua.edu.cn/simple/
 @REM 打包
 pyinstaller -F -n DockerPull.exe -i favicon.ico docker_image_puller.py
 @REM 卸载依赖

docker_image_puller.py

@@ -7,6 +7,7 @@
 import requests
 from requests.adapters import HTTPAdapter
 from urllib3.util.retry import Retry
+from tqdm import tqdm
 import tarfile
 import urllib3
 urllib3.disable_warnings()
@@ -130,13 +131,20 @@ def download_layers(session, registry, repository, layers, auth_head, imgdir, re
         try:
             bresp = session.get(f'https://{registry}/v2/{repository}/blobs/{ublob}', headers=auth_head, stream=True, verify=False, timeout=30)
             bresp.raise_for_status()
+            
+            # 使用 tqdm 显示下载进度
+            total_size = int(bresp.headers.get('content-length', 0))
+            with tqdm(total=total_size, unit='B', unit_scale=True, desc=f'Downloading {ublob[:12]}') as pbar:
+                with open(f'{layerdir}/layer_gzip.tar', 'wb') as file:
+                    for chunk in bresp.iter_content(chunk_size=1024):
+                        if chunk:
+                            file.write(chunk)
+                            pbar.update(len(chunk))
+        
         except requests.exceptions.RequestException as e:
             print(f'下载层错误：{e}')
             exit(1)
 
-        with open(f'{layerdir}/layer_gzip.tar', 'wb') as file:
-            shutil.copyfileobj(bresp.raw, file)
-        
         with open(f'{layerdir}/layer.tar', 'wb') as file:
             with gzip.open(f'{layerdir}/layer_gzip.tar', 'rb') as gz:
                 shutil.copyfileobj(gz, file)
@@ -229,6 +237,7 @@ def main():
     if not os.path.exists(imgdir):
         os.makedirs(imgdir)
 
+    print('开始下载层...')
     download_layers(session, registry, repository, layers, auth_head, imgdir, resp_json, imgparts, img, tag)
 
     create_image_tar(imgdir, repo, img)