nhpdf 1.6__py3-none-any.whl → 1.7__py3-none-any.whl

nhpdf/nhpdf.py

@@ -9,6 +9,8 @@ import time
 import threading
 from pathlib import Path
 from math import trunc
+import io
+from PIL import Image
 
 nhpdf_dir = Path.home()/"Documents"/"nhpdf"
 nhpdf_dir.mkdir(parents=True, exist_ok=True) 
@@ -33,16 +35,32 @@ def download_image(raw_url: str) -> bytes:
     url = f'https://i3.nhentai.net/galleries{img_code}/{page}{f_type}'
     response = requests.get(url)
     if response.status_code == 200:
-        return response.content
+        content = check_alpha(response.content)
+        return 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))
+    for img in raw_images:
+        check_alpha(img)
     nhpdf = nhpdf_dir/f"{name}.pdf"
     with open(nhpdf, "wb") as file:
         file.write(img2pdf.convert(raw_images))
 
+def check_alpha(image: bytes) -> bytes:
+    try:
+        img2pdf.convert(image)
+    except img2pdf.AlphaChannelError as alphaError:
+        buffered = io.BytesIO(image)
+        img = Image.open(buffered)
+        converted: Image = img.convert('RGB')
+        buf = io.BytesIO()
+        converted.save(buf, format='PNG')
+        image = buf.getvalue() 
+    return image
+
+
 def start_nhpdf():
     global la, page, pages
     if len(sys.argv) < 2:

{nhpdf-1.6.dist-info → nhpdf-1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nhpdf
-Version: 1.6
+Version: 1.7
 Summary: Download NHentai doujins as PDF
 Project-URL: Homepage, https://github.com/Aze543/NhentaiDoujinToPDFMaker
 Project-URL: Issues, https://github.com/Aze543/NhentaiDoujinToPDFMaker/issues

nhpdf-1.7.dist-info/RECORD

@@ -0,0 +1,6 @@
+nhpdf/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nhpdf/nhpdf.py,sha256=YQl9bsXDgv7uJTpW_iiCRDFyRahJYVPKvscbtKC-sgs,3758
+nhpdf-1.7.dist-info/METADATA,sha256=NbXvJvEAQTbGQsxL3Ok6vyeQTlXStb9Nkzvsxu7RDMw,2873
+nhpdf-1.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+nhpdf-1.7.dist-info/entry_points.txt,sha256=Dn0ISURSvQrGZMCOxYEA5ocSWU5DUOiT50J7FHZk5Xs,50
+nhpdf-1.7.dist-info/RECORD,,

nhpdf-1.6.dist-info/RECORD

@@ -1,6 +0,0 @@
-nhpdf/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nhpdf/nhpdf.py,sha256=-o0ryeh69xNosrLmP8SD_MIOawLXvKb6x3vo1oFmpPw,3265
-nhpdf-1.6.dist-info/METADATA,sha256=wqI6ZqBEwE6Vc2SSGehddMfwqH10JeaLObON9venzD8,2873
-nhpdf-1.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-nhpdf-1.6.dist-info/entry_points.txt,sha256=Dn0ISURSvQrGZMCOxYEA5ocSWU5DUOiT50J7FHZk5Xs,50
-nhpdf-1.6.dist-info/RECORD,,