summaryrefslogtreecommitdiff
path: root/searx/answerers/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'searx/answerers/__init__.py')
-rw-r--r--searx/answerers/__init__.py50
1 files changed, 50 insertions, 0 deletions
diff --git a/searx/answerers/__init__.py b/searx/answerers/__init__.py
new file mode 100644
index 0000000..444316f
--- /dev/null
+++ b/searx/answerers/__init__.py
@@ -0,0 +1,50 @@
+from os import listdir
+from os.path import realpath, dirname, join, isdir
+from sys import version_info
+from searx.utils import load_module
+from collections import defaultdict
+
+if version_info[0] == 3:
+ unicode = str
+
+
+answerers_dir = dirname(realpath(__file__))
+
+
+def load_answerers():
+ answerers = []
+ for filename in listdir(answerers_dir):
+ if not isdir(join(answerers_dir, filename)) or filename.startswith('_'):
+ continue
+ module = load_module('answerer.py', join(answerers_dir, filename))
+ if not hasattr(module, 'keywords') or not isinstance(module.keywords, tuple) or not len(module.keywords):
+ exit(2)
+ answerers.append(module)
+ return answerers
+
+
+def get_answerers_by_keywords(answerers):
+ by_keyword = defaultdict(list)
+ for answerer in answerers:
+ for keyword in answerer.keywords:
+ for keyword in answerer.keywords:
+ by_keyword[keyword].append(answerer.answer)
+ return by_keyword
+
+
+def ask(query):
+ results = []
+ query_parts = list(filter(None, query.query.split()))
+
+ if query_parts[0].decode('utf-8') not in answerers_by_keywords:
+ return results
+
+ for answerer in answerers_by_keywords[query_parts[0].decode('utf-8')]:
+ result = answerer(query)
+ if result:
+ results.append(result)
+ return results
+
+
+answerers = load_answerers()
+answerers_by_keywords = get_answerers_by_keywords(answerers)