From 675e703bdba6a4823c1f4c117bf6cc827de2546a Mon Sep 17 00:00:00 2001
From: James Miller <james.ryland.miller@gmail.com>
Date: Tue, 18 Feb 2020 21:48:59 -0600
Subject: [PATCH] got immunities working

---
 data/yaml/clean-up-monsters.py | 27 +++++++++++++++++++++++----
 1 file changed, 23 insertions(+), 4 deletions(-)

diff --git a/data/yaml/clean-up-monsters.py b/data/yaml/clean-up-monsters.py
index 9d75e69..606707a 100644
--- a/data/yaml/clean-up-monsters.py
+++ b/data/yaml/clean-up-monsters.py
@@ -5,19 +5,25 @@ def main():
     with open("monsters.yaml", 'r') as content_file:
         data = yaml.full_load(content_file)
 
-    # print(data[0]['ability_mods'])
-
     counter = 0
     for i in data:
-        # counter += 1
-        # print("{}\t{}".format(counter, i['name']))
+        counter += 1
+        print("{}\t{}".format(counter, i['name']))
+
+        # Set some data points before iteration
         i['saves_special'] = {}
+
+        # clean up ability mods
         for k, v in i['ability_mods'].items():
             i['ability_mods'][k] = int(v)
             # print(k, i['ability_mods'][k])
+
+        # clean up skills
         for k, v in i['skills'].items():
             i['skills'][k] = int(v)
             # print(k, i['ability_mods'][k])
+
+        # clean up saves
         for k, v in i['saves'].items():
             # print(k, v)
             i['saves_special'][k] = None
@@ -40,6 +46,19 @@ def main():
             i['saves'][k] = int(i['saves'][k])
             # print(k, i['saves'][k])
 
+        # clean up immunities
+        if i['immunities'] == "None":
+            i['immunities'] = None
+        else:
+            tmp = i['immunities']
+            splits = tmp.split(',')
+            res = []
+            for z in splits:
+                res.append(z.strip())
+            i['immunities'] = res
+
+        print(i['immunities'])
+
     final = yaml.safe_dump(data, allow_unicode=True)