You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
112 lines
3.8 KiB
112 lines
3.8 KiB
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright 2015-2018 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://nhentai.net/"""
|
|
|
|
from .common import Extractor, Message
|
|
from .. import text
|
|
|
|
|
|
class NHentaiExtractor(Extractor):
|
|
"""Base class for nhentai extractors"""
|
|
category = "nhentai"
|
|
root = "https://nhentai.net"
|
|
media_url = "https://i.nhentai.net"
|
|
|
|
@staticmethod
|
|
def transform_to_metadata(ginfo):
|
|
"""Transform an nhentai API response into compatible metadata"""
|
|
title_en = ginfo["title"].get("english", "")
|
|
title_ja = ginfo["title"].get("japanese", "")
|
|
return {
|
|
"gallery_id": ginfo["id"],
|
|
"upload_date": ginfo["upload_date"],
|
|
"media_id": ginfo["media_id"],
|
|
"scanlator": ginfo["scanlator"],
|
|
"count": ginfo["num_pages"],
|
|
"title": title_en or title_ja,
|
|
"title_en": title_en,
|
|
"title_ja": title_ja,
|
|
}
|
|
|
|
|
|
class NhentaiGalleryExtractor(NHentaiExtractor):
|
|
"""Extractor for image galleries from nhentai.net"""
|
|
subcategory = "gallery"
|
|
directory_fmt = ["{category}", "{gallery_id} {title}"]
|
|
filename_fmt = "{category}_{gallery_id}_{num:>03}.{extension}"
|
|
archive_fmt = "{gallery_id}_{num}"
|
|
pattern = [r"(?:https?://)?nhentai\.net/g/(\d+)"]
|
|
test = [("https://nhentai.net/g/147850/", {
|
|
"url": "5179dbf0f96af44005a0ff705a0ad64ac26547d0",
|
|
"keyword": "2f94976e657f3043a89997e22f4de8e1b22d9175",
|
|
})]
|
|
|
|
def __init__(self, match):
|
|
NHentaiExtractor.__init__(self)
|
|
self.gid = match.group(1)
|
|
|
|
def items(self):
|
|
ginfo = self.get_gallery_info(self.gid)
|
|
data = self.transform_to_metadata(ginfo)
|
|
urlfmt = "{}/galleries/{}/{{}}.{{}}".format(
|
|
self.media_url, data["media_id"])
|
|
extdict = {"j": "jpg", "p": "png", "g": "gif"}
|
|
yield Message.Version, 1
|
|
yield Message.Directory, data
|
|
for data["num"], image in enumerate(ginfo["images"]["pages"], 1):
|
|
ext = extdict.get(image["t"], "jpg")
|
|
data["width"] = image["w"]
|
|
data["height"] = image["h"]
|
|
data["extension"] = ext
|
|
yield Message.Url, urlfmt.format(data["num"], ext), data
|
|
|
|
def get_gallery_info(self, gallery_id):
|
|
"""Extract and return info about a gallery by ID"""
|
|
url = "{}/api/gallery/{}".format(self.root, gallery_id)
|
|
return self.request(url).json()
|
|
|
|
|
|
class NhentaiSearchExtractor(NHentaiExtractor):
|
|
"""Extractor for nhentai search results"""
|
|
category = "nhentai"
|
|
subcategory = "search"
|
|
pattern = [r"(?:https?://)?nhentai\.net/search/?\?(.*)"]
|
|
|
|
def __init__(self, match):
|
|
NHentaiExtractor.__init__(self)
|
|
self.params = text.parse_query(match.group(1))
|
|
|
|
if "q" in self.params:
|
|
self.params["query"] = self.params["q"]
|
|
del self.params["q"]
|
|
|
|
def items(self):
|
|
yield Message.Version, 1
|
|
for ginfo in self._pagination("galleries/search", self.params):
|
|
url = "{}/g/{}/".format(self.root, ginfo["id"])
|
|
yield Message.Queue, url, self.transform_to_metadata(ginfo)
|
|
|
|
def _pagination(self, endpoint, params):
|
|
"""Pagination over API responses"""
|
|
url = "{}/api/{}".format(self.root, endpoint)
|
|
params["page"] = text.parse_int(params.get("page"), 1)
|
|
|
|
while True:
|
|
data = self.request(url, params=params, fatal=False).json()
|
|
|
|
if "error" in data:
|
|
self.log.error("API request failed: \"%s\"", data["error"])
|
|
return
|
|
|
|
yield from data["result"]
|
|
|
|
if params["page"] >= data["num_pages"]:
|
|
return
|
|
params["page"] += 1
|