From 54af682848b3d7dcedbe1c5a940c25b22bdc702a Mon Sep 17 00:00:00 2001 From: ricterz Date: Sat, 11 Jan 2025 07:50:41 +0800 Subject: [PATCH] fix #369 --- nhentai/downloader.py | 1 + nhentai/utils.py | 5 +++-- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/nhentai/downloader.py b/nhentai/downloader.py index e4dac6d..4f07c76 100644 --- a/nhentai/downloader.py +++ b/nhentai/downloader.py @@ -152,6 +152,7 @@ class Downloader(Singleton): return True digit_length = len(str(len(queue))) + logger.info(f'Total download pages: {len(queue)}') coroutines = [ self._semaphore_download(url, filename=os.path.basename(urlparse(url).path), length=digit_length) for url in queue diff --git a/nhentai/utils.py b/nhentai/utils.py index 2a280bb..8253e4e 100644 --- a/nhentai/utils.py +++ b/nhentai/utils.py @@ -18,6 +18,7 @@ from nhentai.logger import logger from nhentai.serializer import serialize_json, serialize_comic_xml, set_js_database MAX_FIELD_LENGTH = 100 +EXTENSIONS = ('.png', '.jpg', '.jpeg', '.gif', '.webp') def request(method, url, **kwargs): @@ -130,7 +131,7 @@ def generate_html(output_dir='.', doujinshi_obj=None, template='default'): file_list.sort() for image in file_list: - if not os.path.splitext(image)[1] in ('.jpg', '.png', '.webp'): + if not os.path.splitext(image)[1] in EXTENSIONS: continue image_html += f'\n' @@ -256,7 +257,7 @@ def generate_doc(file_type='', output_dir='.', doujinshi_obj=None, regenerate=Fa import img2pdf """Write images to a PDF file using img2pdf.""" - file_list = [f for f in os.listdir(doujinshi_dir) if f.lower().endswith(('.png', '.jpg', '.jpeg', '.gif', '.webp'))] + file_list = [f for f in os.listdir(doujinshi_dir) if f.lower().endswith(EXTENSIONS)] file_list.sort() logger.info(f'Writing PDF file to path: {filename}')