got immunities working

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)