github/gallery-dl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[imgbb] Fix user extraction, add displayname

Browse Source
This commit is contained in:
HRXN
2023-10-05 12:18:58 +02:00
committed by GitHub
parent 85357c1ef8
commit b846f56c3a
1 changed files with 14 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
gallery_dl/extractor/imgbb.py
View File

@@ -124,12 +124,14 @@ class ImgbbAlbumExtractor(ImgbbExtractor):
self.page_url = "https://ibb.co/album/" + self.album_id self.page_url = "https://ibb.co/album/" + self.album_id
def metadata(self, page): def metadata(self, page):
album, pos = text.extract(page, '"og:title" content="', '"') album , pos = text.extract(page, '"og:title" content="', '"')
user , pos = text.extract(page, 'rel="author">', '<', pos) user , pos = text.extract(page, ',"username":"', '"', pos)
displayname, pos = text.extract(page, '"user":{"name":"', '"')
return { return {
"album_id" : self.album_id, "album_id" : self.album_id,
"album_name": text.unescape(album), "album_name" : text.unescape(album),
"user" : user.lower() if user else "", "user" : user.lower() if user else "",
"displayname": displayname if displayname else "",
} }
def images(self, page): def images(self, page):
@@ -158,7 +160,11 @@ class ImgbbUserExtractor(ImgbbExtractor):
self.page_url = "https://{}.imgbb.com/".format(self.user) self.page_url = "https://{}.imgbb.com/".format(self.user)
def metadata(self, page): def metadata(self, page):
return {"user": self.user} displayname, pos = text.extract(page, '"user":{"name":"', '"')
return {
"user" : self.user,
"displayname": displayname if displayname else "",
}
def images(self, page): def images(self, page):
user = text.extr(page, '.obj.resource={"id":"', '"') user = text.extr(page, '.obj.resource={"id":"', '"')
@@ -185,11 +191,11 @@ class ImgbbImageExtractor(ImgbbExtractor):
image = { image = {
"id" : self.image_id, "id" : self.image_id,
"title" : text.unescape(extr('"og:title" content="', '"')), "title" : text.unescape(extr('"og:title" content="', ' hosted at ImgBB"')),
"url" : extr('"og:image" content="', '"'), "url" : extr('"og:image" content="', '"'),
"width" : text.parse_int(extr('"og:image:width" content="', '"')), "width" : text.parse_int(extr('"og:image:width" content="', '"')),
"height": text.parse_int(extr('"og:image:height" content="', '"')), "height": text.parse_int(extr('"og:image:height" content="', '"')),
"user" : extr('rel="author">', '<').lower(), "user" : extr(',"username":"', '"').lower(),
} }
image["extension"] = text.ext_from_url(image["url"]) image["extension"] = text.ext_from_url(image["url"])