bugfixes

labteral · Jun 1, 2021 · 5fd0b03 · 5fd0b03
1 parent 1b5ac9f
commit 5fd0b03
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 6 deletions.
diff --git a/digsig/__init__.py b/digsig/__init__.py
@@ -3,7 +3,6 @@
 
 from . import *
 from .auto import *
-from .auto import PrivateKey as PrivateKeyAuto
-from .auto import PublicKey as PublicKeyAuto
 
-__version__ = '1.215.1'
+
+__version__ = '1.216.0'
diff --git a/digsig/auto.py b/digsig/auto.py
@@ -49,6 +49,9 @@ def get_instance(
 
         raise ValueError
 
+# Backwards compatible
+PublicKeyAuto = PublicKey
+
 
 class PrivateKey:
     @staticmethod
@@ -91,7 +94,7 @@ def get_instance(
                     raise ValueError
             elif mode in RsaModes.options():
                 private_key_class = RsaPrivateKey
-                if key_format not in EcdsaFormats.options():
+                if key_format not in RsaFormats.options():
                     raise ValueError
             else:
                 raise ValueError
@@ -103,4 +106,7 @@ def get_instance(
                 key_format=key_format,
             )
 
-        raise ValueError
+        raise ValueError
+
+# Backwards compatible
+PrivateKeyAuto = PrivateKey
diff --git a/digsig/rsa.py b/digsig/rsa.py
@@ -1,7 +1,6 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
 
-from build.lib.digsig.ecdsa import EcdsaFormats
 from .hashing import hash_message
 from .digsig import PublicKeyInterface, PrivateKeyInterface
 from .errors import InvalidSignatureError