1#!/usr/bin/env python
2import json
3import os
4
5BASE_PATH = f"{os.environ['ROOT']}/scripts"
6CLASSES_PATH = f"{BASE_PATH}/src/classes.json"
7
8CLASSES = []
9with open(CLASSES_PATH) as f:
10 CLASSES = json.load(open(CLASSES_PATH))
11
12to_add: str = input("Class to add:")
13
14if not to_add.startswith("bi-"):
15 to_add = f"bi-{to_add}"
16
17if to_add not in CLASSES:
18 CLASSES.append(to_add)
19else:
20 raise Exception("Class already exists!")
21
22file = open(CLASSES_PATH, "w")
23
24file.write(json.dumps(sorted(CLASSES), sort_keys=True, indent=2).rstrip())
25file.close()