remove debug print

add webp support

nhentai/doujinshi.py

@@ -12,6 +12,7 @@ EXT_MAP = {
     'j': 'jpg',
     'p': 'png',
     'g': 'gif',
+    'w': 'webp',
 }
 
 

nhentai/downloader.py

@@ -69,7 +69,7 @@ class Downloader(Singleton):
                         if response.status_code != 200:
                             path = urlparse(url).path
                             for mirror in constant.IMAGE_URL_MIRRORS:
-                                print(f'{mirror}{path}')
+                                # print(f'{mirror}{path}')
                                 mirror_url = f'{mirror}{path}'
                                 response = request('get', mirror_url, stream=True,
                                                    timeout=self.timeout, proxies=proxy)

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 = []

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'<img src="{image}" class="image-item"/>\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}')