diff options
Diffstat (limited to 'gallery_dl/extractor/newgrounds.py')
| -rw-r--r-- | gallery_dl/extractor/newgrounds.py | 155 |
1 files changed, 155 insertions, 0 deletions
diff --git a/gallery_dl/extractor/newgrounds.py b/gallery_dl/extractor/newgrounds.py new file mode 100644 index 0000000..9e0aaa3 --- /dev/null +++ b/gallery_dl/extractor/newgrounds.py @@ -0,0 +1,155 @@ +# -*- coding: utf-8 -*- + +# Copyright 2018-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. + +"""Extractors for https://www.newgrounds.com/""" + +from .common import Extractor, Message +from .. import text +import json + + +class NewgroundsExtractor(Extractor): + """Base class for newgrounds extractors""" + category = "newgrounds" + directory_fmt = ("{category}", "{user}") + filename_fmt = "{category}_{index}_{title}.{extension}" + archive_fmt = "{index}" + + def __init__(self, match): + Extractor.__init__(self, match) + self.user = match.group(1) + self.root = "https://{}.newgrounds.com".format(self.user) + + def items(self): + data = self.get_metadata() + yield Message.Version, 1 + yield Message.Directory, data + + for page_url in self.get_page_urls(): + image = self.parse_page_data(page_url) + image.update(data) + url = image["url"] + yield Message.Url, url, text.nameext_from_url(url, image) + + def get_metadata(self): + """Collect metadata for extractor-job""" + return {"user": self.user} + + def get_page_urls(self): + """Return urls of all relevant image pages""" + + def parse_page_data(self, page_url): + """Collect url and metadata from an image page""" + extr = text.extract_from(self.request(page_url).text) + full = text.extract_from(json.loads(extr('"full_image_text":', '});'))) + data = { + "description": text.unescape(extr(':description" content="', '"')), + "date" : extr('itemprop="datePublished" content="', '"'), + "rating" : extr('class="rated-', '"'), + "favorites" : text.parse_int(extr('id="faves_load">', '<')), + "score" : text.parse_float(extr('id="score_number">', '<')), + "url" : full('src="', '"'), + "title" : text.unescape(full('alt="', '"')), + "width" : text.parse_int(full('width="', '"')), + "height" : text.parse_int(full('height="', '"')), + } + + tags = text.split_html(extr('<dd class="tags momag">', '</dd>')) + tags.sort() + data["tags"] = tags + + data["date"] = text.parse_datetime(data["date"]) + data["index"] = text.parse_int( + data["url"].rpartition("/")[2].partition("_")[0]) + return data + + def _pagination(self, url): + headers = { + "Referer": self.root, + "X-Requested-With": "XMLHttpRequest", + "Accept": "application/json, text/javascript, */*; q=0.01", + } + + while True: + data = self.request(url, headers=headers).json() + + for year in data["sequence"]: + for item in data["years"][str(year)]["items"]: + page_url = text.extract(item, 'href="', '"')[0] + yield text.urljoin(self.root, page_url) + + if not data["more"]: + return + url = text.urljoin(self.root, data["more"]) + + +class NewgroundsUserExtractor(NewgroundsExtractor): + """Extractor for all images of a newgrounds user""" + subcategory = "user" + pattern = r"(?:https?://)?([^.]+)\.newgrounds\.com(?:/art)?/?$" + test = ( + ("https://blitzwuff.newgrounds.com/art", { + "url": "24b19c4a135a09889fac7b46a74e427e4308d02b", + "keyword": "2aab0532a894ff3cf88dd01ce5c60f114011b268", + }), + ("https://blitzwuff.newgrounds.com/"), + ) + + def get_page_urls(self): + return self._pagination(self.root + "/art/page/1") + + +class NewgroundsImageExtractor(NewgroundsExtractor): + """Extractor for a single image from newgrounds.com""" + subcategory = "image" + pattern = (r"(?:https?://)?(?:" + r"(?:www\.)?newgrounds\.com/art/view/([^/?&#]+)/[^/?&#]+" + r"|art\.ngfiles\.com/images/\d+/\d+_([^_]+)_([^.]+))") + test = ( + ("https://www.newgrounds.com/art/view/blitzwuff/ffx", { + "url": "e7778c4597a2fb74b46e5f04bb7fa1d80ca02818", + "keyword": "cbe90f8f32da4341938f59b08d70f76137028a7e", + "content": "cb067d6593598710292cdd340d350d14a26fe075", + }), + ("https://art.ngfiles.com/images/587000/587551_blitzwuff_ffx.png", { + "url": "e7778c4597a2fb74b46e5f04bb7fa1d80ca02818", + "keyword": "cbe90f8f32da4341938f59b08d70f76137028a7e", + }), + ) + + def __init__(self, match): + NewgroundsExtractor.__init__(self, match) + if match.group(2): + self.user = match.group(2) + self.page_url = "https://www.newgrounds.com/art/view/{}/{}".format( + self.user, match.group(3)) + else: + self.page_url = match.group(0) + + def get_page_urls(self): + return (self.page_url,) + + +class NewgroundsVideoExtractor(NewgroundsExtractor): + """Extractor for all videos of a newgrounds user""" + subcategory = "video" + filename_fmt = "{category}_{index}.{extension}" + pattern = r"(?:https?://)?([^.]+)\.newgrounds\.com/movies/?$" + test = ("https://twistedgrim.newgrounds.com/movies", { + "pattern": r"ytdl:https?://www\.newgrounds\.com/portal/view/\d+", + "count": ">= 29", + }) + + def get_page_urls(self): + return self._pagination(self.root + "/movies/page/1") + + def parse_page_data(self, page_url): + return { + "url" : "ytdl:" + page_url, + "index": text.parse_int(page_url.rpartition("/")[2]), + } |
