[hentaifoundry] add 'tag' extractor (#6465)

mikf · Nov 13, 2024 · a3276e3 · a3276e3
1 parent b62c466
commit a3276e3
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 1 deletion.
diff --git a/docs/supportedsites.md b/docs/supportedsites.md
@@ -328,7 +328,7 @@ Consider all listed sites to potentially be NSFW.
 <tr>
     <td>Hentai Foundry</td>
     <td>https://www.hentai-foundry.com/</td>
-    <td>Favorites, individual Images, Pictures, Popular Images, Recent Images, Scraps, Stories, User Profiles</td>
+    <td>Favorites, individual Images, Pictures, Popular Images, Recent Images, Scraps, Stories, Tag Searches, User Profiles</td>
     <td></td>
 </tr>
 <tr>

diff --git a/gallery_dl/extractor/hentaifoundry.py b/gallery_dl/extractor/hentaifoundry.py
@@ -254,6 +254,22 @@ def __init__(self, match):
             self.root, self.user)
 
 
+class HentaifoundryTagExtractor(HentaifoundryExtractor):
+    """Extractor for tag searches on hentaifoundry.com"""
+    subcategory = "tag"
+    directory_fmt = ("{category}", "{search_tags}")
+    archive_fmt = "t_{search_tags}_{index}"
+    pattern = BASE_PATTERN + r"/pictures/tagged/([^/?#]+)"
+    example = "https://www.hentai-foundry.com/pictures/tagged/TAG"
+
+    def __init__(self, match):
+        HentaifoundryExtractor.__init__(self, match)
+        self.page_url = "{}/pictures/tagged/{}".format(self.root, self.user)
+
+    def metadata(self):
+        return {"search_tags": self.user}
+
+
 class HentaifoundryRecentExtractor(HentaifoundryExtractor):
     """Extractor for 'Recent Pictures' on hentaifoundry.com"""
     subcategory = "recent"

diff --git a/test/results/hentaifoundry.py b/test/results/hentaifoundry.py
@@ -55,12 +55,24 @@
     "#class"   : hentaifoundry.HentaifoundryFavoriteExtractor,
 },
 
+{
+    "#url"     : "https://www.hentai-foundry.com/pictures/tagged/kancolle",
+    "#category": ("", "hentaifoundry", "tag"),
+    "#class"   : hentaifoundry.HentaifoundryTagExtractor,
+    "#pattern" : r"https://pictures.hentai-foundry.com/[^/]/[^/?#]+/\d+/",
+    "#range"   : "20-30",
+
+    "search_tags": "kancolle",
+},
+
 {
     "#url"     : "https://www.hentai-foundry.com/pictures/recent/2018-09-20",
     "#category": ("", "hentaifoundry", "recent"),
     "#class"   : hentaifoundry.HentaifoundryRecentExtractor,
     "#pattern" : r"https://pictures.hentai-foundry.com/[^/]/[^/?#]+/\d+/",
     "#range"   : "20-30",
+
+    "date": "2018-09-20",
 },
 
 {