.

1 files changed, 31 insertions, 0 deletions

diff --git a/util.py b/util.py
new file mode 100644
index 00000000..dae3f67c
--- /dev/null
+++ b/util.py
@@ -0,0 +1,31 @@
+import os
+import csv
+import simplejson as json
+
+def read_citation_list(index=0):
+  filename = './datasets/citations.csv'
+  if index > 0:
+    fn, ext = os.path.splitext(filename)
+    filename = fn + '-' + str(index) + ext
+  with open(filename, 'r') as f:
+    reader = csv.reader(f)
+    lines = list(reader)
+    keys = lines[0]
+    lines = lines[1:]
+    return keys, lines
+
+def read_json(fn):
+  with open(fn, 'r') as json_file:
+    return json.load(json_file)
+
+def write_json(fn, data):
+  with open(fn, 'w') as outfile:
+    json.dump(data, outfile)
+
+def write_csv(fn, keys, rows):
+  with open(fn, 'w') as f:
+    writer = csv.writer(f)
+    if keys is not None:
+      writer.writerow(keys)
+    for row in rows:
+      writer.writerow(row)