Split out function for reading dingbats

1 files changed, 16 insertions, 10 deletions

diff --git a/generate.py b/generate.py
index 7350769..d6cbaa9 100644
--- a/generate.py
+++ b/generate.py
@@ -7,19 +7,25 @@ _root = os.path.dirname(__file__)
 
 
 def _main():
-    with open(os.path.join(_root, "dingbats.csv")) as dingbats_file:
-        dingbat_reader = csv.DictReader(dingbats_file)
+    dingbats = _read_dingbats()
+
+    with open(os.path.join(_root, "js/src/dingbats.ts"), "wt", encoding="utf-8") as typescript_file:
+        typescript_file.write("const dingbats = [\n")
+
+        for dingbat in dingbats:
+            typescript_file.write("    ")
+            json.dump(dingbat, typescript_file)
+            typescript_file.write(",\n")
 
-        with open(os.path.join(_root, "js/src/dingbats.ts"), "wt", encoding="utf-8") as typescript_file:
-            typescript_file.write("const dingbats = [\n")
+        typescript_file.write("];\n")
+        typescript_file.write("export default dingbats;");
 
-            for dingbat in dingbat_reader:
-                typescript_file.write("    ")
-                json.dump(dingbat, typescript_file)
-                typescript_file.write(",\n")
 
-            typescript_file.write("];\n")
-            typescript_file.write("export default dingbats;");
+def _read_dingbats():
+    with open(os.path.join(_root, "dingbats.csv")) as dingbats_file:
+        dingbat_reader = csv.DictReader(dingbats_file)
+
+        return list(dingbat_reader)
 
 
 if __name__ == "__main__":