diff --git a/gallery_dl/extractor/imagefap.py b/gallery_dl/extractor/imagefap.py
index 3bdcfdfd..32fe3715 100644
--- a/gallery_dl/extractor/imagefap.py
+++ b/gallery_dl/extractor/imagefap.py
@@ -161,11 +161,12 @@ class ImagefapFolderExtractor(ImagefapExtractor):
self.user = user or profile
def items(self):
- for gallery_id, name in self.galleries(self.folder_id):
+ for gallery_id, name, folder in self.galleries(self.folder_id):
url = "{}/gallery/{}".format(self.root, gallery_id)
data = {
"gallery_id": gallery_id,
"title" : text.unescape(name),
+ "folder" : text.unescape(folder),
"_extractor": ImagefapGalleryExtractor,
}
yield Message.Queue, url, data
@@ -173,6 +174,7 @@ class ImagefapFolderExtractor(ImagefapExtractor):
def galleries(self, folder_id):
"""Yield gallery IDs and titles of a folder"""
if folder_id == "-1":
+ folder_name = "Uncategorized"
if self._id:
url = "{}/usergallery.php?userid={}&folderid=-1".format(
self.root, self.user)
@@ -180,23 +182,28 @@ class ImagefapFolderExtractor(ImagefapExtractor):
url = "{}/profile/{}/galleries?folderid=-1".format(
self.root, self.user)
else:
+ folder_name = None
url = "{}/organizer/{}/".format(self.root, folder_id)
params = {"page": 0}
+ extr = text.extract_from(self.request(url, params=params).text)
+ if not folder_name:
+ folder_name = extr("class'blk_galleries'>", "")
+
while True:
- extr = text.extract_from(self.request(url, params=params).text)
cnt = 0
while True:
gid = extr('", "<")
+ yield gid, extr("", "<"), folder_name
cnt += 1
if cnt < 20:
break
params["page"] += 1
+ extr = text.extract_from(self.request(url, params=params).text)
class ImagefapUserExtractor(ImagefapExtractor):
diff --git a/test/results/imagefap.py b/test/results/imagefap.py
index bec94011..7ac1631f 100644
--- a/test/results/imagefap.py
+++ b/test/results/imagefap.py
@@ -124,7 +124,11 @@ __tests__ = (
"#url" : "https://www.imagefap.com/usergallery.php?userid=1981976&folderid=409758",
"#category": ("", "imagefap", "folder"),
"#class" : imagefap.ImagefapFolderExtractor,
- "#sha1_url": "37822523e6e4a56feb9dea35653760c86b44ff89",
+ "#urls" : "https://www.imagefap.com/gallery/7876223",
+
+ "folder" : "Softcore",
+ "gallery_id": "7876223",
+ "title" : "Kelsi Monroe in lingerie",
},
{
@@ -140,6 +144,8 @@ __tests__ = (
"#class" : imagefap.ImagefapFolderExtractor,
"#pattern" : imagefap.ImagefapGalleryExtractor.pattern,
"#range" : "1-40",
+
+ "folder": "Uncategorized",
},
{