leveling None handling

tassaron · Aug 8, 2023 · 12bd687 · 12bd687
1 parent 25e109e
commit 12bd687
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/dnd_character/character.py b/dnd_character/character.py
@@ -663,12 +663,15 @@ def apply_class_level(self) -> None:
         if new_cfd is None:
             self.class_features_data = None
         else:
-            self.class_features_data = {
-                k: v
-                if "available" not in k and "days" not in k
-                else self.class_features_data[k]
-                for k, v in new_cfd.items()
-            }
+            if self.class_features_data is None:
+                self.class_features_data = new_cfd
+            else:
+                self.class_features_data = {
+                    k: v
+                    if "available" not in k and "days" not in k
+                    else self.class_features_data[k]
+                    for k, v in new_cfd.items()
+                }
 
     def set_spell_slots(self, new_spell_slots: dict[str, int]) -> dict[str, int]:
         default_spell_slots = {