mirror of
https://github.com/mikf/gallery-dl.git
synced 2024-11-22 10:42:34 +01:00
283 lines
9.3 KiB
Python
283 lines
9.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright 2015-2023 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://hitomi.la/"""
|
|
|
|
from .common import GalleryExtractor, Extractor, Message
|
|
from .nozomi import decode_nozomi
|
|
from ..cache import memcache
|
|
from .. import text, util
|
|
import string
|
|
import re
|
|
|
|
|
|
class HitomiGalleryExtractor(GalleryExtractor):
|
|
"""Extractor for image galleries from hitomi.la"""
|
|
category = "hitomi"
|
|
root = "https://hitomi.la"
|
|
pattern = (r"(?:https?://)?hitomi\.la"
|
|
r"/(?:manga|doujinshi|cg|gamecg|imageset|galleries|reader)"
|
|
r"/(?:[^/?#]+-)?(\d+)")
|
|
example = "https://hitomi.la/manga/TITLE-867789.html"
|
|
|
|
def __init__(self, match):
|
|
self.gid = match.group(1)
|
|
url = "https://ltn.hitomi.la/galleries/{}.js".format(self.gid)
|
|
GalleryExtractor.__init__(self, match, url)
|
|
|
|
def _init(self):
|
|
self.session.headers["Referer"] = "{}/reader/{}.html".format(
|
|
self.root, self.gid)
|
|
|
|
def metadata(self, page):
|
|
self.info = info = util.json_loads(page.partition("=")[2])
|
|
iget = info.get
|
|
|
|
language = iget("language")
|
|
if language:
|
|
language = language.capitalize()
|
|
|
|
date = iget("date")
|
|
if date:
|
|
date += ":00"
|
|
|
|
tags = []
|
|
for tinfo in iget("tags") or ():
|
|
tag = string.capwords(tinfo["tag"])
|
|
if tinfo.get("female"):
|
|
tag += " ♀"
|
|
elif tinfo.get("male"):
|
|
tag += " ♂"
|
|
tags.append(tag)
|
|
|
|
return {
|
|
"gallery_id": text.parse_int(info["id"]),
|
|
"title" : info["title"],
|
|
"title_jpn" : info.get("japanese_title") or "",
|
|
"type" : info["type"].capitalize(),
|
|
"language" : language,
|
|
"lang" : util.language_to_code(language),
|
|
"date" : text.parse_datetime(date, "%Y-%m-%d %H:%M:%S%z"),
|
|
"tags" : tags,
|
|
"artist" : [o["artist"] for o in iget("artists") or ()],
|
|
"group" : [o["group"] for o in iget("groups") or ()],
|
|
"parody" : [o["parody"] for o in iget("parodys") or ()],
|
|
"characters": [o["character"] for o in iget("characters") or ()]
|
|
}
|
|
|
|
def images(self, _):
|
|
# see https://ltn.hitomi.la/gg.js
|
|
gg_m, gg_b, gg_default = _parse_gg(self)
|
|
|
|
fmt = self.config("format") or "webp"
|
|
if fmt == "original":
|
|
subdomain, path, ext, check = "b", "images", None, False
|
|
else:
|
|
subdomain, path, ext, check = "a", fmt, fmt, (fmt != "webp")
|
|
|
|
result = []
|
|
for image in self.info["files"]:
|
|
if check:
|
|
if image.get("has" + fmt):
|
|
path = ext = fmt
|
|
else:
|
|
path = ext = "webp"
|
|
ihash = image["hash"]
|
|
idata = text.nameext_from_url(image["name"])
|
|
idata["extension_original"] = idata["extension"]
|
|
if ext:
|
|
idata["extension"] = ext
|
|
|
|
# see https://ltn.hitomi.la/common.js
|
|
inum = int(ihash[-1] + ihash[-3:-1], 16)
|
|
url = "https://{}{}.hitomi.la/{}/{}/{}/{}.{}".format(
|
|
chr(97 + gg_m.get(inum, gg_default)),
|
|
subdomain, path, gg_b, inum, ihash, idata["extension"],
|
|
)
|
|
result.append((url, idata))
|
|
return result
|
|
|
|
|
|
class HitomiTagExtractor(Extractor):
|
|
"""Extractor for galleries from tag searches on hitomi.la"""
|
|
category = "hitomi"
|
|
subcategory = "tag"
|
|
root = "https://hitomi.la"
|
|
pattern = (r"(?:https?://)?hitomi\.la"
|
|
r"/(tag|artist|group|series|type|character)"
|
|
r"/([^/?#]+)\.html")
|
|
example = "https://hitomi.la/tag/TAG-LANG.html"
|
|
|
|
def __init__(self, match):
|
|
Extractor.__init__(self, match)
|
|
self.type, self.tag = match.groups()
|
|
|
|
tag, _, num = self.tag.rpartition("-")
|
|
if num.isdecimal():
|
|
self.tag = tag
|
|
|
|
def items(self):
|
|
data = {"_extractor": HitomiGalleryExtractor}
|
|
nozomi_url = "https://ltn.hitomi.la/{}/{}.nozomi".format(
|
|
self.type, self.tag)
|
|
headers = {
|
|
"Origin": self.root,
|
|
"Cache-Control": "max-age=0",
|
|
}
|
|
|
|
offset = 0
|
|
total = None
|
|
while True:
|
|
headers["Referer"] = "{}/{}/{}.html?page={}".format(
|
|
self.root, self.type, self.tag, offset // 100 + 1)
|
|
headers["Range"] = "bytes={}-{}".format(offset, offset+99)
|
|
response = self.request(nozomi_url, headers=headers)
|
|
|
|
for gallery_id in decode_nozomi(response.content):
|
|
gallery_url = "{}/galleries/{}.html".format(
|
|
self.root, gallery_id)
|
|
yield Message.Queue, gallery_url, data
|
|
|
|
offset += 100
|
|
if total is None:
|
|
total = text.parse_int(
|
|
response.headers["content-range"].rpartition("/")[2])
|
|
if offset >= total:
|
|
return
|
|
|
|
|
|
class HitomiIndexExtractor(HitomiTagExtractor):
|
|
"""Extractor for galleries from index searches on hitomi.la"""
|
|
subcategory = "index"
|
|
pattern = r"(?:https?://)?hitomi\.la/(\w+)-(\w+)\.html"
|
|
example = "https://hitomi.la/index-LANG.html"
|
|
|
|
def __init__(self, match):
|
|
Extractor.__init__(self, match)
|
|
self.tag, self.language = match.groups()
|
|
|
|
def items(self):
|
|
data = {"_extractor": HitomiGalleryExtractor}
|
|
nozomi_url = "https://ltn.hitomi.la/{}-{}.nozomi".format(
|
|
self.tag, self.language)
|
|
headers = {
|
|
"Origin": self.root,
|
|
"Cache-Control": "max-age=0",
|
|
}
|
|
|
|
offset = 0
|
|
total = None
|
|
while True:
|
|
headers["Referer"] = "{}/{}-{}.html?page={}".format(
|
|
self.root, self.tag, self.language, offset // 100 + 1)
|
|
headers["Range"] = "bytes={}-{}".format(offset, offset+99)
|
|
response = self.request(nozomi_url, headers=headers)
|
|
|
|
for gallery_id in decode_nozomi(response.content):
|
|
gallery_url = "{}/galleries/{}.html".format(
|
|
self.root, gallery_id)
|
|
yield Message.Queue, gallery_url, data
|
|
|
|
offset += 100
|
|
if total is None:
|
|
total = text.parse_int(
|
|
response.headers["content-range"].rpartition("/")[2])
|
|
if offset >= total:
|
|
return
|
|
|
|
|
|
class HitomiSearchExtractor(Extractor):
|
|
"""Extractor for galleries from multiple tag searches on hitomi.la"""
|
|
category = "hitomi"
|
|
subcategory = "search"
|
|
root = "https://hitomi.la"
|
|
pattern = r"(?:https?://)?hitomi\.la/search\.html\?([^/?#]+)"
|
|
example = "https://hitomi.la/search.html?QUERY"
|
|
|
|
def __init__(self, match):
|
|
Extractor.__init__(self, match)
|
|
self.query = match.group(1)
|
|
self.tags = text.unquote(self.query).split(" ")
|
|
|
|
def items(self):
|
|
data = {"_extractor": HitomiGalleryExtractor}
|
|
|
|
results = [self.get_nozomi_items(tag) for tag in self.tags]
|
|
intersects = set.intersection(*results)
|
|
|
|
for gallery_id in sorted(intersects, reverse=True):
|
|
gallery_url = "{}/galleries/{}.html".format(
|
|
self.root, gallery_id)
|
|
yield Message.Queue, gallery_url, data
|
|
|
|
def get_nozomi_items(self, full_tag):
|
|
area, tag, language = self.get_nozomi_args(full_tag)
|
|
|
|
if area:
|
|
referer_base = "{}/n/{}/{}-{}.html".format(
|
|
self.root, area, tag, language)
|
|
nozomi_url = "https://ltn.hitomi.la/{}/{}-{}.nozomi".format(
|
|
area, tag, language)
|
|
else:
|
|
referer_base = "{}/n/{}-{}.html".format(
|
|
self.root, tag, language)
|
|
nozomi_url = "https://ltn.hitomi.la/{}-{}.nozomi".format(
|
|
tag, language)
|
|
|
|
headers = {
|
|
"Origin": self.root,
|
|
"Cache-Control": "max-age=0",
|
|
"Referer": "{}/search.html?{}".format(referer_base, self.query),
|
|
}
|
|
|
|
response = self.request(nozomi_url, headers=headers)
|
|
return set(decode_nozomi(response.content))
|
|
|
|
def get_nozomi_args(self, query):
|
|
ns, _, tag = query.strip().partition(":")
|
|
area = ns
|
|
language = "all"
|
|
|
|
if ns == "female" or ns == "male":
|
|
area = "tag"
|
|
tag = query
|
|
elif ns == "language":
|
|
area = None
|
|
language = tag
|
|
tag = "index"
|
|
|
|
return area, tag, language
|
|
|
|
|
|
@memcache(maxage=1800)
|
|
def _parse_gg(extr):
|
|
page = extr.request("https://ltn.hitomi.la/gg.js").text
|
|
|
|
m = {}
|
|
|
|
keys = []
|
|
for match in re.finditer(
|
|
r"case\s+(\d+):(?:\s*o\s*=\s*(\d+))?", page):
|
|
key, value = match.groups()
|
|
keys.append(int(key))
|
|
|
|
if value:
|
|
value = int(value)
|
|
for key in keys:
|
|
m[key] = value
|
|
keys.clear()
|
|
|
|
for match in re.finditer(
|
|
r"if\s+\(g\s*===?\s*(\d+)\)[\s{]*o\s*=\s*(\d+)", page):
|
|
m[int(match.group(1))] = int(match.group(2))
|
|
|
|
d = re.search(r"(?:var\s|default:)\s*o\s*=\s*(\d+)", page)
|
|
b = re.search(r"b:\s*[\"'](.+)[\"']", page)
|
|
|
|
return m, b.group(1).strip("/"), int(d.group(1)) if d else 1
|