diff --git a/utils/downloader.py b/utils/downloader.py index 7d2e159..0e08b17 100644 --- a/utils/downloader.py +++ b/utils/downloader.py @@ -11,6 +11,7 @@ import os import concurrent.futures import requests import time +from PIL import Image from utils.Ntfy import ntfy from utils.comic.ComicInfo import comicInfo from utils.HtmlUtils import htmlUtils @@ -51,6 +52,14 @@ def download_image(image_url, dst_dir, file_name, timeout=20, proxy_type=None, p with open(temp_path, 'wb') as f: f.write(response.content) response.close() + #验证是否是图像 + try: + img = Image.open(temp_path) + img.verify() + except: + os.remove(temp_path) + print(f"{temp_path}已损坏,删除重试中") + return None shutil.move(temp_path, file_path) print("## OK: {} {}".format(file_path, image_url)) except Exception as e: