From 04bb3fc0cd974cbae1db16e1c4413e05f2f3cd3c Mon Sep 17 00:00:00 2001
From: neodarz <neodarz@neodarz.net>
Date: Wed, 16 Jan 2019 19:35:47 +0100
Subject: Add manticoresearch search function

---
 sphinx/sphinx.py | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 sphinx/sphinx.py

(limited to 'sphinx/sphinx.py')

diff --git a/sphinx/sphinx.py b/sphinx/sphinx.py
new file mode 100644
index 0000000..1fffa3e
--- /dev/null
+++ b/sphinx/sphinx.py
@@ -0,0 +1,39 @@
+from .sphinxapi import *
+
+import config
+
+sphinxclient = SphinxClient()
+
+sphinxclient.SetServer(host=config.SPHINX_HOST, port=config.SPHINX_PORT)
+
+def search(request):
+    """
+    Simple sphinx seach function, return a array of all documents matching the
+    search terms.
+    """
+    res = sphinxclient.Query(request)
+
+    opts = {'before_match': '<b>', 'after_match':'</b>', 'chunk_separator': '...', 'limit': 400, 'around': 15}
+    index = 'datas'
+
+    response = []
+
+    if 'matches' in res:
+        n = 1
+        for match in res['matches']:
+            attrsdump = ''
+            for attr in res['attrs']:
+                attrname = attr[0]
+                attrtype = attr[1]
+                if attrname != "content":
+                    value = match['attrs'][attrname]
+                    if attrtype==SPH_ATTR_TIMESTAMP:
+                        value = time.strftime ( '%Y-%m-%d %H:%M:%S', time.localtime(value) )
+                    attrsdump = '%s, \'%s\'=\'%s\'' % ( attrsdump, attrname, value)
+            docs = []
+            docs.append(''.join([line.strip('\n') for line in match['attrs']['content']]))
+            res_excerpts = sphinxclient.BuildExcerpts(index=index, docs=docs, opts=opts, words=request)
+            response.append({'id': match['id'], 'weight': match['weight'], 'url': match['attrs']['url'], 'title': match['attrs']['title'], 'excerpts': res_excerpts})
+            n += 1
+
+    return response
-- 
cgit v1.2.1