mirror of
https://github.com/mikf/gallery-dl.git
synced 2024-11-24 03:32:33 +01:00
233 lines
8.0 KiB
Python
233 lines
8.0 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, util, exception
|
|
from ..cache import cache
|
|
|
|
|
|
class ImgbbExtractor(Extractor):
|
|
"""Base class for imgbb extractors"""
|
|
category = "imgbb"
|
|
directory_fmt = ("{category}", "{user}")
|
|
filename_fmt = "{title} {id}.{extension}"
|
|
archive_fmt = "{id}"
|
|
root = "https://imgbb.com"
|
|
|
|
def __init__(self, match):
|
|
Extractor.__init__(self, match)
|
|
self.page_url = self.sort = None
|
|
|
|
def items(self):
|
|
self.login()
|
|
|
|
url = self.page_url
|
|
params = {"sort": self.sort}
|
|
while True:
|
|
response = self.request(url, params=params, allow_redirects=False)
|
|
if response.status_code < 300:
|
|
break
|
|
url = response.headers["location"]
|
|
if url.startswith(self.root):
|
|
raise exception.NotFoundError(self.subcategory)
|
|
|
|
page = response.text
|
|
data = self.metadata(page)
|
|
first = True
|
|
|
|
for img in self.images(page):
|
|
image = {
|
|
"id" : img["url_viewer"].rpartition("/")[2],
|
|
"user" : img["user"]["username"] if "user" in img else "",
|
|
"title" : text.unescape(img["title"]),
|
|
"url" : img["image"]["url"],
|
|
"extension": img["image"]["extension"],
|
|
"size" : text.parse_int(img["image"]["size"]),
|
|
"width" : text.parse_int(img["width"]),
|
|
"height" : text.parse_int(img["height"]),
|
|
}
|
|
image.update(data)
|
|
if first:
|
|
first = False
|
|
yield Message.Directory, data
|
|
yield Message.Url, image["url"], image
|
|
|
|
def login(self):
|
|
username, password = self._get_auth_info()
|
|
if username:
|
|
self._update_cookies(self._login_impl(username, password))
|
|
|
|
@cache(maxage=360*24*3600, keyarg=1)
|
|
def _login_impl(self, username, password):
|
|
self.log.info("Logging in as %s", username)
|
|
|
|
url = self.root + "/login"
|
|
page = self.request(url).text
|
|
token = text.extr(page, 'PF.obj.config.auth_token="', '"')
|
|
|
|
headers = {"Referer": url}
|
|
data = {
|
|
"auth_token" : token,
|
|
"login-subject": username,
|
|
"password" : password,
|
|
}
|
|
response = self.request(url, method="POST", headers=headers, data=data)
|
|
|
|
if not response.history:
|
|
raise exception.AuthenticationError()
|
|
return self.session.cookies
|
|
|
|
def _pagination(self, page, endpoint, params):
|
|
data = None
|
|
seek, pos = text.extract(page, 'data-seek="', '"')
|
|
tokn, pos = text.extract(page, 'PF.obj.config.auth_token="', '"', pos)
|
|
params["action"] = "list"
|
|
params["list"] = "images"
|
|
params["sort"] = self.sort
|
|
params["seek"] = seek
|
|
params["page"] = 2
|
|
params["auth_token"] = tokn
|
|
|
|
while True:
|
|
for img in text.extract_iter(page, "data-object='", "'"):
|
|
yield util.json_loads(text.unquote(img))
|
|
if data:
|
|
if params["seek"] == data["seekEnd"]:
|
|
return
|
|
params["seek"] = data["seekEnd"]
|
|
params["page"] += 1
|
|
elif not seek or 'class="pagination-next"' not in page:
|
|
return
|
|
data = self.request(endpoint, method="POST", data=params).json()
|
|
page = data["html"]
|
|
|
|
|
|
class ImgbbAlbumExtractor(ImgbbExtractor):
|
|
"""Extractor for albums on imgbb.com"""
|
|
subcategory = "album"
|
|
directory_fmt = ("{category}", "{user}", "{album_name} {album_id}")
|
|
pattern = r"(?:https?://)?ibb\.co/album/([^/?#]+)/?(?:\?([^#]+))?"
|
|
test = (
|
|
("https://ibb.co/album/i5PggF", {
|
|
"range": "1-80",
|
|
"url": "70afec9fcc3a6de62a6b644b487d892d8d47cf1a",
|
|
"keyword": "569e1d88ebdd27655387559cdf1cd526a3e1ab69",
|
|
}),
|
|
("https://ibb.co/album/i5PggF?sort=title_asc", {
|
|
"range": "1-80",
|
|
"url": "afdf5fc95d8e09d77e8f44312f3e9b843987bb5a",
|
|
"keyword": "f090e14d0e5f7868595082b2c95da1309c84872d",
|
|
}),
|
|
# no user data (#471)
|
|
("https://ibb.co/album/kYKpwF", {
|
|
"url": "ac0abcfcb89f4df6adc2f7e4ff872f3b03ef1bc7",
|
|
"keyword": {"user": ""},
|
|
}),
|
|
# private
|
|
("https://ibb.co/album/hqgWrF", {
|
|
"exception": exception.HttpError,
|
|
}),
|
|
)
|
|
|
|
def __init__(self, match):
|
|
ImgbbExtractor.__init__(self, match)
|
|
self.album_name = None
|
|
self.album_id = match.group(1)
|
|
self.sort = text.parse_query(match.group(2)).get("sort", "date_desc")
|
|
self.page_url = "https://ibb.co/album/" + self.album_id
|
|
|
|
def metadata(self, page):
|
|
album, pos = text.extract(page, '"og:title" content="', '"')
|
|
user , pos = text.extract(page, 'rel="author">', '<', pos)
|
|
return {
|
|
"album_id" : self.album_id,
|
|
"album_name": text.unescape(album),
|
|
"user" : user.lower() if user else "",
|
|
}
|
|
|
|
def images(self, page):
|
|
url = text.extr(page, '"og:url" content="', '"')
|
|
album_id = url.rpartition("/")[2].partition("?")[0]
|
|
|
|
return self._pagination(page, "https://ibb.co/json", {
|
|
"from" : "album",
|
|
"albumid" : album_id,
|
|
"params_hidden[list]" : "images",
|
|
"params_hidden[from]" : "album",
|
|
"params_hidden[albumid]": album_id,
|
|
})
|
|
|
|
|
|
class ImgbbUserExtractor(ImgbbExtractor):
|
|
"""Extractor for user profiles in imgbb.com"""
|
|
subcategory = "user"
|
|
pattern = r"(?:https?://)?([\w-]+)\.imgbb\.com/?(?:\?([^#]+))?$"
|
|
test = ("https://folkie.imgbb.com", {
|
|
"range": "1-80",
|
|
"pattern": r"https?://i\.ibb\.co/\w+/[^/?#]+",
|
|
})
|
|
|
|
def __init__(self, match):
|
|
ImgbbExtractor.__init__(self, match)
|
|
self.user = match.group(1)
|
|
self.sort = text.parse_query(match.group(2)).get("sort", "date_desc")
|
|
self.page_url = "https://{}.imgbb.com/".format(self.user)
|
|
|
|
def metadata(self, page):
|
|
return {"user": self.user}
|
|
|
|
def images(self, page):
|
|
user = text.extr(page, '.obj.resource={"id":"', '"')
|
|
return self._pagination(page, self.page_url + "json", {
|
|
"from" : "user",
|
|
"userid" : user,
|
|
"params_hidden[userid]": user,
|
|
"params_hidden[from]" : "user",
|
|
})
|
|
|
|
|
|
class ImgbbImageExtractor(ImgbbExtractor):
|
|
subcategory = "image"
|
|
pattern = r"(?:https?://)?ibb\.co/(?!album/)([^/?#]+)"
|
|
test = ("https://ibb.co/fUqh5b", {
|
|
"pattern": r"https://i\.ibb\.co/g3kvx80/Arundel-Ireeman-5\.jpg",
|
|
"content": "c5a0965178a8b357acd8aa39660092918c63795e",
|
|
"keyword": {
|
|
"id" : "fUqh5b",
|
|
"title" : "Arundel Ireeman 5",
|
|
"url" : "https://i.ibb.co/g3kvx80/Arundel-Ireeman-5.jpg",
|
|
"width" : 960,
|
|
"height": 719,
|
|
"user" : "folkie",
|
|
"extension": "jpg",
|
|
},
|
|
})
|
|
|
|
def __init__(self, match):
|
|
ImgbbExtractor.__init__(self, match)
|
|
self.image_id = match.group(1)
|
|
|
|
def items(self):
|
|
url = "https://ibb.co/" + self.image_id
|
|
extr = text.extract_from(self.request(url).text)
|
|
|
|
image = {
|
|
"id" : self.image_id,
|
|
"title" : text.unescape(extr('"og:title" content="', '"')),
|
|
"url" : extr('"og:image" content="', '"'),
|
|
"width" : text.parse_int(extr('"og:image:width" content="', '"')),
|
|
"height": text.parse_int(extr('"og:image:height" content="', '"')),
|
|
"user" : extr('rel="author">', '<').lower(),
|
|
}
|
|
image["extension"] = text.ext_from_url(image["url"])
|
|
|
|
yield Message.Directory, image
|
|
yield Message.Url, image["url"], image
|