summaryrefslogtreecommitdiff
path: root/scraper/s2.py
diff options
context:
space:
mode:
Diffstat (limited to 'scraper/s2.py')
-rw-r--r--scraper/s2.py17
1 files changed, 9 insertions, 8 deletions
diff --git a/scraper/s2.py b/scraper/s2.py
index 01c0b4d5..ec6a9172 100644
--- a/scraper/s2.py
+++ b/scraper/s2.py
@@ -177,19 +177,20 @@ class SemanticScholarAPI(object):
return None if resp.status_code != 200 else resp.json() # Paper(**resp.json())
@staticmethod
- def search(q):
- resp = requests.post(SemanticScholarAPI.SEARCH_ENDPOINT, json={
+ def search(q, page=1, pageSize=10, yearFilter=None):
+ query = {
'authors': [],
'coAuthors': [],
- 'facets': {},
- 'page': 1,
- 'pageSize': 10,
+ #'facets': {},
+ 'page': page,
+ 'pageSize': pageSize,
'publicationTypes': [],
'queryString': q,
'requireViewablePdf': False,
'sort': "relevance",
'venues': [],
- 'yearFilter': None,
- }, headers=SemanticScholarAPI.headers)
- # print(resp.status_code)
+ 'yearFilter': yearFilter,
+ }
+ resp = requests.post(SemanticScholarAPI.SEARCH_ENDPOINT, json=query, headers=SemanticScholarAPI.headers)
return None if resp.status_code != 200 else resp.json()
+