1
0
mirror of https://github.com/mikf/gallery-dl.git synced 2024-11-25 12:12:34 +01:00

[flickr] add search extractor

This commit is contained in:
Mike Fährmann 2017-06-13 08:01:32 +02:00
parent 5f55c854b9
commit 05ed95e5b0
No known key found for this signature in database
GPG Key ID: 5680CA389D365A88
3 changed files with 34 additions and 2 deletions

View File

@ -11,6 +11,7 @@
from .common import Extractor, Message
from .. import text, util, exception
from . import oauth
import urllib.parse
class FlickrExtractor(Extractor):
@ -173,7 +174,7 @@ class FlickrGroupExtractor(FlickrExtractor):
pattern = [r"(?:https?://)?(?:www\.)?flickr\.com/groups/([^/]+)"]
test = [("https://www.flickr.com/groups/bird_headshots/", {
"url": "40b5586fa0cd1578c3b8cc874fc6e3ae7af70786",
"keyword": "24e721cc510c1f74bc3d8f7bd6130773ba3faef4",
"keyword": "cd056422203e0a6cae93cd866a6ee0920cebb3be",
})]
def data(self):
@ -212,6 +213,32 @@ class FlickrFavoriteExtractor(FlickrExtractor):
return self.api.favorites_getList(self.user["nsid"])
class FlickrSearchExtractor(FlickrExtractor):
subcategory = "search"
directory_fmt = ["{category}", "{subcategory}", "{search[text]}"]
pattern = [r"(?:https?://)?(?:www\.)?flickr\.com/search/?\?([^#]+)"]
test = [(("https://flickr.com/search/?text=tree%20cloud%20house"
"&color_codes=4&styles=minimalism"), {
"url": "ba7ca5a0da857775f68ab63ff5653bf4b16f2ceb",
"keyword": "20cd3b153f52ec88e949d1167439583461a515b3",
})]
def __init__(self, match):
FlickrExtractor.__init__(self, match)
self.search = {
key: vlist[0]
for key, vlist in urllib.parse.parse_qs(match.group(1)).items()
}
if "text" not in self.search:
self.search["text"] = ""
def data(self):
return {"search": self.search}
def photos(self):
return self.api.photos_search(self.search)
class FlickrAPI():
"""Minimal interface for the flickr API"""
API_URL = "https://api.flickr.com/services/rest/"
@ -272,6 +299,10 @@ class FlickrAPI():
params = {"photo_id": photo_id}
return self._call("photos.getSizes", params)["sizes"]["size"]
def photos_search(self, params):
"""Return a list of photos matching some criteria."""
return self._listing("photos.search", params.copy())
def photosets_getPhotos(self, photoset_id):
"""Get the list of photos in a set."""
params = {"photoset_id": photoset_id}

View File

@ -45,7 +45,7 @@ class OAuthSession():
message = self.concat("GET", url, query).encode()
key = self.concat(self.consumer_secret, self.token_secret).encode()
signature = hmac.new(key, message, hashlib.sha1).digest()
return base64.b64encode(signature)
return base64.b64encode(signature).decode()
@staticmethod
def concat(*args):

View File

@ -51,6 +51,7 @@ skip = [
# dont work on travis-ci
"exhentai", "kissmanga", "mangafox", "dynastyscans", "nijie",
# temporary issues
"gomanga", "mangapark",
]
# enable selective testing for direct calls
if __name__ == '__main__' and len(sys.argv) > 1: