blob: 44bab08bbd586cd7d343671b7d335bdc8b72241a (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
import os
import sys
import csv
import subprocess
import time
import random
import re
import simplejson as json
import click
from s2 import SemanticScholarAPI
from util import *
'''
s2 search API format:
results
matchedAuthors
matchedPresentations
query
querySuggestions
results
stats
totalPages
totalResults
'''
@click.command()
@click.option('--index', '-n', default=0, help='Index of CSV (query,)')
def fetch_entries(index):
keys, lines = read_citation_list(index)
s2 = SemanticScholarAPI()
for line in lines:
label = line[0]
title = re.sub(r'[^-0-9a-zA-Z ]+', '', line[1])
entry_fn = './datasets/s2/entries/{}.json'.format(title)
if not os.path.exists(entry_fn):
results = s2.search(title)
write_json(dump_fn, results)
if len(results['results']) == 0:
print("No results for {}".format(title))
else:
print(title)
write_json(entry_fn, results['results'][0])
time.sleep(random.randint(10, 20))
if __name__ == '__main__':
fetch_entries()
|
