export: entry point for impact graph record export

modules/websearch/lib/websearch_webinterface.py

@@ -109,6 +109,13 @@
 from invenio.bibfield import get_record
 from invenio.shellutils import mymkdir
 
+# imports for impact graphs
+import json
+import re
+from invenio.bibauthorid_dbinterface import get_title_of_paper
+from invenio.bibrank_citation_searcher import get_refers_to
+from invenio.bibrank_citation_searcher import get_cited_by
+
 import invenio.template
 websearch_templates = invenio.template.load('websearch')
 
@@ -119,7 +126,8 @@
                       get_output_formats(with_attributes=True).values()]
 except KeyError:
     output_formats = ['xd', 'xm', 'hd', 'hb', 'hs', 'hx']
-output_formats.extend(['hm', 't', 'h'])
+output_formats.extend(['hm', 't', 'h', 'impact'])
+
 
 def wash_search_urlargd(form):
     """
@@ -159,6 +167,48 @@ def wash_search_urlargd(form):
 
     return argd
 
+
+def get_year(recid):
+    for tag in ["773__y", "260__c", "269__c", "909C4y", "925__a"]:
+        date = get_fieldvalues([recid], tag)
+
+        if len(date) == 1:
+            date = date[0]
+            match_obj = re.search("\d\d\d\d", date)
+            if match_obj is not None:
+                return int(match_obj.group())
+
+    return None
+
+
+def get_impact_graph(root_recid, max_depth=1):
+    records = {}
+    stack = [(int(root_recid), 0)]
+
+    while len(stack) > 0:
+        recid, depth = stack.pop(0)
+
+        if recid in records:
+            continue
+
+        record = {}
+        records[recid] = record
+
+        record["recid"] = recid
+        record["references"] = list(get_refers_to(recid))
+        record["citations"] = list(get_cited_by(recid))
+        record["title"] = get_title_of_paper(recid)
+        record["year"] = get_year(recid)
+
+        if depth < max_depth:
+            for neighbor in record["references"]:
+                stack.append((neighbor, depth + 1))
+            for neighbor in record["citations"]:
+                stack.append((neighbor, depth + 1))
+
+    return json.dumps(records)
+
+
 class WebInterfaceUnAPIPages(WebInterfaceDirectory):
     """ Handle /unapi set of pages."""
     _exports = ['']
@@ -1193,6 +1243,13 @@ def __call__(self, req, form):
                 text=auth_msg, \
                 navmenuid='search')
 
+        # hack in impact graphs as an additional export format
+        if self.format == "impact":
+            req.content_type = "application/json"
+            req.send_http_header()
+            req.write(get_impact_graph(self.recid))
+            return
+
         # mod_python does not like to return [] in case when of=id:
         out = perform_request_search(req, **argd)
         if isinstance(out, intbitset):