From f1d44f48c7e63ba53b46b2c7bb59404a91a29e7a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jan-Marten=20Br=C3=BCggemann?= Date: Sat, 14 Sep 2024 01:37:07 +0200 Subject: [PATCH] abort if digest could not be retreived --- sync_kiwix.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/sync_kiwix.py b/sync_kiwix.py index 282baeb..ea7a00f 100644 --- a/sync_kiwix.py +++ b/sync_kiwix.py @@ -146,6 +146,8 @@ def download_wiki(session: requests.Session, title: str, src: str, dst: str): if method == "SHA-256": digest_value = base64.b64decode(data) break + else: + raise SystemExit(f"Could not get SHA-256 digest for {title}. Aborting.") sha256 = hashlib.sha256() with session.get(url=src, stream=True) as response: @@ -194,7 +196,7 @@ def main(): filelist = retreive_filelist(session=session, url=KIWIX_BASE_URL) save_cached_filelist(cache_file=args.cache_file, filelist=filelist) - # Get downlaod files list + # Get download files list download_files = get_download_files(wikis=args.wiki, filelist=filelist) # Download files