[feat] engine: implementation of wallhaven

This commit is contained in:
Bnyro 2023-09-17 20:14:05 +02:00 committed by MatthieuBarbu
parent 8bea311d5a
commit 03123675a0
3 changed files with 102 additions and 0 deletions

View File

@ -0,0 +1,13 @@
.. _wallhaven engine:
=========
Wallhaven
=========
.. contents:: Contents
:depth: 2
:local:
:backlinks: entry
.. automodule:: searx.engines.wallhaven
:members:

View File

@ -0,0 +1,84 @@
# SPDX-License-Identifier: AGPL-3.0-or-later
# lint: pylint
"""Wallhaven_ is a site created by and for people who like wallpapers.
.. _Wallhaven: https://wallhaven.cc/about#Copyright
"""
from datetime import datetime
from urllib.parse import urlencode
about = {
'website': 'https://wallhaven.cc/',
'official_api_documentation': 'https://wallhaven.cc/help/api',
'use_official_api': True,
'require_api_key': False,
'results': 'JSON',
}
categories = ['images']
paging = True
base_url = "https://wallhaven.cc"
api_key = ''
"""If you own an API key you can add it here, further read `Rate Limiting and
Errors`_.
.. _Rate Limiting and Errors: https://wallhaven.cc/help/api#limits
"""
# Possible categories: sfw, sketchy, nsfw
safesearch_map = {0: '111', 1: '110', 2: '100'}
"""Turn purities on(1) or off(0) NSFW requires a valid API key.
.. code:: text
100/110/111 <-- Bits stands for: SFW, Sketchy and NSFW
`What are SFW, Sketchy and NSFW all about?`_:
- SFW = "Safe for work" wallpapers. *Grandma approves.*
- Sketchy = Not quite SFW not quite NSFW. *Grandma might be uncomfortable.*
- NSFW = "Not safe for work". *Grandma isn't sure who you are anymore.*
.. _What are SFW, Sketchy and NSFW all about?:
https://wallhaven.cc/faq#What-are-SFW-Sketchy-and-NSFW-all-about
"""
def request(query, params):
args = {
'q': query,
'page': params['pageno'],
'purity': safesearch_map[params['safesearch']],
}
if api_key:
params['api_key'] = api_key
params['url'] = f"{base_url}/api/v1/search?{urlencode(args)}"
return params
def response(resp):
results = []
json = resp.json()
for result in json['data']:
results.append(
{
'template': 'images.html',
'title': '',
'content': f"{result['category']} / {result['purity']}",
'url': result['url'],
'img_src': result['path'],
'thumbnail_src': result['thumbs']['small'],
'img_format': result['resolution'],
'publishedDate': datetime.strptime(result['created_at'], '%Y-%m-%d %H:%M:%S'),
}
)
return results

View File

@ -1908,6 +1908,11 @@ engines:
timeout: 10.0 timeout: 10.0
disabled: true disabled: true
- name: wallhaven
engine: wallhaven
# api_key: abcdefghijklmnopqrstuvwxyz
shortcut: wh
# wikimini: online encyclopedia for children # wikimini: online encyclopedia for children
# The fulltext and title parameter is necessary for Wikimini because # The fulltext and title parameter is necessary for Wikimini because
# sometimes it will not show the results and redirect instead # sometimes it will not show the results and redirect instead