|
- from jose.backends.base import Key
- from jose.constants import ALGORITHMS
- from jose.exceptions import JWKError
-
- try:
- from jose.backends import RSAKey # noqa: F401
- except ImportError:
- pass
-
- try:
- from jose.backends import ECKey # noqa: F401
- except ImportError:
- pass
-
- try:
- from jose.backends import AESKey # noqa: F401
- except ImportError:
- pass
-
- try:
- from jose.backends import DIRKey # noqa: F401
- except ImportError:
- pass
-
- try:
- from jose.backends import HMACKey # noqa: F401
- except ImportError:
- pass
-
-
- def get_key(algorithm):
- if algorithm in ALGORITHMS.KEYS:
- return ALGORITHMS.KEYS[algorithm]
- elif algorithm in ALGORITHMS.HMAC: # noqa: F811
- return HMACKey
- elif algorithm in ALGORITHMS.RSA:
- from jose.backends import RSAKey # noqa: F811
-
- return RSAKey
- elif algorithm in ALGORITHMS.EC:
- from jose.backends import ECKey # noqa: F811
-
- return ECKey
- elif algorithm in ALGORITHMS.AES:
- from jose.backends import AESKey # noqa: F811
-
- return AESKey
- elif algorithm == ALGORITHMS.DIR:
- from jose.backends import DIRKey # noqa: F811
-
- return DIRKey
- return None
-
-
- def register_key(algorithm, key_class):
- if not issubclass(key_class, Key):
- raise TypeError("Key class is not a subclass of jwk.Key")
- ALGORITHMS.KEYS[algorithm] = key_class
- ALGORITHMS.SUPPORTED.add(algorithm)
- return True
-
-
- def construct(key_data, algorithm=None):
- """
- Construct a Key object for the given algorithm with the given
- key_data.
- """
-
- # Allow for pulling the algorithm off of the passed in jwk.
- if not algorithm and isinstance(key_data, dict):
- algorithm = key_data.get("alg", None)
-
- if not algorithm:
- raise JWKError("Unable to find an algorithm for key")
-
- key_class = get_key(algorithm)
- if not key_class:
- raise JWKError("Unable to find an algorithm for key")
- return key_class(key_data, algorithm)
|
