Import Upstream version 1.8.7upstream/1.8.7

1 files changed, 195 insertions, 0 deletions

diff --git a/gallery_dl/extractor/imagefap.py b/gallery_dl/extractor/imagefap.py
new file mode 100644
index 0000000..152b631
--- /dev/null
+++ b/gallery_dl/extractor/imagefap.py
@@ -0,0 +1,195 @@
+# -*- coding: utf-8 -*-
+
+# Copyright 2016-2019 Mike Fährmann
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License version 2 as
+# published by the Free Software Foundation.
+
+"""Extract images from https://imagefap.com/"""
+
+from .common import Extractor, Message
+from .. import text
+import json
+
+
+class ImagefapExtractor(Extractor):
+    """Base class for imagefap extractors"""
+    category = "imagefap"
+    directory_fmt = ("{category}", "{gallery_id} {title}")
+    filename_fmt = "{category}_{gallery_id}_{filename}.{extension}"
+    archive_fmt = "{gallery_id}_{image_id}"
+    root = "https://www.imagefap.com"
+
+    def __init__(self, match):
+        Extractor.__init__(self, match)
+        self.session.headers["Referer"] = self.root
+
+
+class ImagefapGalleryExtractor(ImagefapExtractor):
+    """Extractor for image galleries from imagefap.com"""
+    subcategory = "gallery"
+    pattern = (r"(?:https?://)?(?:www\.)?imagefap\.com/"
+               r"(?:gallery\.php\?gid=|gallery/|pictures/)(\d+)")
+    test = (
+        ("https://www.imagefap.com/pictures/7102714", {
+            "url": "268995eac5d01ddecd0fe58cfa9828390dc85a84",
+            "keyword": "b5bd65ab2ff574ed1639db9a43c7b1b8583c85ef",
+            "content": "694a0a57385980a6f90fbc296cadcd6c11ba2dab",
+        }),
+        ("https://www.imagefap.com/gallery/5486966", {
+            "url": "14906b4f0b8053d1d69bc730a325acb793cbc898",
+            "keyword": "ab90972f3527a2011478fabc621a2c99a541f752",
+        }),
+        ("https://www.imagefap.com/gallery.php?gid=7102714"),
+    )
+
+    def __init__(self, match):
+        ImagefapExtractor.__init__(self, match)
+        self.gid = match.group(1)
+        self.image_id = ""
+
+    def items(self):
+        url = "{}/pictures/{}/".format(self.root, self.gid)
+        page = self.request(url).text
+        data = self.get_job_metadata(page)
+        yield Message.Version, 1
+        yield Message.Directory, data
+        for url, image in self.get_images():
+            data.update(image)
+            yield Message.Url, url, data
+
+    def get_job_metadata(self, page):
+        """Collect metadata for extractor-job"""
+        descr, pos = text.extract(
+            page, '<meta name="description" content="Browse ', '"')
+        count, pos = text.extract(page, ' 1 of ', ' pics"', pos)
+        self.image_id = text.extract(page, 'id="img_ed_', '"', pos)[0]
+
+        title, _, descr = descr.partition(" porn picture gallery by ")
+        uploader, _, tags = descr.partition(" to see hottest ")
+        return {
+            "gallery_id": text.parse_int(self.gid),
+            "title": text.unescape(title),
+            "uploader": uploader,
+            "tags": tags[:-11].split(", "),
+            "count": text.parse_int(count),
+        }
+
+    def get_images(self):
+        """Collect image-urls and -metadata"""
+        num = 0
+        url = "{}/photo/{}/".format(self.root, self.image_id)
+        params = {"gid": self.gid, "idx": 0, "partial": "true"}
+        while True:
+            pos = 0
+            page = self.request(url, params=params).text
+            for _ in range(24):
+                imgurl, pos = text.extract(page, '<a href="', '"', pos)
+                if not imgurl:
+                    return
+                num += 1
+                _, imgid, name = imgurl.rsplit("/", 2)
+                data = {"image_id": text.parse_int(imgid), "num": num}
+                yield imgurl, text.nameext_from_url(name, data)
+            params["idx"] += 24
+
+
+class ImagefapImageExtractor(ImagefapExtractor):
+    """Extractor for single images from imagefap.com"""
+    subcategory = "image"
+    pattern = r"(?:https?://)?(?:www\.)?imagefap\.com/photo/(\d+)"
+    test = ("https://www.imagefap.com/photo/1369341772/", {
+        "url": "b31ee405b61ff0450020a1bf11c0581ca9adb471",
+        "keyword": "eadaa8f8012298384996efd21cf1f9e9e0dddb9b",
+    })
+
+    def __init__(self, match):
+        ImagefapExtractor.__init__(self, match)
+        self.image_id = match.group(1)
+
+    def items(self):
+        data = self.get_job_metadata()
+        yield Message.Version, 1
+        yield Message.Directory, data
+        yield Message.Url, data["url"], data
+
+    def get_job_metadata(self):
+        """Collect metadata for extractor-job"""
+        url = "{}/photo/{}/".format(self.root, self.image_id)
+        page = self.request(url).text
+        info = json.loads(text.extract(
+            page, '<script type="application/ld+json">', '</script>')[0])
+        parts = info["contentUrl"].rsplit("/", 3)
+        return text.nameext_from_url(parts[3], {
+            "url": info["contentUrl"],
+            "title": text.unescape(info["name"]),
+            "uploader": info["author"],
+            "date": info["datePublished"],
+            "width": text.parse_int(info["width"]),
+            "height": text.parse_int(info["height"]),
+            "gallery_id": text.parse_int(parts[1]),
+            "image_id": text.parse_int(parts[2]),
+        })
+
+
+class ImagefapUserExtractor(ImagefapExtractor):
+    """Extractor for all galleries from a user at imagefap.com"""
+    subcategory = "user"
+    categorytransfer = True
+    pattern = (r"(?:https?://)?(?:www\.)?imagefap\.com/"
+               r"(?:profile(?:\.php\?user=|/)([^/?&#]+)"
+               r"|usergallery\.php\?userid=(\d+))")
+    test = (
+        ("https://www.imagefap.com/profile/LucyRae/galleries", {
+            "url": "d941aa906f56a75972a7a5283030eb9a8d27a4fd",
+        }),
+        ("https://www.imagefap.com/usergallery.php?userid=1862791", {
+            "url": "d941aa906f56a75972a7a5283030eb9a8d27a4fd",
+        }),
+        ("https://www.imagefap.com/profile.php?user=LucyRae"),
+    )
+
+    def __init__(self, match):
+        ImagefapExtractor.__init__(self, match)
+        self.user, self.user_id = match.groups()
+
+    def items(self):
+        yield Message.Version, 1
+        for gid, name in self.get_gallery_data():
+            url = "{}/gallery/{}".format(self.root, gid)
+            data = {
+                "gallery_id": text.parse_int(gid),
+                "title": text.unescape(name),
+                "_extractor": ImagefapGalleryExtractor,
+            }
+            yield Message.Queue, url, data
+
+    def get_gallery_data(self):
+        """Yield all gallery_ids of a specific user"""
+        folders = self.get_gallery_folders()
+        url = "{}/ajax_usergallery_folder.php".format(self.root)
+        params = {"userid": self.user_id}
+        for folder_id in folders:
+            params["id"] = folder_id
+            page = self.request(url, params=params).text
+
+            pos = 0
+            while True:
+                gid, pos = text.extract(page, '<a  href="/gallery/', '"', pos)
+                if not gid:
+                    break
+                name, pos = text.extract(page, "<b>", "<", pos)
+                yield gid, name
+
+    def get_gallery_folders(self):
+        """Create a list of all folder_ids of a specific user"""
+        if self.user:
+            url = "{}/profile/{}/galleries".format(self.root, self.user)
+        else:
+            url = "{}/usergallery.php?userid={}".format(
+                self.root, self.user_id)
+        page = self.request(url).text
+        self.user_id, pos = text.extract(page, '?userid=', '"')
+        folders, pos = text.extract(page, ' id="tgl_all" value="', '"', pos)
+        return folders.split("|")[:-1]