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", }, {