diff --git a/infrastructure/abri/abri_client.py b/infrastructure/abri/abri_client.py index e1049bd0e..99ef095e8 100644 --- a/infrastructure/abri/abri_client.py +++ b/infrastructure/abri/abri_client.py @@ -56,10 +56,13 @@ class AbriClient: raise AbriResponseParseError( "Job_logged element missing from relay response" ) - return JobLogged( - job_no=job_logged.attrib["job_no"], - logged_info=job_logged.attrib["logged_info"], - ) + job_no = job_logged.get("job_no") + logged_info = job_logged.get("logged_info") + if job_no is None or logged_info is None: + raise AbriResponseParseError( + "Job_logged element missing job_no or logged_info" + ) + return JobLogged(job_no=job_no, logged_info=logged_info) @staticmethod def _parse_rejection(root: ET.Element) -> AbriRequestRejected: