From 4c5cd5840dd26fa17c8becfd717681a003a22126 Mon Sep 17 00:00:00 2001 From: Kentaro Wada Date: Fri, 2 Feb 2024 18:27:53 +0900 Subject: [PATCH] Customize download log messages in cached_download function --- gdown/cached_download.py | 20 +++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/gdown/cached_download.py b/gdown/cached_download.py index 2d714140..32b081fc 100644 --- a/gdown/cached_download.py +++ b/gdown/cached_download.py @@ -129,15 +129,17 @@ def cached_download( try: temp_path = osp.join(temp_root, "dl") - if not quiet: - msg = "Cached Downloading" - if path: - msg = f"{msg}: {path}" - else: - msg = f"{msg}..." - print(msg, file=sys.stderr) - - download(url, temp_path, quiet=quiet, **kwargs) + log_message_hash = f"Hash: {hash}\n" if hash else "" + download( + url=url, + output=temp_path, + quiet=quiet, + log_messages={ + "start": f"Cached downloading...\n{log_message_hash}", + "output": f"To: {path}\n", + }, + **kwargs, + ) with filelock.FileLock(lock_path): shutil.move(temp_path, path) except Exception: