forked from zaclys/searxng
[lint] pylint searx/search/processors files / BTW add some doc-strings
Signed-off-by: Markus Heiser <markus.heiser@darmarit.de>
This commit is contained in:
parent
b1557b5443
commit
924f9afea3
|
@ -1,37 +1,49 @@
|
||||||
# SPDX-License-Identifier: AGPL-3.0-or-later
|
# SPDX-License-Identifier: AGPL-3.0-or-later
|
||||||
|
# lint: pylint
|
||||||
|
|
||||||
|
"""Implement request processores used by engine-types.
|
||||||
|
|
||||||
|
"""
|
||||||
|
|
||||||
|
__all__ = [
|
||||||
|
'EngineProcessor',
|
||||||
|
'OfflineProcessor',
|
||||||
|
'OnlineProcessor',
|
||||||
|
'OnlineDictionaryProcessor',
|
||||||
|
'OnlineCurrencyProcessor',
|
||||||
|
'processors',
|
||||||
|
]
|
||||||
|
|
||||||
|
from searx import logger
|
||||||
|
import searx.engines as engines
|
||||||
|
|
||||||
from .online import OnlineProcessor
|
from .online import OnlineProcessor
|
||||||
from .offline import OfflineProcessor
|
from .offline import OfflineProcessor
|
||||||
from .online_dictionary import OnlineDictionaryProcessor
|
from .online_dictionary import OnlineDictionaryProcessor
|
||||||
from .online_currency import OnlineCurrencyProcessor
|
from .online_currency import OnlineCurrencyProcessor
|
||||||
from .abstract import EngineProcessor
|
from .abstract import EngineProcessor
|
||||||
from searx import logger
|
|
||||||
import searx.engines as engines
|
|
||||||
|
|
||||||
|
|
||||||
__all__ = ['EngineProcessor', 'OfflineProcessor', 'OnlineProcessor',
|
|
||||||
'OnlineDictionaryProcessor', 'OnlineCurrencyProcessor', 'processors']
|
|
||||||
logger = logger.getChild('search.processors')
|
logger = logger.getChild('search.processors')
|
||||||
processors = {}
|
processors = {}
|
||||||
|
"""Cache request processores, stored by *engine-name* (:py:func:`initialize`)"""
|
||||||
|
|
||||||
def get_processor_class(engine_type):
|
def get_processor_class(engine_type):
|
||||||
|
"""Return processor class according to the ``engine_type``"""
|
||||||
for c in [OnlineProcessor, OfflineProcessor, OnlineDictionaryProcessor, OnlineCurrencyProcessor]:
|
for c in [OnlineProcessor, OfflineProcessor, OnlineDictionaryProcessor, OnlineCurrencyProcessor]:
|
||||||
if c.engine_type == engine_type:
|
if c.engine_type == engine_type:
|
||||||
return c
|
return c
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
|
||||||
def get_processor(engine, engine_name):
|
def get_processor(engine, engine_name):
|
||||||
|
"""Return processor instance that fits to ``engine.engine.type``)"""
|
||||||
engine_type = getattr(engine, 'engine_type', 'online')
|
engine_type = getattr(engine, 'engine_type', 'online')
|
||||||
processor_class = get_processor_class(engine_type)
|
processor_class = get_processor_class(engine_type)
|
||||||
if processor_class:
|
if processor_class:
|
||||||
return processor_class(engine, engine_name)
|
return processor_class(engine, engine_name)
|
||||||
else:
|
return None
|
||||||
return None
|
|
||||||
|
|
||||||
|
|
||||||
def initialize(engine_list):
|
def initialize(engine_list):
|
||||||
|
"""Initialize all engines and store a processor for each engine in :py:obj:`processors`."""
|
||||||
engines.initialize_engines(engine_list)
|
engines.initialize_engines(engine_list)
|
||||||
for engine_name, engine in engines.engines.items():
|
for engine_name, engine in engines.engines.items():
|
||||||
processor = get_processor(engine, engine_name)
|
processor = get_processor(engine, engine_name)
|
||||||
|
|
|
@ -1,4 +1,9 @@
|
||||||
# SPDX-License-Identifier: AGPL-3.0-or-later
|
# SPDX-License-Identifier: AGPL-3.0-or-later
|
||||||
|
# lint: pylint
|
||||||
|
|
||||||
|
"""Abstract base classes for engine request processores.
|
||||||
|
|
||||||
|
"""
|
||||||
|
|
||||||
import threading
|
import threading
|
||||||
from abc import abstractmethod, ABC
|
from abc import abstractmethod, ABC
|
||||||
|
@ -10,12 +15,13 @@ from searx.network import get_time_for_thread, get_network
|
||||||
from searx.metrics import histogram_observe, counter_inc, count_exception, count_error
|
from searx.metrics import histogram_observe, counter_inc, count_exception, count_error
|
||||||
from searx.exceptions import SearxEngineAccessDeniedException
|
from searx.exceptions import SearxEngineAccessDeniedException
|
||||||
|
|
||||||
|
|
||||||
logger = logger.getChild('searx.search.processor')
|
logger = logger.getChild('searx.search.processor')
|
||||||
SUSPENDED_STATUS = {}
|
SUSPENDED_STATUS = {}
|
||||||
|
|
||||||
|
# pylint: disable=missing-function-docstring
|
||||||
|
|
||||||
class SuspendedStatus:
|
class SuspendedStatus:
|
||||||
|
"""Class to handle suspend state."""
|
||||||
|
|
||||||
__slots__ = 'suspend_end_time', 'suspend_reason', 'continuous_errors', 'lock'
|
__slots__ = 'suspend_end_time', 'suspend_reason', 'continuous_errors', 'lock'
|
||||||
|
|
||||||
|
@ -49,6 +55,7 @@ class SuspendedStatus:
|
||||||
|
|
||||||
|
|
||||||
class EngineProcessor(ABC):
|
class EngineProcessor(ABC):
|
||||||
|
"""Base classes used for all types of reqest processores."""
|
||||||
|
|
||||||
__slots__ = 'engine', 'engine_name', 'lock', 'suspended_status'
|
__slots__ = 'engine', 'engine_name', 'lock', 'suspended_status'
|
||||||
|
|
||||||
|
@ -143,9 +150,7 @@ class EngineProcessor(ABC):
|
||||||
if tests is None:
|
if tests is None:
|
||||||
tests = getattr(self.engine, 'additional_tests', {})
|
tests = getattr(self.engine, 'additional_tests', {})
|
||||||
tests.update(self.get_default_tests())
|
tests.update(self.get_default_tests())
|
||||||
return tests
|
return tests
|
||||||
else:
|
|
||||||
return tests
|
|
||||||
|
|
||||||
def get_default_tests(self):
|
def get_default_tests(self): # pylint: disable=no-self-use
|
||||||
return {}
|
return {}
|
||||||
|
|
|
@ -1,13 +1,17 @@
|
||||||
# SPDX-License-Identifier: AGPL-3.0-or-later
|
# SPDX-License-Identifier: AGPL-3.0-or-later
|
||||||
|
# lint: pylint
|
||||||
|
|
||||||
|
"""Processores for engine-type: ``offline``
|
||||||
|
|
||||||
|
"""
|
||||||
|
|
||||||
from searx import logger
|
from searx import logger
|
||||||
from searx.search.processors.abstract import EngineProcessor
|
from .abstract import EngineProcessor
|
||||||
|
|
||||||
|
|
||||||
logger = logger.getChild('searx.search.processor.offline')
|
logger = logger.getChild('searx.search.processor.offline')
|
||||||
|
|
||||||
|
|
||||||
class OfflineProcessor(EngineProcessor):
|
class OfflineProcessor(EngineProcessor):
|
||||||
|
"""Processor class used by ``offline`` engines"""
|
||||||
|
|
||||||
engine_type = 'offline'
|
engine_type = 'offline'
|
||||||
|
|
||||||
|
@ -21,6 +25,6 @@ class OfflineProcessor(EngineProcessor):
|
||||||
except ValueError as e:
|
except ValueError as e:
|
||||||
# do not record the error
|
# do not record the error
|
||||||
logger.exception('engine {0} : invalid input : {1}'.format(self.engine_name, e))
|
logger.exception('engine {0} : invalid input : {1}'.format(self.engine_name, e))
|
||||||
except Exception as e:
|
except Exception as e: # pylint: disable=broad-except
|
||||||
self.handle_exception(result_container, e)
|
self.handle_exception(result_container, e)
|
||||||
logger.exception('engine {0} : exception : {1}'.format(self.engine_name, e))
|
logger.exception('engine {0} : exception : {1}'.format(self.engine_name, e))
|
||||||
|
|
|
@ -1,24 +1,29 @@
|
||||||
# SPDX-License-Identifier: AGPL-3.0-or-later
|
# SPDX-License-Identifier: AGPL-3.0-or-later
|
||||||
|
# lint: pylint
|
||||||
|
|
||||||
|
"""Processores for engine-type: ``online``
|
||||||
|
|
||||||
|
"""
|
||||||
|
|
||||||
from time import time
|
from time import time
|
||||||
import asyncio
|
import asyncio
|
||||||
|
|
||||||
import httpx
|
import httpx
|
||||||
|
|
||||||
import searx.network
|
import searx.network
|
||||||
from searx import logger
|
from searx import logger
|
||||||
from searx.utils import gen_useragent
|
from searx.utils import gen_useragent
|
||||||
from searx.exceptions import (SearxEngineAccessDeniedException, SearxEngineCaptchaException,
|
from searx.exceptions import (
|
||||||
SearxEngineTooManyRequestsException,)
|
SearxEngineAccessDeniedException,
|
||||||
|
SearxEngineCaptchaException,
|
||||||
|
SearxEngineTooManyRequestsException,
|
||||||
|
)
|
||||||
from searx.metrics.error_recorder import count_error
|
from searx.metrics.error_recorder import count_error
|
||||||
|
from .abstract import EngineProcessor
|
||||||
from searx.search.processors.abstract import EngineProcessor
|
|
||||||
|
|
||||||
|
|
||||||
logger = logger.getChild('searx.search.processor.online')
|
logger = logger.getChild('searx.search.processor.online')
|
||||||
|
|
||||||
|
|
||||||
def default_request_params():
|
def default_request_params():
|
||||||
|
"""Default request parameters for ``online`` engines."""
|
||||||
return {
|
return {
|
||||||
'method': 'GET',
|
'method': 'GET',
|
||||||
'headers': {},
|
'headers': {},
|
||||||
|
@ -31,6 +36,7 @@ def default_request_params():
|
||||||
|
|
||||||
|
|
||||||
class OnlineProcessor(EngineProcessor):
|
class OnlineProcessor(EngineProcessor):
|
||||||
|
"""Processor class for ``online`` engines."""
|
||||||
|
|
||||||
engine_type = 'online'
|
engine_type = 'online'
|
||||||
|
|
||||||
|
@ -153,7 +159,7 @@ class OnlineProcessor(EngineProcessor):
|
||||||
except SearxEngineAccessDeniedException as e:
|
except SearxEngineAccessDeniedException as e:
|
||||||
self.handle_exception(result_container, e, suspend=True)
|
self.handle_exception(result_container, e, suspend=True)
|
||||||
logger.exception('engine {0} : Searx is blocked'.format(self.engine_name))
|
logger.exception('engine {0} : Searx is blocked'.format(self.engine_name))
|
||||||
except Exception as e:
|
except Exception as e: # pylint: disable=broad-except
|
||||||
self.handle_exception(result_container, e)
|
self.handle_exception(result_container, e)
|
||||||
logger.exception('engine {0} : exception : {1}'.format(self.engine_name, e))
|
logger.exception('engine {0} : exception : {1}'.format(self.engine_name, e))
|
||||||
|
|
||||||
|
|
|
@ -1,4 +1,8 @@
|
||||||
# SPDX-License-Identifier: AGPL-3.0-or-later
|
# SPDX-License-Identifier: AGPL-3.0-or-later
|
||||||
|
# lint: pylint
|
||||||
|
"""Processores for engine-type: ``online_currency``
|
||||||
|
|
||||||
|
"""
|
||||||
|
|
||||||
import unicodedata
|
import unicodedata
|
||||||
import re
|
import re
|
||||||
|
@ -6,32 +10,31 @@ import re
|
||||||
from searx.data import CURRENCIES
|
from searx.data import CURRENCIES
|
||||||
from .online import OnlineProcessor
|
from .online import OnlineProcessor
|
||||||
|
|
||||||
|
|
||||||
parser_re = re.compile('.*?(\\d+(?:\\.\\d+)?) ([^.0-9]+) (?:in|to) ([^.0-9]+)', re.I)
|
parser_re = re.compile('.*?(\\d+(?:\\.\\d+)?) ([^.0-9]+) (?:in|to) ([^.0-9]+)', re.I)
|
||||||
|
|
||||||
|
# pylint: disable=missing-function-docstring
|
||||||
|
|
||||||
def normalize_name(name):
|
def normalize_name(name):
|
||||||
name = name.lower().replace('-', ' ').rstrip('s')
|
name = name.lower().replace('-', ' ').rstrip('s')
|
||||||
name = re.sub(' +', ' ', name)
|
name = re.sub(' +', ' ', name)
|
||||||
return unicodedata.normalize('NFKD', name).lower()
|
return unicodedata.normalize('NFKD', name).lower()
|
||||||
|
|
||||||
|
|
||||||
def name_to_iso4217(name):
|
def name_to_iso4217(name):
|
||||||
global CURRENCIES
|
global CURRENCIES # pylint: disable=global-statement
|
||||||
name = normalize_name(name)
|
name = normalize_name(name)
|
||||||
currency = CURRENCIES['names'].get(name, [name])
|
currency = CURRENCIES['names'].get(name, [name])
|
||||||
if isinstance(currency, str):
|
if isinstance(currency, str):
|
||||||
return currency
|
return currency
|
||||||
return currency[0]
|
return currency[0]
|
||||||
|
|
||||||
|
|
||||||
def iso4217_to_name(iso4217, language):
|
def iso4217_to_name(iso4217, language):
|
||||||
global CURRENCIES
|
global CURRENCIES # pylint: disable=global-statement
|
||||||
return CURRENCIES['iso4217'].get(iso4217, {}).get(language, iso4217)
|
return CURRENCIES['iso4217'].get(iso4217, {}).get(language, iso4217)
|
||||||
|
|
||||||
|
|
||||||
class OnlineCurrencyProcessor(OnlineProcessor):
|
class OnlineCurrencyProcessor(OnlineProcessor):
|
||||||
|
|
||||||
|
"""Processor class used by ``online_currency`` engines."""
|
||||||
|
|
||||||
engine_type = 'online_currency'
|
engine_type = 'online_currency'
|
||||||
|
|
||||||
def get_params(self, search_query, engine_category):
|
def get_params(self, search_query, engine_category):
|
||||||
|
|
|
@ -1,15 +1,18 @@
|
||||||
# SPDX-License-Identifier: AGPL-3.0-or-later
|
# SPDX-License-Identifier: AGPL-3.0-or-later
|
||||||
|
# lint: pylint
|
||||||
|
"""Processores for engine-type: ``online_dictionary``
|
||||||
|
|
||||||
|
"""
|
||||||
|
|
||||||
import re
|
import re
|
||||||
|
|
||||||
from searx.utils import is_valid_lang
|
from searx.utils import is_valid_lang
|
||||||
from .online import OnlineProcessor
|
from .online import OnlineProcessor
|
||||||
|
|
||||||
|
|
||||||
parser_re = re.compile('.*?([a-z]+)-([a-z]+) ([^ ]+)$', re.I)
|
parser_re = re.compile('.*?([a-z]+)-([a-z]+) ([^ ]+)$', re.I)
|
||||||
|
|
||||||
|
|
||||||
class OnlineDictionaryProcessor(OnlineProcessor):
|
class OnlineDictionaryProcessor(OnlineProcessor):
|
||||||
|
"""Processor class used by ``online_dictionnary`` engines."""
|
||||||
|
|
||||||
engine_type = 'online_dictionnary'
|
engine_type = 'online_dictionnary'
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue