From 9767c4cc9473dbc6eff402874363015fd70689b9 Mon Sep 17 00:00:00 2001
From: Étienne Loks <etienne.loks@iggdrasil.net>
Date: Fri, 29 Sep 2017 12:20:49 +0200
Subject: Internal searches: fix categories and properties search index (manage
 many fields)

---
 chimere/utils.py | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

(limited to 'chimere/utils.py')

diff --git a/chimere/utils.py b/chimere/utils.py
index ffa5789..2ed2fd3 100644
--- a/chimere/utils.py
+++ b/chimere/utils.py
@@ -1455,3 +1455,42 @@ class IcalManager(ImportManager):
             if created:
                 new_item += 1
         return (new_item, updated_item, msg)
+
+
+def merge_tsvectors(vectors):
+    """
+    Parse tsvector to merge them in one string
+    :param vectors: list of tsvector string
+    :return: merged tsvector
+    """
+    result_dict = {}
+    for vector in vectors:
+        if not vector:
+            continue
+
+        current_position = 0
+        if result_dict:
+            for key in result_dict:
+                max_position = max(result_dict[key])
+                if max_position > current_position:
+                    current_position = max_position
+
+        for dct_member in vector.split(" "):
+            key, positions = dct_member.split(':')
+            key = key[1:-1]  # remove quotes
+            positions = [int(pos) + current_position
+                         for pos in positions.split(',')]
+            if key in result_dict:
+                result_dict[key] += positions
+            else:
+                result_dict[key] = positions
+
+    # {'lamelie': [1, 42, 5]} => {'lamelie': "1,42,5"}
+    result_dict = {k: ",".join([str(val) for val in result_dict[k]])
+                   for k in result_dict}
+    # {'lamelie': "1,5", "hagarde": "2", "regarde": "4"} =>
+    # "lamelie':1,5 hagarde:2 regarde:4"
+    result = " ".join(["{}:{}".format(k, result_dict[k]) for k in result_dict])
+
+    return result
+
-- 
cgit v1.2.3