scripts to fetch papers from main s2 api

1 files changed, 10 insertions, 0 deletions

diff --git a/util.py b/util.py
index dae3f67c..9f321465 100644
--- a/util.py
+++ b/util.py
@@ -14,6 +14,16 @@ def read_citation_list(index=0):
     lines = lines[1:]
     return keys, lines
 
+def read_csv(fn, keys=True):
+  with open(fn, 'r') as f:
+    reader = csv.reader(f)
+    lines = list(reader)
+    if keys:
+      keys = lines[0]
+      lines = lines[1:]
+      return keys, lines
+    return lines
+
 def read_json(fn):
   with open(fn, 'r') as json_file:
     return json.load(json_file)