Building minimal JSON datafiles + rewrites from ranking page URLs

1 files changed, 15 insertions, 0 deletions

diff --git a/scripts/datafiles-generate-minimal.py b/scripts/datafiles-generate-minimal.py
new file mode 100644
index 0000000..896a1ec
--- /dev/null
+++ b/scripts/datafiles-generate-minimal.py
@@ -0,0 +1,15 @@
+import json, sys
+
+from pyranking.fetch import fetch_ranking
+
+ranking_date = sys.argv[1]
+ranking = fetch_ranking(ranking_date)
+
+output = {}
+for row in ranking:
+    output[row['pid']] = {
+        'p': row['place'],
+        'r': float(row['score'])
+    }
+
+print json.dumps(output)