searxng/__init__.py at 35db278b0b2a521c4575bf27cab4bef4e8d9704a

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

 # SPDX-License-Identifier: AGPL-3.0-or-later
 """Calculate mathematical expressions using ack#eval
 """
 import decimal
 import re
 import sys
 import subprocess
 from pathlib import Path
 import flask
 import babel
 from flask_babel import gettext
 from searx.plugins import logger
 name = "Basic Calculator"
 description = gettext("Calculate mathematical expressions via the search bar")
 default_on = True
 preference_section = 'general'
 plugin_id = 'calculator'
 logger = logger.getChild(plugin_id)
 def call_calculator(query_py_formatted, timeout):
     # see https://docs.python.org/3/using/cmdline.html
     # -S Disable the import of the module site and the site-dependent manipulations
     #    of sys.path that it entails. Also disable these manipulations if site is
     #    explicitly imported later (call site.main() if you want them to be triggered).
     # -I Run Python in isolated mode. This also implies -E, -P and -s options.
     # -E Ignore all PYTHON* environment variables, e.g. PYTHONPATH and PYTHONHOME, that might be set.
     # -P Don’t prepend a potentially unsafe path to sys.path
     # -s Don’t add the user site-packages directory to sys.path.
     calculator_process_py_path = Path(__file__).parent.absolute() / "calculator_process.py"
     cmd = [sys.executable, "-S", "-I", str(calculator_process_py_path), query_py_formatted]
     try:
         result = subprocess.run(
             cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, timeout=timeout, check=False
         )
         if result.returncode == 0 and not result.stderr:
             return result.stdout
         logger.debug("calculator exited with stderr: %s", result.stderr)
         return None
     except subprocess.TimeoutExpired as e:
         logger.debug("calculator did not exit in time")
         # subprocess.run automatically attempts to terminate the process on timeout.
         # Additional killing is generally not necessary, but we can ensure it if needed.
         # Check if stdout or stderr are captured and log them if available
         if e.stdout:
             logger.debug("Partial stdout before timeout: %s", e.stdout)
         if e.stderr:
             logger.debug("Partial stderr before timeout: %s", e.stderr)
         return None
 def post_search(_request, search):
     # only show the result of the expression on the first page
     if search.search_query.pageno > 1:
         return True
     query = search.search_query.query
     # in order to avoid DoS attacks with long expressions, ignore long expressions
     if len(query) > 100:
         return True
     # replace commonly used math operators with their proper Python operator
     query = query.replace("x", "*").replace(":", "/")
     # use UI language
     ui_locale = babel.Locale.parse(flask.request.preferences.get_value('locale'), sep='-')
     # parse the number system in a localized way
     def _decimal(match: re.Match) -> str:
         val = match.string[match.start() : match.end()]
         val = babel.numbers.parse_decimal(val, ui_locale, numbering_system="latn")
         return str(val)
     loc_decimal = ui_locale.number_symbols["latn"]["decimal"]
     loc_group = ui_locale.number_symbols["latn"]["group"]
     query = re.sub(f"[0-9]+[{loc_decimal}|{loc_group}][0-9]+[{loc_decimal}|{loc_group}]?[0-9]?", _decimal, query)
     # only numbers and math operators are accepted
     if any(str.isalpha(c) for c in query):
         return True
     # in python, powers are calculated via **
     query_py_formatted = query.replace("^", "**")
     # Prevent the runtime from being longer than 50 ms
     result = call_calculator(query_py_formatted, 0.05)
     if result is None or result == "":
         return True
     if len(result) < 15:  # arbitrary number, TODO : check the actual limit
         try:
             result = babel.numbers.format_decimal(result, locale=ui_locale)
         except decimal.InvalidOperation:
             pass
     search.result_container.answers['calculate'] = {'answer': f"{search.search_query.query} = {result}"}
     return True