merge

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()
+