From 5ab32f0abdb56721110bbff3051dc7231c79086b Mon Sep 17 00:00:00 2001 From: Lester Celestial Date: Wed, 31 Jan 2024 02:00:36 +0800 Subject: [PATCH] Prevent attribute error when auth.message is not found --- openconnect_sso/authenticator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/openconnect_sso/authenticator.py b/openconnect_sso/authenticator.py index d8da09f..83d2aaa 100644 --- a/openconnect_sso/authenticator.py +++ b/openconnect_sso/authenticator.py @@ -150,7 +150,7 @@ def parse_auth_request_response(xml): resp = AuthRequestResponse( auth_id=xml.auth.get("id"), auth_title=getattr(xml.auth, "title", ""), - auth_message=xml.auth.message, + auth_message=getattr(xml.auth, "message", ""), auth_error=getattr(xml.auth, "error", ""), opaque=xml.opaque, login_url=xml.auth["sso-v2-login"], @@ -185,7 +185,7 @@ def parse_auth_complete_response(xml): assert xml.auth.get("id") == "success" resp = AuthCompleteResponse( auth_id=xml.auth.get("id"), - auth_message=xml.auth.message, + auth_message=getattr(xml.auth, "message", ""), session_token=xml["session-token"], server_cert_hash=xml.config["vpn-base-config"]["server-cert-hash"], )