diff --git a/aurora/models.py b/aurora/models.py index a7b81a7..f00f0b5 100644 --- a/aurora/models.py +++ b/aurora/models.py @@ -118,14 +118,14 @@ class Change(AuroraBaseModel): @classmethod def from_dict(cls, bot: Red, data: dict) -> "Change": - if data["duration"] is not None: + if "duration" in data and data["duration"] is not None: hours, minutes, seconds = map(int, data["duration"].split(':')) duration = timedelta(hours=hours, minutes=minutes, seconds=seconds) else: duration = None data.update({ "timestamp": datetime.fromtimestamp(data["timestamp"]), - "end_timestamp": datetime.fromtimestamp(data["end_timestamp"]) if data["end_timestamp"] else None, + "end_timestamp": datetime.fromtimestamp(data["end_timestamp"]) if "end_timestamp" in data and data["end_timestamp"] else None, "duration": duration }) return cls(bot=bot, **data)