diff --git a/kiauh/utils/system_utils.py b/kiauh/utils/system_utils.py index 39d9cac..588f48d 100644 --- a/kiauh/utils/system_utils.py +++ b/kiauh/utils/system_utils.py @@ -11,10 +11,12 @@ import os import shutil +import socket import subprocess import sys import time -import socket +import urllib.error +import urllib.request from pathlib import Path from typing import List @@ -269,3 +271,32 @@ def get_ipv4_addr() -> str: return "127.0.0.1" finally: s.close() + + +def download_file(url: str, target_folder: str, target_name: str, show_progress=True): + target_path = os.path.join(target_folder, target_name) + try: + if show_progress: + urllib.request.urlretrieve(url, target_path, download_progress) + else: + urllib.request.urlretrieve(url, target_path) + except urllib.error.HTTPError as e: + Logger.print_error(f"Download failed! HTTP error occured: {e}") + raise + except urllib.error.URLError as e: + Logger.print_error(f"Download failed! URL error occured: {e}") + raise + except Exception as e: + Logger.print_error(f"Download failed! An error occured: {e}") + raise + + +def download_progress(block_num, block_size, total_size): + downloaded = block_num * block_size + percent = 100 if downloaded >= total_size else downloaded / total_size * 100 + mb = 1024 * 1024 + progress = int(percent / 5) + remaining = "-" * (20 - progress) + dl = f"\rDownloading: [{'#' * progress}{remaining}]{percent:.2f}% ({downloaded/mb:.2f}/{total_size/mb:.2f}MB)" + sys.stdout.write(dl) + sys.stdout.flush()