self_info.py 940 B

123456789101112131415161718192021222324252627282930
  1. # SPDX-License-Identifier: AGPL-3.0-or-later
  2. # lint: pylint
  3. # pylint: disable=missing-module-docstring,invalid-name
  4. import re
  5. from flask_babel import gettext
  6. from searx.botdetection._helpers import get_real_ip
  7. name = gettext('Self Information')
  8. description = gettext('Displays your IP if the query is "ip" and your user agent if the query contains "user agent".')
  9. default_on = True
  10. preference_section = 'query'
  11. query_keywords = ['user-agent']
  12. query_examples = ''
  13. # Self User Agent regex
  14. p = re.compile('.*user[ -]agent.*', re.IGNORECASE)
  15. def post_search(request, search):
  16. if search.search_query.pageno > 1:
  17. return True
  18. if search.search_query.query == 'ip':
  19. ip = get_real_ip(request)
  20. search.result_container.answers['ip'] = {'answer': ip}
  21. elif p.match(search.search_query.query):
  22. ua = request.user_agent
  23. search.result_container.answers['user-agent'] = {'answer': ua}
  24. return True