forked from Ponysearch/Ponysearch
72 lines
1.8 KiB
Python
72 lines
1.8 KiB
Python
|
# SPDX-License-Identifier: AGPL-3.0-or-later
|
||
|
|
"""Pixiv (images)"""
|
||
|
|
|
||
|
|
from urllib.parse import urlencode
|
||
|
|
import random
|
||
|
|
|
||
|
|
# Engine metadata
|
||
|
|
about = {
|
||
|
|
"website": 'https://www.pixiv.net/',
|
||
|
|
"wikidata_id": 'Q306956',
|
||
|
|
"official_api_documentation": None,
|
||
|
|
"use_official_api": False,
|
||
|
|
"require_api_key": False,
|
||
|
|
"results": 'JSON',
|
||
|
|
}
|
||
|
|
|
||
|
|
# Engine configuration
|
||
|
|
paging = True
|
||
|
|
categories = ['images']
|
||
|
|
|
||
|
|
# Search URL
|
||
|
|
base_url = "https://www.pixiv.net/ajax/search/illustrations"
|
||
|
|
pixiv_image_proxies: list = []
|
||
|
|
|
||
|
|
|
||
|
|
def request(query, params):
|
||
|
|
query_params = {
|
||
|
|
"word": query,
|
||
|
|
"order": "date_d",
|
||
|
|
"mode": "all",
|
||
|
|
"p": params["pageno"],
|
||
|
|
"s_mode": "s_tag_full",
|
||
|
|
"type": "illust_and_ugoira",
|
||
|
|
"lang": "en",
|
||
|
|
}
|
||
|
|
|
||
|
|
params["url"] = f"{base_url}/{query}?{urlencode(query_params)}"
|
||
|
|
|
||
|
|
return params
|
||
|
|
|
||
|
|
|
||
|
|
def response(resp):
|
||
|
|
results = []
|
||
|
|
data = resp.json()
|
||
|
|
|
||
|
|
for item in data["body"]["illust"]["data"]:
|
||
|
|
|
||
|
|
image_url = item["url"]
|
||
|
|
pixiv_proxy = random.choice(pixiv_image_proxies)
|
||
|
|
proxy_image_url = image_url.replace("https://i.pximg.net", pixiv_proxy)
|
||
|
|
proxy_full_image_url = (
|
||
|
|
proxy_image_url.replace("/c/250x250_80_a2/", "/")
|
||
|
|
.replace("_square1200.jpg", "_master1200.jpg")
|
||
|
|
.replace("custom-thumb", "img-master")
|
||
|
|
.replace("_custom1200.jpg", "_master1200.jpg")
|
||
|
|
)
|
||
|
|
|
||
|
|
results.append(
|
||
|
|
{
|
||
|
|
"title": item.get("title"),
|
||
|
|
"url": proxy_full_image_url,
|
||
|
|
'content': item.get('alt'),
|
||
|
|
"author": f"{item.get('userName')} (ID: {item.get('userId')})",
|
||
|
|
"img_src": proxy_full_image_url,
|
||
|
|
"thumbnail_src": proxy_image_url,
|
||
|
|
"source": 'pixiv.net',
|
||
|
|
"template": "images.html",
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
return results
|