1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
# -*- 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 LolisafelbumExtractor(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://cdn\.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":
data["_http_validate"] = self._check_rewrite
else:
data["_http_validate"] = None
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"],
}
@staticmethod
def _check_rewrite(response):
if response.history and response.headers.get(
"Content-Type").startswith("text/html"):
# consume content to reuse connection
response.content
# rewrite to download URL
return response.url.replace("/v/", "/d/", 1)
return True
|
