mirror of
https://github.com/searxng/searxng
synced 2024-01-01 19:24:07 +01:00
217 lines
7.4 KiB
Python
Executable file
217 lines
7.4 KiB
Python
Executable file
#!/usr/bin/env python
|
|
"""Script to run searxng from terminal.
|
|
|
|
Getting categories without initiate the engine will only return `['general']`
|
|
|
|
>>> import searxng.engines
|
|
... list(searxng.engines.categories.keys())
|
|
['general']
|
|
>>> import searxng.search
|
|
... searxng.search.initialize()
|
|
... list(searxng.engines.categories.keys())
|
|
['general', 'it', 'science', 'images', 'news', 'videos', 'music', 'files', 'social media', 'map']
|
|
|
|
Example to use this script:
|
|
|
|
.. code:: bash
|
|
|
|
$ python3 searxng_extra/standalone_searxng.py rain
|
|
|
|
Example to run it from python:
|
|
|
|
>>> import importlib
|
|
... import json
|
|
... import sys
|
|
... import searxng.engines
|
|
... import searxng.search
|
|
... search_query = 'rain'
|
|
... # initialize engines
|
|
... searxng.search.initialize()
|
|
... # load engines categories once instead of each time the function called
|
|
... engine_cs = list(searxng.engines.categories.keys())
|
|
... # load module
|
|
... spec = importlib.util.spec_from_file_location(
|
|
... 'utils.standalone_searx', 'searxng_extra/standalone_searxng.py')
|
|
... sas = importlib.util.module_from_spec(spec)
|
|
... spec.loader.exec_module(sas)
|
|
... # use function from module
|
|
... prog_args = sas.parse_argument([search_query], category_choices=engine_cs)
|
|
... search_q = sas.get_search_query(prog_args, engine_categories=engine_cs)
|
|
... res_dict = sas.to_dict(search_q)
|
|
... sys.stdout.write(json.dumps(
|
|
... res_dict, sort_keys=True, indent=4, ensure_ascii=False,
|
|
... default=sas.json_serial))
|
|
{
|
|
"answers": [],
|
|
"infoboxes": [ {...} ],
|
|
"paging": true,
|
|
"results": [... ],
|
|
"results_number": 820000000.0,
|
|
"search": {
|
|
"lang": "all",
|
|
"pageno": 1,
|
|
"q": "rain",
|
|
"safesearch": 0,
|
|
"timerange": null
|
|
},
|
|
"suggestions": [...]
|
|
}
|
|
""" # noqa: E501
|
|
# pylint: disable=pointless-string-statement
|
|
'''
|
|
searxng is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Affero General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
searxng is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU Affero General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Affero General Public License
|
|
along with searxng. If not, see < http://www.gnu.org/licenses/ >.
|
|
|
|
(C) 2016- by Alexandre Flament, <alex@al-f.net>
|
|
'''
|
|
# pylint: disable=wrong-import-position
|
|
import argparse
|
|
import sys
|
|
from datetime import datetime
|
|
from json import dumps
|
|
from typing import Any, Dict, List, Optional
|
|
|
|
import searxng
|
|
import searxng.preferences
|
|
import searxng.query
|
|
import searxng.search
|
|
import searxng.webadapter
|
|
|
|
EngineCategoriesVar = Optional[List[str]]
|
|
|
|
|
|
def get_search_query(
|
|
args: argparse.Namespace, engine_categories: EngineCategoriesVar = None
|
|
) -> searxng.search.SearchQuery:
|
|
"""Get search results for the query"""
|
|
if engine_categories is None:
|
|
engine_categories = list(searxng.engines.categories.keys())
|
|
try:
|
|
category = args.category.decode('utf-8')
|
|
except AttributeError:
|
|
category = args.category
|
|
form = {
|
|
"q": args.query,
|
|
"categories": category,
|
|
"pageno": str(args.pageno),
|
|
"language": args.lang,
|
|
"time_range": args.timerange
|
|
}
|
|
preferences = searxng.preferences.Preferences(
|
|
['oscar'], engine_categories, searxng.engines.engines, [])
|
|
preferences.key_value_settings['safesearch'].parse(args.safesearch)
|
|
|
|
search_query = searxng.webadapter.get_search_query_from_webapp(
|
|
preferences, form)[0]
|
|
return search_query
|
|
|
|
|
|
def no_parsed_url(results: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
|
|
"""Remove parsed url from dict."""
|
|
for result in results:
|
|
del result['parsed_url']
|
|
return results
|
|
|
|
|
|
def json_serial(obj: Any) -> Any:
|
|
"""JSON serializer for objects not serializable by default json code.
|
|
|
|
:raise TypeError: raised when **obj** is not serializable
|
|
"""
|
|
if isinstance(obj, datetime):
|
|
serial = obj.isoformat()
|
|
return serial
|
|
if isinstance(obj, bytes):
|
|
return obj.decode('utf8')
|
|
if isinstance(obj, set):
|
|
return list(obj)
|
|
raise TypeError("Type ({}) not serializable".format(type(obj)))
|
|
|
|
|
|
def to_dict(search_query: searxng.search.SearchQuery) -> Dict[str, Any]:
|
|
"""Get result from parsed arguments."""
|
|
result_container = searxng.search.Search(search_query).search()
|
|
result_container_json = {
|
|
"search": {
|
|
"q": search_query.query,
|
|
"pageno": search_query.pageno,
|
|
"lang": search_query.lang,
|
|
"safesearch": search_query.safesearch,
|
|
"timerange": search_query.time_range,
|
|
},
|
|
"results": no_parsed_url(result_container.get_ordered_results()),
|
|
"infoboxes": result_container.infoboxes,
|
|
"suggestions": list(result_container.suggestions),
|
|
"answers": list(result_container.answers),
|
|
"paging": result_container.paging,
|
|
"results_number": result_container.results_number()
|
|
}
|
|
return result_container_json
|
|
|
|
|
|
def parse_argument(
|
|
args: Optional[List[str]]=None,
|
|
category_choices: EngineCategoriesVar=None
|
|
) -> argparse.Namespace:
|
|
"""Parse command line.
|
|
|
|
:raise SystemExit: Query argument required on `args`
|
|
|
|
Examples:
|
|
|
|
>>> import importlib
|
|
... # load module
|
|
... spec = importlib.util.spec_from_file_location(
|
|
... 'utils.standalone_searx', 'utils/standalone_searxng.py')
|
|
... sas = importlib.util.module_from_spec(spec)
|
|
... spec.loader.exec_module(sas)
|
|
... sas.parse_argument()
|
|
usage: ptipython [-h] [--category [{general}]] [--lang [LANG]] [--pageno [PAGENO]] [--safesearch [{0,1,2}]] [--timerange [{day,week,month,year}]]
|
|
query
|
|
SystemExit: 2
|
|
>>> sas.parse_argument(['rain'])
|
|
Namespace(category='general', lang='all', pageno=1, query='rain', safesearch='0', timerange=None)
|
|
""" # noqa: E501
|
|
if not category_choices:
|
|
category_choices = list(searxng.engines.categories.keys())
|
|
parser = argparse.ArgumentParser(description='Standalone searxng.')
|
|
parser.add_argument('query', type=str,
|
|
help='Text query')
|
|
parser.add_argument('--category', type=str, nargs='?',
|
|
choices=category_choices,
|
|
default='general',
|
|
help='Search category')
|
|
parser.add_argument('--lang', type=str, nargs='?', default='all',
|
|
help='Search language')
|
|
parser.add_argument('--pageno', type=int, nargs='?', default=1,
|
|
help='Page number starting from 1')
|
|
parser.add_argument(
|
|
'--safesearch', type=str, nargs='?',
|
|
choices=['0', '1', '2'], default='0',
|
|
help='Safe content filter from none to strict')
|
|
parser.add_argument(
|
|
'--timerange', type=str,
|
|
nargs='?', choices=['day', 'week', 'month', 'year'],
|
|
help='Filter by time range')
|
|
return parser.parse_args(args)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
searxng.search.initialize()
|
|
engine_cs = list(searxng.engines.categories.keys())
|
|
prog_args = parse_argument(category_choices=engine_cs)
|
|
search_q = get_search_query(prog_args, engine_categories=engine_cs)
|
|
res_dict = to_dict(search_q)
|
|
sys.stdout.write(dumps(
|
|
res_dict, sort_keys=True, indent=4, ensure_ascii=False,
|
|
default=json_serial))
|