86 lines
2.7 KiB
Python
86 lines
2.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright 2019 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://imgbb.com/"""
|
|
|
|
from .common import Extractor, Message
|
|
from .. import text
|
|
import json
|
|
|
|
|
|
class ImgbbAlbumExtractor(Extractor):
|
|
"""Extractor for albums on imgbb.com"""
|
|
category = "imgbb"
|
|
subcategory = "album"
|
|
root = "https://imgbb.com"
|
|
directory_fmt = ("{category}", "{user[username]}",
|
|
"{album_id} {album_name}")
|
|
filename_fmt = "{id}{title:?_//}.{extension}"
|
|
archive_fmt = "{id}"
|
|
pattern = r"(?:https?://)?ibb\.co/album/([^/?&#]+)(?:\?([^#]+))?"
|
|
test = ("https://ibb.co/album/c6p5Yv", {
|
|
"range": "1-100",
|
|
"url": "",
|
|
"keyword": "",
|
|
})
|
|
|
|
def __init__(self, match):
|
|
Extractor.__init__(self, match)
|
|
self.album_id = match.group(1)
|
|
self.album_name = None
|
|
self.params = text.parse_query(match.group(2))
|
|
|
|
def items(self):
|
|
first = True
|
|
yield Message.Version, 1
|
|
|
|
for img in self.images():
|
|
url = img["image"]["url"]
|
|
img["album_id"] = self.album_id
|
|
img["album_name"] = self.album_name
|
|
img["id"] = img["url_viewer"].rpartition("/")[2]
|
|
if first:
|
|
first = False
|
|
yield Message.Directory, img
|
|
yield Message.Url, url, img
|
|
|
|
def images(self):
|
|
url = "https://ibb.co/album/" + self.album_id
|
|
page = self.request(url).text
|
|
|
|
self.album_name, pos = text.extract(page, '"og:title" content="', '"')
|
|
seek, pos = text.extract(page, 'data-seek="', '"', pos)
|
|
tokn, pos = text.extract(page, 'PF.obj.config.auth_token="', '"', pos)
|
|
|
|
endpoint = "https://ibb.co/json"
|
|
data = None
|
|
params = {
|
|
"action" : "list",
|
|
"list" : "images",
|
|
"from" : "album",
|
|
"sort" : "date_desc",
|
|
"page" : 2,
|
|
"albumid": self.album_id,
|
|
"params_hidden[list]" : "images",
|
|
"params_hidden[from]" : "album",
|
|
"params_hidden[albumid]": self.album_id,
|
|
"seek" : seek,
|
|
"auth_token": tokn,
|
|
}
|
|
|
|
while True:
|
|
for img in text.extract_iter(page, "data-object='", "'"):
|
|
yield json.loads(text.unquote(img))
|
|
if data:
|
|
if params["seek"] == data["seekEnd"]:
|
|
return
|
|
params["seek"] = data["seekEnd"]
|
|
params["page"] += 1
|
|
data = self.request(endpoint, "POST", data=params).json()
|
|
page = data["html"]
|