zaclys/searxng
1
0
Fork 1
mirror of https://github.com/searxng/searxng synced 2024-01-01 19:24:07 +01:00
Code Issues Projects Releases Packages Wiki Activity

[refactor] introduce searx.engines._get_supported_languages

Browse source
This commit is contained in:
Martin Fischer 2022-01-28 11:45:47 +01:00
parent ce5be98f39
commit 73e71508e4
1 changed files with 12 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
searx/engines/__init__.py
View file

@ -13,7 +13,7 @@ usage::
import sys import sys
import copy import copy
from typing import Dict, List, Optional from typing import Collection, Dict, List, Optional
from os.path import realpath, dirname from os.path import realpath, dirname
from searx import logger, settings from searx import logger, settings
@ -172,10 +172,10 @@ def update_engine_attributes(engine: ConfiguredEngine, engine_data):
setattr(engine, arg_name, copy.deepcopy(arg_value)) setattr(engine, arg_name, copy.deepcopy(arg_value))
def set_language_attributes(engine: ConfiguredEngine): def _get_supported_languages(engine: ConfiguredEngine) -> Collection[str]:
# assign supported languages from json file # assign supported languages from json file
if engine.name in ENGINES_LANGUAGES: if engine.name in ENGINES_LANGUAGES:
engine.supported_languages = ENGINES_LANGUAGES[engine.name] return ENGINES_LANGUAGES[engine.name]
elif engine.engine in ENGINES_LANGUAGES: elif engine.engine in ENGINES_LANGUAGES:
# The key of the dictionary ENGINES_LANGUAGES is the *engine name* # The key of the dictionary ENGINES_LANGUAGES is the *engine name*
@ -183,7 +183,7 @@ def set_language_attributes(engine: ConfiguredEngine):
# settings.yml to use the same origin engine (python module) these # settings.yml to use the same origin engine (python module) these
# additional engines can use the languages from the origin engine. # additional engines can use the languages from the origin engine.
# For this use the configured ``engine: ...`` from settings.yml # For this use the configured ``engine: ...`` from settings.yml
engine.supported_languages = ENGINES_LANGUAGES[engine.engine] return ENGINES_LANGUAGES[engine.engine]
if hasattr(engine, 'language'): if hasattr(engine, 'language'):
# For an engine, when there is `language: ...` in the YAML settings, the # For an engine, when there is `language: ...` in the YAML settings, the
@ -196,9 +196,15 @@ def set_language_attributes(engine: ConfiguredEngine):
) )
if isinstance(engine.supported_languages, dict): if isinstance(engine.supported_languages, dict):
engine.supported_languages = {engine.language: engine.supported_languages[engine.language]} return {engine.language: engine.supported_languages[engine.language]}
else: else:
engine.supported_languages = [engine.language] return [engine.language]
return engine.supported_languages
def set_language_attributes(engine: ConfiguredEngine):
engine.supported_languages = _get_supported_languages(engine)
# find custom aliases for non standard language codes # find custom aliases for non standard language codes
engine.language_aliases.update(find_language_aliases(engine.supported_languages)) engine.language_aliases.update(find_language_aliases(engine.supported_languages))