self_info.py 925 B

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