diff options
Diffstat (limited to 'scraper/util.py')
| -rw-r--r-- | scraper/util.py | 25 |
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')): |