Sem-Ver: feature Add a handle_retrieval_exception method to the HTTPSMultiRepositoryPublicKeyRetriever class.

Signed-off-by: David Black <[email protected]>

Author: dbaxa

atlassian_jwt_auth/key.py

@@ -136,14 +136,20 @@ def _create_retrievers(self, key_repository_urls):
         return [HTTPSPublicKeyRetriever(url) for url
                 in key_repository_urls]
 
+    def handle_retrieval_exception(self, retriever, exception):
+        """ Handles working with exceptions encountered during key
+            retrieval.
+        """
+        if isinstance(exception, PublicKeyRetrieverException):
+            if exception.status_code is None or exception.status_code < 500:
+                raise
+
     def retrieve(self, key_identifier, **requests_kwargs):
         for retriever in self._retrievers:
             try:
                 return retriever.retrieve(key_identifier, **requests_kwargs)
             except (RequestException, PublicKeyRetrieverException) as e:
-                if isinstance(e, PublicKeyRetrieverException):
-                    if e.status_code is None or e.status_code < 500:
-                        raise
+                self.handle_retrieval_exception(retriever, e)
                 logger = logging.getLogger(__name__)
                 logger.warn('Unable to retrieve public key from store',
                             extra={'underlying_error': str(e),