From f76d80de543996f9a90eaf46e226354ad4681731 Mon Sep 17 00:00:00 2001
From: asciimoo <asciimoo@gmail.com>
Date: Thu, 17 Oct 2013 21:06:28 +0200
Subject: [enh] category support

---
 searx/engines/__init__.py | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'searx/engines/__init__.py')

diff --git a/searx/engines/__init__.py b/searx/engines/__init__.py
index 083ce06cd..cc531bbbe 100644
--- a/searx/engines/__init__.py
+++ b/searx/engines/__init__.py
@@ -27,15 +27,23 @@ engine_dir = dirname(realpath(__file__))
 
 engines = {}
 
+categories = {'general': []}
+
 for filename in listdir(engine_dir):
     modname = splitext(filename)[0]
     if filename.startswith('_') or not filename.endswith('.py'):
         continue
     filepath = join(engine_dir, filename)
     engine = load_source(modname, filepath)
+    engine.name = modname
     if not hasattr(engine, 'request') or not hasattr(engine, 'response'):
         continue
     engines[modname] = engine
+    if not hasattr(engine, 'categories'):
+        categories['general'].append(engine)
+    else:
+        for category_name in engine.categories:
+            categories.setdefault(category_name, []).append(engine)
 
 def default_request_params():
     return {'method': 'GET', 'headers': {}, 'data': {}, 'url': ''}
-- 
cgit v1.2.3