nhpdf 1.0__py3-none-any.whl → 1.1__py3-none-any.whl

nhpdf/nhpdf.py

@@ -2,37 +2,42 @@ import img2pdf # type: ignore
 import requests
 from typing import NoReturn
 from bs4 import BeautifulSoup
+from concurrent.futures import ThreadPoolExecutor
 import re
 import sys
 import time
 import threading
 from pathlib import Path
+from math import trunc
 
 nhpdf_dir = Path.home()/"Documents"/"nhpdf"
 nhpdf_dir.mkdir(parents=True, exist_ok=True) 
 
-def loading_animation():
+def loading_animation() -> NoReturn:
     global la, pages, page
     spinner = ['|', '/', '-', '\\']
     while not la:
         for frame in spinner:
-            sys.stdout.write(f'\rDownloading the pages ({page}/{pages})...{frame}')
+            sys.stdout.write(f'\rDownloading the pages ({trunc((page/pages)*100)}%/{100}%)...{frame}')
             sys.stdout.flush()
             time.sleep(0.1)
             if la:  
                 break
-        
-def compile_images(raw: list, name: str) -> NoReturn:
+
+def download_image(raw_url: str) -> bytes:
     global pages, page
-    raw_images = []
-    for item in raw:
-        page += 1
-        img_code, f_type = re.search(r'/(\d+)', item['data-src']).group(), re.search(r'\.([a-zA-Z0-9]+)$', item['data-src']).group()
-        url = f'https://i3.nhentai.net/galleries{img_code}/{page}{f_type}'
-        response = requests.get(url)
-        raw_images.append(response.content)
-        if page == pages:
-            break
+    page += 1
+    img_code, f_type = re.search(r'/(\d+)', raw_url['data-src']).group(), re.search(r'\b(.(jpg|jpeg|png|webp))\b', raw_url['data-src']).group()
+    url = f'https://i3.nhentai.net/galleries{img_code}/{page}{f_type}'
+    response = requests.get(url)
+    if response.status_code == 200:
+        return response.content
+    return None
+
+
+def compile_images(raw: list, name: str) -> NoReturn:
+    with ThreadPoolExecutor(max_workers=10) as executor:
+        raw_images = list(executor.map(download_image, raw))
     nhpdf = nhpdf_dir/f"{name}.pdf"
     with open(nhpdf, "wb") as file:
         file.write(img2pdf.convert(raw_images))
@@ -68,9 +73,10 @@ def start_nhpdf():
             print("\n[ERROR]: The code cannot be found in the website.\n\ngoing to the next code...")
             continue
         print(f"\nH-Doujin Details:\nname: {name}\nauthor: {author}\npages: {pages}\n")
+        raw_data.remove(raw_data[0])
         function_thread = threading.Thread(target=loading_animation)
         function_thread.start()
-        compile_images(raw_data, name)
+        compile_images(raw_data[:pages], name)
         la = True
         function_thread.join()
         if item != items:

{nhpdf-1.0.dist-info → nhpdf-1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nhpdf
-Version: 1.0
+Version: 1.1
 Summary: Download NHentai doujins as PDF
 Project-URL: Homepage, https://github.com/Aze543/NhentaiDoujinToPDFMaker
 Project-URL: Issues, https://github.com/Aze543/NhentaiDoujinToPDFMaker/issues
@@ -8,7 +8,7 @@ Author-email: Aze543 <eizen.9258437663@gmail.com>
 Maintainer-email: Aze543 <eizen.9258437663@gmail.com>
 License-Expression: MIT
 Keywords: goon,pdf,scraper
-Classifier: Development Status :: 5 - Production/Stable
+Classifier: Development Status :: 4 - Beta
 Classifier: Programming Language :: Python
 Requires-Python: >=3.8
 Requires-Dist: beautifulsoup4

nhpdf-1.1.dist-info/RECORD

@@ -0,0 +1,6 @@
+nhpdf/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nhpdf/nhpdf.py,sha256=UCOMrCbjVINbv9tj-dxM--ajzOloU_4ti3DenI0arCY,3131
+nhpdf-1.1.dist-info/METADATA,sha256=euYbpRlYPAQGxNFW0V3thJ3gV67qRJq4BkHqWtZ9C8E,2864
+nhpdf-1.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+nhpdf-1.1.dist-info/entry_points.txt,sha256=Dn0ISURSvQrGZMCOxYEA5ocSWU5DUOiT50J7FHZk5Xs,50
+nhpdf-1.1.dist-info/RECORD,,

nhpdf-1.0.dist-info/RECORD

@@ -1,6 +0,0 @@
-nhpdf/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nhpdf/nhpdf.py,sha256=H8fl4xUlBbO16MPuJVQ0WzYnHJSQvW7HItj5-AhjxoY,2873
-nhpdf-1.0.dist-info/METADATA,sha256=p0xAaaQL0xmpRk3xwRbyOm8En2Ue83VslE3ZbDFiFoc,2877
-nhpdf-1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-nhpdf-1.0.dist-info/entry_points.txt,sha256=Dn0ISURSvQrGZMCOxYEA5ocSWU5DUOiT50J7FHZk5Xs,50
-nhpdf-1.0.dist-info/RECORD,,