summaryrefslogtreecommitdiff
path: root/searx/plugins/self_info.py
diff options
context:
space:
mode:
Diffstat (limited to 'searx/plugins/self_info.py')
-rw-r--r--searx/plugins/self_info.py75
1 files changed, 50 insertions, 25 deletions
diff --git a/searx/plugins/self_info.py b/searx/plugins/self_info.py
index 7cad040d2..d1d7f12b9 100644
--- a/searx/plugins/self_info.py
+++ b/searx/plugins/self_info.py
@@ -1,32 +1,57 @@
# SPDX-License-Identifier: AGPL-3.0-or-later
-# pylint: disable=missing-module-docstring,invalid-name
+# pylint: disable=missing-module-docstring, missing-class-docstring
+from __future__ import annotations
+import typing
import re
from flask_babel import gettext
from searx.botdetection._helpers import get_real_ip
+from searx.result_types import Answer
-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
+from . import Plugin, PluginInfo
+
+if typing.TYPE_CHECKING:
+ from searx.search import SearchWithPlugins
+ from searx.extended_types import SXNG_Request
+
+
+class SXNGPlugin(Plugin):
+ """Simple plugin that displays information about user's request, including
+ the IP or HTTP User-Agent. The information is displayed in area for the
+ "answers".
+ """
+
+ id = "self_info"
+ default_on = True
+ keywords = ["ip", "user-agent"]
+
+ def __init__(self):
+ super().__init__()
+
+ self.ip_regex = re.compile(r"^ip", re.IGNORECASE)
+ self.ua_regex = re.compile(r"^user-agent", re.IGNORECASE)
+
+ self.info = PluginInfo(
+ id=self.id,
+ name=gettext("Self Information"),
+ description=gettext(
+ """Displays your IP if the query is "ip" and your user agent if the query is "user-agent"."""
+ ),
+ preference_section="query",
+ )
+
+ def post_search(self, request: "SXNG_Request", search: "SearchWithPlugins") -> list[Answer]:
+ """Returns a result list only for the first page."""
+ results = []
+
+ if search.search_query.pageno > 1:
+ return results
+
+ if self.ip_regex.search(search.search_query.query):
+ Answer(results=results, answer=gettext("Your IP is: ") + get_real_ip(request))
+
+ if self.ua_regex.match(search.search_query.query):
+ Answer(results=results, answer=gettext("Your user-agent is: ") + str(request.user_agent))
+
+ return results