Fixed NoneType error on next rocket. Closes #25.

djtimca · Jan 25, 2021 · 4d90bb8 · 4d90bb8
1 parent f2e53b5
commit 4d90bb8
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/custom_components/spacex/sensor.py b/custom_components/spacex/sensor.py
@@ -327,13 +327,14 @@ def device_state_attributes(self):
         elif self._kind == "spacex_next_launch_rocket":
             core_counter = 1
             for this_core in launch_data["cores_detail"]:
-                self.attrs["core_" + str(core_counter) + "_serial"] = this_core.get(
-                    "details",{}).get("serial")
+                if this_core.get("details"):
+                    self.attrs["core_" + str(core_counter) + "_serial"] = this_core["details"].get("serial")
+                    self.attrs["core_" + str(core_counter) + "_block"] = this_core["details"].get("block")
+
                 self.attrs["core_" + str(core_counter) + "_flight"] = this_core.get(
                     "flight"
                 )
-                self.attrs["core_" + str(core_counter) + "_block"] = this_core.get(
-                    "details",{}).get("block")
+
                 self.attrs[
                     "core_" + str(core_counter) + "_landing_intent"
                 ] = this_core.get("landing_attempt")