forked from Ponysearch/Ponysearch
[enh] Add engine for discourse forums
This commit is contained in:
parent
91882aedf1
commit
ee146dbc07
3 changed files with 164 additions and 0 deletions
8
docs/dev/engines/online/discourse.rst
Normal file
8
docs/dev/engines/online/discourse.rst
Normal file
|
@ -0,0 +1,8 @@
|
||||||
|
.. _discourse engine:
|
||||||
|
|
||||||
|
================
|
||||||
|
Discourse Forums
|
||||||
|
================
|
||||||
|
|
||||||
|
.. automodule:: searx.engines.discourse
|
||||||
|
:members:
|
153
searx/engines/discourse.py
Normal file
153
searx/engines/discourse.py
Normal file
|
@ -0,0 +1,153 @@
|
||||||
|
# SPDX-License-Identifier: AGPL-3.0-or-later
|
||||||
|
""".. sidebar:: info
|
||||||
|
|
||||||
|
- `builtwith.com Discourse <https://trends.builtwith.com/websitelist/Discourse>`_
|
||||||
|
|
||||||
|
Discourse is an open source Internet forum system. To search in a forum this
|
||||||
|
engine offers some additional settings:
|
||||||
|
|
||||||
|
- :py:obj:`base_url`
|
||||||
|
- :py:obj:`api_order`
|
||||||
|
- :py:obj:`search_endpoint`
|
||||||
|
- :py:obj:`show_avatar`
|
||||||
|
|
||||||
|
Example
|
||||||
|
=======
|
||||||
|
|
||||||
|
To search in your favorite Discourse forum, add a configuration like shown here
|
||||||
|
for the ``paddling.com`` forum:
|
||||||
|
|
||||||
|
.. code:: yaml
|
||||||
|
|
||||||
|
- name: paddling
|
||||||
|
engine: discourse
|
||||||
|
shortcut: paddle
|
||||||
|
base_url: 'https://forums.paddling.com/'
|
||||||
|
api_order: views
|
||||||
|
categories: ['social media', 'sports']
|
||||||
|
show_avatar: true
|
||||||
|
|
||||||
|
|
||||||
|
Implementations
|
||||||
|
===============
|
||||||
|
|
||||||
|
"""
|
||||||
|
|
||||||
|
from urllib.parse import urlencode
|
||||||
|
from datetime import datetime, timedelta
|
||||||
|
import html
|
||||||
|
|
||||||
|
from dateutil import parser
|
||||||
|
|
||||||
|
from flask_babel import gettext
|
||||||
|
|
||||||
|
about = {
|
||||||
|
"website": "https://discourse.org/",
|
||||||
|
"wikidata_id": "Q15054354",
|
||||||
|
"official_api_documentation": "https://docs.discourse.org/",
|
||||||
|
"use_official_api": True,
|
||||||
|
"require_api_key": False,
|
||||||
|
"results": "JSON",
|
||||||
|
}
|
||||||
|
|
||||||
|
base_url: str = None # type: ignore
|
||||||
|
"""URL of the Discourse forum."""
|
||||||
|
|
||||||
|
search_endpoint = '/search.json'
|
||||||
|
"""URL path of the `search endpoint`_.
|
||||||
|
|
||||||
|
.. _search endpoint: https://docs.discourse.org/#tag/Search
|
||||||
|
"""
|
||||||
|
|
||||||
|
api_order = 'likes'
|
||||||
|
"""Order method, valid values are: ``latest``, ``likes``, ``views``, ``latest_topic``"""
|
||||||
|
|
||||||
|
show_avatar = False
|
||||||
|
"""Show avatar of the user who send the post."""
|
||||||
|
|
||||||
|
paging = True
|
||||||
|
time_range_support = True
|
||||||
|
|
||||||
|
AGO_TIMEDELTA = {
|
||||||
|
'day': timedelta(days=1),
|
||||||
|
'week': timedelta(days=7),
|
||||||
|
'month': timedelta(days=31),
|
||||||
|
'year': timedelta(days=365),
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def request(query, params):
|
||||||
|
|
||||||
|
if len(query) <= 2:
|
||||||
|
return None
|
||||||
|
|
||||||
|
q = [query, f'order:{api_order}']
|
||||||
|
time_range = params.get('time_range')
|
||||||
|
if time_range:
|
||||||
|
after_date = datetime.now() - AGO_TIMEDELTA[time_range]
|
||||||
|
q.append('after:' + after_date.strftime('%Y-%m-%d'))
|
||||||
|
|
||||||
|
args = {
|
||||||
|
'q': ' '.join(q),
|
||||||
|
'page': params['pageno'],
|
||||||
|
}
|
||||||
|
|
||||||
|
params['url'] = f'{base_url}{search_endpoint}?{urlencode(args)}'
|
||||||
|
params['headers'] = {
|
||||||
|
'Accept': 'application/json, text/javascript, */*; q=0.01',
|
||||||
|
'X-Requested-With': 'XMLHttpRequest',
|
||||||
|
}
|
||||||
|
|
||||||
|
return params
|
||||||
|
|
||||||
|
|
||||||
|
def response(resp):
|
||||||
|
|
||||||
|
results = []
|
||||||
|
json_data = resp.json()
|
||||||
|
|
||||||
|
if ('topics' or 'posts') not in json_data.keys():
|
||||||
|
return []
|
||||||
|
|
||||||
|
topics = {}
|
||||||
|
|
||||||
|
for item in json_data['topics']:
|
||||||
|
topics[item['id']] = item
|
||||||
|
|
||||||
|
for post in json_data['posts']:
|
||||||
|
result = topics.get(post['topic_id'], {})
|
||||||
|
|
||||||
|
url = f"{base_url}/p/{post['id']}"
|
||||||
|
status = gettext("closed") if result.get('closed', '') else gettext("open")
|
||||||
|
comments = result.get('posts_count', 0)
|
||||||
|
publishedDate = parser.parse(result['created_at'])
|
||||||
|
|
||||||
|
metadata = []
|
||||||
|
metadata.append('@' + post.get('username', ''))
|
||||||
|
|
||||||
|
if int(comments) > 1:
|
||||||
|
metadata.append(f'{gettext("comments")}: {comments}')
|
||||||
|
|
||||||
|
if result.get('has_accepted_answer'):
|
||||||
|
metadata.append(gettext("answered"))
|
||||||
|
elif int(comments) > 1:
|
||||||
|
metadata.append(status)
|
||||||
|
|
||||||
|
result = {
|
||||||
|
'url': url,
|
||||||
|
'title': html.unescape(result['title']),
|
||||||
|
'content': html.unescape(post.get('blurb', '')),
|
||||||
|
'metadata': ' | '.join(metadata),
|
||||||
|
'publishedDate': publishedDate,
|
||||||
|
'upstream': {'topics': result},
|
||||||
|
}
|
||||||
|
|
||||||
|
avatar = post.get('avatar_template', '').replace('{size}', '96')
|
||||||
|
if show_avatar and avatar:
|
||||||
|
result['thumbnail'] = base_url + avatar
|
||||||
|
|
||||||
|
results.append(result)
|
||||||
|
|
||||||
|
results.append({'number_of_results': len(json_data['topics'])})
|
||||||
|
|
||||||
|
return results
|
|
@ -94,4 +94,7 @@ SOCIAL_MEDIA_TERMS = {
|
||||||
'POINTS': 'points',
|
'POINTS': 'points',
|
||||||
'TITLE': 'title',
|
'TITLE': 'title',
|
||||||
'AUTHOR': 'author',
|
'AUTHOR': 'author',
|
||||||
|
'THREAD OPEN': 'open',
|
||||||
|
'THREAD CLOSED': 'closed',
|
||||||
|
'THREAD ANSWERED': 'answered',
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue