Skip to content

Commit ac9a96d

Browse files
committed
Catch remaining payload data issues.
1 parent fd0d107 commit ac9a96d

File tree

1 file changed

+18
-12
lines changed

1 file changed

+18
-12
lines changed

custom_components/spacex/sensor.py

Lines changed: 18 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -370,20 +370,26 @@ def extra_state_attributes(self):
370370
else:
371371
self.attrs["manufacturer"] = "NA"
372372

373-
self.attrs["payload_type"] = launch_data["payloads_detail"][0]["type"]
374-
self.attrs["payload_mass"] = (
375-
str(
376-
launch_data["payloads_detail"][0]["mass_kg"]
373+
if launch_data["payloads_detail"]:
374+
self.attrs["payload_type"] = launch_data["payloads_detail"][0]["type"]
375+
self.attrs["payload_mass"] = (
376+
str(
377+
launch_data["payloads_detail"][0]["mass_kg"]
378+
)
379+
+ " kg"
377380
)
378-
+ " kg"
379-
)
380-
self.attrs["payload_mass_us"] = (
381-
str(
382-
launch_data["payloads_detail"][0]["mass_lbs"]
381+
self.attrs["payload_mass_us"] = (
382+
str(
383+
launch_data["payloads_detail"][0]["mass_lbs"]
384+
)
385+
+ " lbs"
383386
)
384-
+ " lbs"
385-
)
386-
self.attrs["orbit"] = launch_data["payloads_detail"][0]["orbit"]
387+
self.attrs["orbit"] = launch_data["payloads_detail"][0]["orbit"]
388+
else:
389+
self.attrs["payload_type"] = ""
390+
self.attrs["payload_mass"] = ""
391+
self.attrs["payload_mass_us"] = ""
392+
387393

388394
elif self._kind == "spacex_latest_launch_mission":
389395
self.attrs["mission_patch"] = latest_launch_data["links"].get("patch",{}).get("large")

0 commit comments

Comments
 (0)