From 556fbb1a44759882c416a22ca2d135a4a550a5c0 Mon Sep 17 00:00:00 2001
From: arebokert <arebokert@hotmail.com>
Date: Sat, 18 Jan 2025 17:19:29 +0100
Subject: [PATCH] [turboimagehost] add support for galleries

- added support
- raise error if gallery not found
- fix test
- fix lint issues
- simplify
---
 docs/supportedsites.md             |  2 +-
 gallery_dl/extractor/imagehosts.py | 28 ++++++++++++++++++++++++++++
 test/results/turboimagehost.py     | 10 ++++++++++
 3 files changed, 39 insertions(+), 1 deletion(-)

diff --git a/docs/supportedsites.md b/docs/supportedsites.md
index e55e868a84..9f37f6eb77 100644
--- a/docs/supportedsites.md
+++ b/docs/supportedsites.md
@@ -958,7 +958,7 @@ Consider all listed sites to potentially be NSFW.
 <tr>
     <td>Turboimagehost</td>
     <td>https://www.turboimagehost.com/</td>
-    <td>individual Images</td>
+    <td>Galleries, individual Images</td>
     <td></td>
 </tr>
 <tr>
diff --git a/gallery_dl/extractor/imagehosts.py b/gallery_dl/extractor/imagehosts.py
index 5f1e0f4852..d6b36cb517 100644
--- a/gallery_dl/extractor/imagehosts.py
+++ b/gallery_dl/extractor/imagehosts.py
@@ -286,6 +286,34 @@ def get_info(self, page):
         return url, url
 
 
+class TurboimagehostGalleryExtractor(ImagehostImageExtractor):
+    """Extractor for image galleries from turboimagehost.com"""
+    category = "turboimagehost"
+    subcategory = "gallery"
+    pattern = (r"(?:https?://)?((?:www\.)?turboimagehost\.com"
+               r"/album/(\d+)/([^/?#]*))")
+    example = "https://www.turboimagehost.com/album/12345/GALLERY_NAME"
+
+    def items(self):
+        data = {"_extractor": TurboimagehostImageExtractor}
+        params = {"p": 1}
+
+        while True:
+            page = self.request(self.page_url, params=params).text
+
+            if params["p"] == 1 and \
+                    "Requested gallery don`t exist on our website." in page:
+                raise exception.NotFoundError("gallery")
+
+            thumb_url = None
+            for thumb_url in text.extract_iter(page, '"><a href="', '"'):
+                yield Message.Queue, thumb_url, data
+            if thumb_url is None:
+                return
+
+            params["p"] += 1
+
+
 class ViprImageExtractor(ImagehostImageExtractor):
     """Extractor for single images from vipr.im"""
     category = "vipr"
diff --git a/test/results/turboimagehost.py b/test/results/turboimagehost.py
index 642d932193..ff3b6723e8 100644
--- a/test/results/turboimagehost.py
+++ b/test/results/turboimagehost.py
@@ -17,7 +17,17 @@
     "#sha1_content" : (
         "f38b54b17cd7462e687b58d83f00fca88b1b105a",
         "0c8768055e4e20e7c7259608b67799171b691140",
+        "961b25d85b5f5bd18cbe3e847ac55925f14d0286"
     ),
 },
 
+{
+    "#url"     : "https://www.turboimagehost.com/album/344597/testimagegallery",
+    "#category": ("imagehost", "turboimagehost", "gallery"),
+    "#class"   : imagehosts.TurboimagehostGalleryExtractor,
+    "#pattern" : imagehosts.TurboimagehostImageExtractor.pattern,
+    "#sha1_url": "f2d4fe102fdd71dd1f595cdb0c16ce999d6bb19b",
+    "#count"   : 110,
+},
+
 )