search-deep.py

1 files changed, 4 insertions, 4 deletions

diff --git a/scraper/s2.py b/scraper/s2.py
index 01c0b4d5..26334de8 100644
--- a/scraper/s2.py
+++ b/scraper/s2.py
@@ -177,19 +177,19 @@ class SemanticScholarAPI(object):
         return None if resp.status_code != 200 else resp.json() # Paper(**resp.json())
 
     @staticmethod
-    def search(q):
+    def search(q, page=1, pageSize=10, yearFilter=None):
         resp = requests.post(SemanticScholarAPI.SEARCH_ENDPOINT, json={
             'authors': [],
             'coAuthors': [],
             'facets': {},
-            'page': 1,
-            'pageSize': 10,
+            'page': page,
+            'pageSize': pageSize,
             'publicationTypes': [],
             'queryString': q,
             'requireViewablePdf': False,
             'sort': "relevance",
             'venues': [],
-            'yearFilter': None,
+            'yearFilter': yearFilter,
         }, headers=SemanticScholarAPI.headers)
         # print(resp.status_code)
         return None if resp.status_code != 200 else resp.json()