diff --git a/nhentai/doujinshi.py b/nhentai/doujinshi.py index 5478cb2..d5efe24 100644 --- a/nhentai/doujinshi.py +++ b/nhentai/doujinshi.py @@ -12,6 +12,7 @@ EXT_MAP = { 'j': 'jpg', 'p': 'png', 'g': 'gif', + 'w': 'webp', } diff --git a/nhentai/parser.py b/nhentai/parser.py index 5b7f26c..fcee9ad 100644 --- a/nhentai/parser.py +++ b/nhentai/parser.py @@ -148,7 +148,7 @@ def doujinshi_parser(id_, counter=0): doujinshi['subtitle'] = subtitle.text if subtitle else '' doujinshi_cover = html.find('div', attrs={'id': 'cover'}) - img_id = re.search('/galleries/([0-9]+)/cover.(jpg|png|gif)$', + img_id = re.search('/galleries/([0-9]+)/cover.(jpg|png|gif|webp)$', doujinshi_cover.a.img.attrs['data-src']) ext = [] 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}')