Removed code duplication

sybrenstuvel · sybrenstuvel · commit 4a6a2b7b7fac · 2016-01-27T15:43:28.000+01:00
diff --git a/rsa/key.py b/rsa/key.py
@@ -70,14 +70,21 @@ def load_pkcs1(cls, keyfile, format='PEM'):
             'DER': cls._load_pkcs1_der,
         }
 
-        if format not in methods:
+        method = cls._assert_format_exists(format, methods)
+        return method(keyfile)
+
+    @staticmethod
+    def _assert_format_exists(file_format, methods):
+        """Checks whether the given file format exists in 'methods'.
+        """
+
+        try:
+            return methods[file_format]
+        except KeyError:
             formats = ', '.join(sorted(methods.keys()))
-            raise ValueError('Unsupported format: %r, try one of %s' % (format,
+            raise ValueError('Unsupported format: %r, try one of %s' % (file_format,
                                                                         formats))
 
-        method = methods[format]
-        return method(keyfile)
-
     def save_pkcs1(self, format='PEM'):
         """Saves the public key in PKCS#1 DER or PEM format.
 
@@ -90,12 +97,7 @@ def save_pkcs1(self, format='PEM'):
             'DER': self._save_pkcs1_der,
         }
 
-        if format not in methods:
-            formats = ', '.join(sorted(methods.keys()))
-            raise ValueError('Unsupported format: %r, try one of %s' % (format,
-                                                                        formats))
-
-        method = methods[format]
+        method = self._assert_format_exists(format, methods)
         return method()
 
     def blind(self, message, r):
