get institutions from spreadsheet

1 files changed, 23 insertions, 2 deletions

diff --git a/scraper/util.py b/scraper/util.py
index 899089f7..4a43a913 100644
--- a/scraper/util.py
+++ b/scraper/util.py
@@ -338,8 +338,29 @@ class AddressBook (object):
       }
     return None
 
-
-def load_institutions(paperId):
+def load_institutions(paperId, paper_location_lookup={}):
+  if paperId in paper_location_lookup:
+    items = paper_location_lookup[paperId]
+    result = []
+    if items['institution_1']:
+      result.append([items['institution_1'], '', ''])
+    if items['institution_2']:
+      result.append([items['institution_2'], '', ''])
+    if items['institution_3']:
+      result.append([items['institution_3'], '', ''])
+    if items['institution_4']:
+      result.append([items['institution_4'], '', ''])
+    if items['institution_5']:
+      result.append([items['institution_5'], '', ''])
+    if items['institution_6']:
+      result.append([items['institution_6'], '', ''])
+    if items['institution_7']:
+      result.append([items['institution_7'], '', ''])
+    if items['institution_8']:
+      result.append([items['institution_8'], '', ''])
+    return {
+      'institutions': result
+    }
   if os.path.exists(file_path('pdf', paperId, 'institutions.json')):
     return read_json(file_path('pdf', paperId, 'institutions.json'))['institutions']
   elif os.path.exists(file_path('doi', paperId, 'institutions.json')):