records: implement collaborations name autocomplete

Signed-off-by: Iuliana Voinea <[email protected]>

inspirehep/config.py

@@ -366,7 +366,10 @@
             )),
             book_title=dict(completion=dict(
                 field='bookautocomplete'
-            ))
+            )),
+            collaboration_name=dict(completion=dict(
+                            field='collaborations.collaboration_suggest'
+                        ))
         ),
         list_route='/literature/',
         item_route=(

inspirehep/modules/records/mappings/records/hep.json

@@ -310,6 +310,9 @@
                         },
                         "value": {
                             "type": "string"
+                        },
+                        "collaboration_suggest": {
+                            "type": "completion"
                         }
                     },
                     "type": "object"

inspirehep/modules/records/receivers.py

@@ -138,6 +138,7 @@ def enhance_after_index(sender, json, *args, **kwargs):
     """
     populate_recid_from_ref(sender, json, *args, **kwargs)
     populate_bookautocomplete(sender, json, *args, **kwargs)
+    populate_collaboration_suggest(sender, json, *args, **kwargs)
     populate_abstract_source_suggest(sender, json, *args, **kwargs)
     populate_affiliation_suggest(sender, json, *args, **kwargs)
     populate_author_count(sender, json, *args, **kwargs)
@@ -184,6 +185,24 @@ def populate_bookautocomplete(sender, json, *args, **kwargs):
     })
 
 
+def populate_collaboration_suggest(sender, json, *args, **kwargs):
+    """Populate the ```collaboration_suggest`` field of Literature records."""
+    if 'hep.json' not in json.get('$schema'):
+        return
+
+    collaborations = json.get('collaborations', [])
+
+    for collaboration in collaborations:
+        name = collaboration.get('value')
+        if name:
+            collaboration.update({
+                'collaboration_suggest': {
+                    'input': name,
+                    'output': name,
+                },
+            })
+
+
 def populate_inspire_document_type(sender, json, *args, **kwargs):
     """Populate the ``facet_inspire_doc_type`` field of Literature records."""
     if 'hep.json' not in json.get('$schema'):