1
0
mirror of https://github.com/mikf/gallery-dl.git synced 2024-11-23 19:22:32 +01:00
gallery-dl/gallery_dl/extractor/nhentai.py
2018-03-21 12:58:41 +01:00

65 lines
2.3 KiB
Python

# -*- 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
class NhentaiGalleryExtractor(Extractor):
"""Extractor for image galleries from nhentai.net"""
category = "nhentai"
subcategory = "gallery"
directory_fmt = ["{category}", "{gallery_id} {title}"]
filename_fmt = "{category}_{gallery_id}_{num:>03}.{extension}"
archive_fmt = "{gallery_id}_{num}"
pattern = [r"(?:https?://)?(?:www\.)?nhentai\.net/g/(\d+)"]
test = [("https://nhentai.net/g/147850/", {
"url": "5179dbf0f96af44005a0ff705a0ad64ac26547d0",
"keyword": "2f94976e657f3043a89997e22f4de8e1b22d9175",
})]
def __init__(self, match):
Extractor.__init__(self)
self.gid = match.group(1)
def items(self):
ginfo = self.get_gallery_info()
data = self.get_job_metadata(ginfo)
urlfmt = "https://i.nhentai.net/galleries/{}/{{}}.{{}}".format(
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):
"""Extract and return gallery-info"""
url = "https://nhentai.net/api/gallery/" + self.gid
return self.request(url).json()
def get_job_metadata(self, ginfo):
"""Collect metadata for extractor-job"""
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,
}