From 1504ee779ff11bedc6dfc889cfbba24c983bd5ce Mon Sep 17 00:00:00 2001 From: JustAHumanBean <17130206+JustAHumanBean@users.noreply.github.com> Date: Fri, 8 Nov 2024 07:49:20 +0000 Subject: [PATCH] Update utils.py --- nhentai/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nhentai/utils.py b/nhentai/utils.py index 554669b..6e50a7e 100644 --- a/nhentai/utils.py +++ b/nhentai/utils.py @@ -104,7 +104,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'): + if not os.path.splitext(image)[1] in ('.jpg', '.png', '.webp'): continue image_html += f'\n' @@ -230,7 +230,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'))] + file_list = [f for f in os.listdir(doujinshi_dir) if f.lower().endswith(('.png', '.jpg', '.jpeg', '.gif', '.webp'))] file_list.sort() logger.info(f'Writing PDF file to path: {filename}')