diff options
Diffstat (limited to 'scraper/s2-final-report.py')
| -rw-r--r-- | scraper/s2-final-report.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/scraper/s2-final-report.py b/scraper/s2-final-report.py index e20b3d98..c99edaee 100644 --- a/scraper/s2-final-report.py +++ b/scraper/s2-final-report.py @@ -23,20 +23,23 @@ def s2_final_report(): def process_paper(row, addresses): aggregate_citations = {} + address = None papers = [] print(row['paper_ids']) for paper_id in row['paper_ids']: res = process_single_paper(row, paper_id, addresses, aggregate_citations) if res: papers.append(res) + if res['address']: + address = res['address'] if not len(papers): return - print(papers) + for papers[0]['address'] with open('{}/{}.json'.format(DIR_PUBLIC_CITATIONS, row['key']), 'w') as f: json.dump({ 'id': papers[0]['paper_id'], 'paper': papers[0], - 'address': papers[0]['address'], + 'address': address, 'additional_papers': papers[1:], 'citations': [aggregate_citations[key] for key in aggregate_citations.keys()], }, f) |