summaryrefslogtreecommitdiffstats
path: root/gallery_dl/extractor/newgrounds.py
diff options
context:
space:
mode:
Diffstat (limited to 'gallery_dl/extractor/newgrounds.py')
-rw-r--r--gallery_dl/extractor/newgrounds.py155
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]),
+ }