48 lines
1.2 KiB
Python
48 lines
1.2 KiB
Python
|
import requests
|
||
|
|
import urllib.parse
|
||
|
|
from . import utils
|
||
|
|
|
||
|
|
commons_start = "http://commons.wikimedia.org/wiki/Special:FilePath/"
|
||
|
|
commons_url = "https://www.wikidata.org/w/api.php"
|
||
|
|
page_size = 50
|
||
|
|
|
||
|
|
|
||
|
|
def commons_uri_to_filename(uri):
|
||
|
|
return urllib.parse.unquote(utils.drop_start(uri, commons_start))
|
||
|
|
|
||
|
|
|
||
|
|
def api_call(params):
|
||
|
|
call_params = {
|
||
|
|
"format": "json",
|
||
|
|
"formatversion": 2,
|
||
|
|
**params,
|
||
|
|
}
|
||
|
|
|
||
|
|
return requests.get(commons_url, params=call_params, timeout=5)
|
||
|
|
|
||
|
|
|
||
|
|
def image_detail(filenames, thumbheight=None, thumbwidth=None):
|
||
|
|
params = {
|
||
|
|
"action": "query",
|
||
|
|
"prop": "imageinfo",
|
||
|
|
"iiprop": "url",
|
||
|
|
}
|
||
|
|
if thumbheight is not None:
|
||
|
|
params["iiurlheight"] = thumbheight
|
||
|
|
if thumbwidth is not None:
|
||
|
|
params["iiurlwidth"] = thumbwidth
|
||
|
|
|
||
|
|
images = {}
|
||
|
|
|
||
|
|
for cur in utils.chunk(filenames, page_size):
|
||
|
|
call_params = params.copy()
|
||
|
|
call_params["titles"] = "|".join(f"File:{f}" for f in cur)
|
||
|
|
|
||
|
|
r = api_call(call_params)
|
||
|
|
|
||
|
|
for image in r.json()["query"]["pages"]:
|
||
|
|
filename = utils.drop_start(image["title"], "File:")
|
||
|
|
images[filename] = image["imageinfo"][0] if "imageinfo" in image else None
|
||
|
|
|
||
|
|
return images
|