mirror of
https://github.com/mikf/gallery-dl.git
synced 2024-11-23 19:22:32 +01:00
89 lines
2.9 KiB
Python
89 lines
2.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright 2021-2022 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 lolisafe/chibisafe instances"""
|
|
|
|
from .common import BaseExtractor, Message
|
|
from .. import text
|
|
|
|
|
|
class LolisafeExtractor(BaseExtractor):
|
|
"""Base class for lolisafe extractors"""
|
|
basecategory = "lolisafe"
|
|
directory_fmt = ("{category}", "{album_name} ({album_id})")
|
|
archive_fmt = "{album_id}_{id}"
|
|
|
|
|
|
BASE_PATTERN = LolisafeExtractor.update({
|
|
"bunkr": {"root": "https://bunkr.is", "pattern": r"bunkr\.(?:is|to)"},
|
|
"zzzz" : {"root": "https://zz.ht" , "pattern": r"zz\.(?:ht|fo)"},
|
|
})
|
|
|
|
|
|
class LolisafeAlbumExtractor(LolisafeExtractor):
|
|
subcategory = "album"
|
|
pattern = BASE_PATTERN + "/a/([^/?#]+)"
|
|
test = (
|
|
("https://bunkr.is/a/Lktg9Keq", {
|
|
"pattern": r"https://cdn\.bunkr\.is/test-テスト-\"&>-QjgneIQv\.png",
|
|
"content": "0c8768055e4e20e7c7259608b67799171b691140",
|
|
"keyword": {
|
|
"album_id": "Lktg9Keq",
|
|
"album_name": 'test テスト "&>',
|
|
"count": 1,
|
|
"filename": 'test-テスト-"&>-QjgneIQv',
|
|
"id": "QjgneIQv",
|
|
"name": 'test-テスト-"&>',
|
|
"num": int,
|
|
},
|
|
}),
|
|
# mp4 (#2239)
|
|
("https://bunkr.is/a/ptRHaCn2", {
|
|
"pattern": r"https://media-files\.bunkr\.is/_-RnHoW69L\.mp4",
|
|
"content": "80e61d1dbc5896ae7ef9a28734c747b28b320471",
|
|
}),
|
|
("https://bunkr.to/a/Lktg9Keq"),
|
|
("https://zz.ht/a/lop7W6EZ", {
|
|
"pattern": r"https://z\.zz\.fo/(4anuY|ih560)\.png",
|
|
"count": 2,
|
|
"keyword": {
|
|
"album_id": "lop7W6EZ",
|
|
"album_name": "ferris",
|
|
},
|
|
}),
|
|
("https://zz.fo/a/lop7W6EZ"),
|
|
)
|
|
|
|
def __init__(self, match):
|
|
LolisafeExtractor.__init__(self, match)
|
|
self.album_id = match.group(match.lastindex)
|
|
|
|
def items(self):
|
|
files, data = self.fetch_album(self.album_id)
|
|
|
|
yield Message.Directory, data
|
|
for data["num"], file in enumerate(files, 1):
|
|
url = file["file"]
|
|
text.nameext_from_url(url, data)
|
|
data["name"], sep, data["id"] = data["filename"].rpartition("-")
|
|
|
|
if data["extension"] == "mp4":
|
|
url = url.replace(
|
|
"//cdn.bunkr.is/", "//media-files.bunkr.is/", 1)
|
|
yield Message.Url, url, data
|
|
|
|
def fetch_album(self, album_id):
|
|
url = "{}/api/album/get/{}".format(self.root, album_id)
|
|
data = self.request(url).json()
|
|
|
|
return data["files"], {
|
|
"album_id" : self.album_id,
|
|
"album_name": text.unescape(data["title"]),
|
|
"count" : data["count"],
|
|
}
|