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
|
# -*- coding: utf-8 -*-
# Copyright 2021-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 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({
})
class LolisafeAlbumExtractor(LolisafeExtractor):
subcategory = "album"
pattern = BASE_PATTERN + "/a/([^/?#]+)"
example = "https://xbunkr.com/a/ID"
def __init__(self, match):
LolisafeExtractor.__init__(self, match)
self.album_id = self.groups[-1]
def _init(self):
domain = self.config("domain")
if domain == "auto":
self.root = text.root_from_url(self.url)
elif domain:
self.root = text.ensure_http_scheme(domain)
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"]
file.update(data)
if "extension" not in file:
text.nameext_from_url(url, file)
if "name" in file:
name = file["name"]
file["name"] = name.rpartition(".")[0] or name
file["id"] = file["filename"].rpartition("-")[2]
elif "id" in file:
file["name"] = file["filename"]
file["filename"] = "{}-{}".format(file["name"], file["id"])
else:
file["name"], sep, file["id"] = \
file["filename"].rpartition("-")
yield Message.Url, url, file
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"],
}
|
