From 5396ae284e0a4494ceb4a52c0dc225e60e5f9f23 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Mon, 23 Dec 2019 12:07:34 +0100
Subject: add categories_to_json.py

---
 inversion/categories_to_json.py | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)
 create mode 100644 inversion/categories_to_json.py

(limited to 'inversion/categories_to_json.py')

diff --git a/inversion/categories_to_json.py b/inversion/categories_to_json.py
new file mode 100644
index 0000000..4931fdc
--- /dev/null
+++ b/inversion/categories_to_json.py
@@ -0,0 +1,31 @@
+import os
+import sys
+import json
+
+if len(sys.argv) < 2:
+  sys.exit('usage: categories_to_json.py infile.txt')
+
+in_fn = sys.argv[1]
+
+with open(in_fn, "r") as f:
+  groups = f.read().lower().split("\n\n")
+data = {}
+for g in groups:
+  lines = g.split("\n")
+  title = lines[0].lower()
+  group = {}
+  for line in lines[1:]:
+    partz = line.split(") ")
+    if len(partz) < 2:
+      continue
+    klass = int(partz[0])
+    label = partz[1].strip()
+    if len(partz) > 2:
+      print(line)
+    group[label] = klass
+  data[title] = group
+
+name, ext = os.path.splitext(in_fn)
+out_fn = name + '.json'
+with open(out_fn, 'w') as outfile:
+  json.dump(data, outfile, sort_keys=True, indent=2)
-- 
cgit v1.2.3-70-g09d2