forked from Ponysearch/Ponysearch
e86c96974d
The user_agent attribute of the Flask request object is an instance of
the werkzeug.user_agent.UserAgent class.
This will fix the following error of the self_info plugin:
> ERROR:searx.plugins.self_info: Exception while calling post_search
> Traceback (most recent call last):
> File "searx/plugins/__init__.py", line 203, in call
> ret = getattr(plugin, plugin_type)(*args, **kwargs)
> File "searx/plugins/self_info.py", line 31, in post_search
> search.result_container.answers['user-agent'] = {'answer': gettext('Your user-agent is: ') + ua}
> TypeError: can only concatenate str (not "UserAgent") to str
32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
# SPDX-License-Identifier: AGPL-3.0-or-later
|
|
# pylint: disable=missing-module-docstring,invalid-name
|
|
|
|
import re
|
|
from flask_babel import gettext
|
|
|
|
from searx.botdetection._helpers import get_real_ip
|
|
|
|
name = gettext('Self Information')
|
|
description = gettext('Displays your IP if the query is "ip" and your user agent if the query contains "user agent".')
|
|
default_on = True
|
|
preference_section = 'query'
|
|
query_keywords = ['user-agent']
|
|
query_examples = ''
|
|
|
|
# "ip" or "my ip" regex
|
|
ip_regex = re.compile('^ip$|my ip', re.IGNORECASE)
|
|
|
|
# Self User Agent regex
|
|
ua_regex = re.compile('.*user[ -]agent.*', re.IGNORECASE)
|
|
|
|
|
|
def post_search(request, search):
|
|
if search.search_query.pageno > 1:
|
|
return True
|
|
if ip_regex.search(search.search_query.query):
|
|
ip = get_real_ip(request)
|
|
search.result_container.answers['ip'] = {'answer': gettext('Your IP is: ') + ip}
|
|
elif ua_regex.match(search.search_query.query):
|
|
ua = request.user_agent
|
|
search.result_container.answers['user-agent'] = {'answer': gettext('Your user-agent is: ') + ua.string}
|
|
return True
|