Slightly better handling of outstanding_certs.

IdentityPython · Mar 11, 2015 · 13ff5e8 · 13ff5e8
1 parent 055b314
commit 13ff5e8
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/src/saml2/entity.py b/src/saml2/entity.py
@@ -1,5 +1,5 @@
 import base64
-from binascii import hexlify
+#from binascii import hexlify
 import logging
 from hashlib import sha1
 from Crypto.PublicKey import RSA
@@ -937,13 +937,13 @@ def _parse_response(self, xmlstr, response_cls, service, binding,
 
             if response:
                 if outstanding_certs:
-                    cert = outstanding_certs[
-                        response.in_response_to]
-                    if cert:
-                        _, key_file = make_temp(
-                            "%s" % cert["key"], decode=False)
-                    else:
+                    try:
+                        cert = outstanding_certs[response.in_response_to]
+                    except KeyError:
                         key_file = ""
+                    else:
+                        _, key_file = make_temp("%s" % cert["key"],
+                                                decode=False)
                 else:
                     key_file = ""
                 response = response.verify(key_file)