diff options
| author | José Padilla <jpadilla@webapplicate.com> | 2015-03-17 15:25:12 -0400 |
|---|---|---|
| committer | José Padilla <jpadilla@webapplicate.com> | 2015-03-17 15:25:12 -0400 |
| commit | 1e6b6c588d4cee1a1e44c380a79f7dace0ecad67 (patch) | |
| tree | 0ad329d3648a374d86a8508b582df26dc22e7483 /jwt/algorithms.py | |
| parent | d47163117bef52392f314406ad0a4177e2a65e16 (diff) | |
| parent | a72cf6df7a35e32562cb02aa17369cdc913834c9 (diff) | |
| download | pyjwt-1e6b6c588d4cee1a1e44c380a79f7dace0ecad67.tar.gz | |
Merge pull request #101 from mark-adams/pyjwt-obj
Refactor api.py functions into an object (PyJWT)
Diffstat (limited to 'jwt/algorithms.py')
| -rw-r--r-- | jwt/algorithms.py | 23 |
1 files changed, 11 insertions, 12 deletions
diff --git a/jwt/algorithms.py b/jwt/algorithms.py index 720d675..bd3595a 100644 --- a/jwt/algorithms.py +++ b/jwt/algorithms.py @@ -1,7 +1,6 @@ import hashlib import hmac -from .api import register_algorithm from .compat import constant_time_compare, string_types, text_type try: @@ -18,23 +17,23 @@ except ImportError: has_crypto = False -def _register_default_algorithms(): +def _register_default_algorithms(pyjwt_obj): """ Registers the algorithms that are implemented by the library. """ - register_algorithm('none', NoneAlgorithm()) - register_algorithm('HS256', HMACAlgorithm(HMACAlgorithm.SHA256)) - register_algorithm('HS384', HMACAlgorithm(HMACAlgorithm.SHA384)) - register_algorithm('HS512', HMACAlgorithm(HMACAlgorithm.SHA512)) + pyjwt_obj.register_algorithm('none', NoneAlgorithm()) + pyjwt_obj.register_algorithm('HS256', HMACAlgorithm(HMACAlgorithm.SHA256)) + pyjwt_obj.register_algorithm('HS384', HMACAlgorithm(HMACAlgorithm.SHA384)) + pyjwt_obj.register_algorithm('HS512', HMACAlgorithm(HMACAlgorithm.SHA512)) if has_crypto: - register_algorithm('RS256', RSAAlgorithm(RSAAlgorithm.SHA256)) - register_algorithm('RS384', RSAAlgorithm(RSAAlgorithm.SHA384)) - register_algorithm('RS512', RSAAlgorithm(RSAAlgorithm.SHA512)) + pyjwt_obj.register_algorithm('RS256', RSAAlgorithm(RSAAlgorithm.SHA256)) + pyjwt_obj.register_algorithm('RS384', RSAAlgorithm(RSAAlgorithm.SHA384)) + pyjwt_obj.register_algorithm('RS512', RSAAlgorithm(RSAAlgorithm.SHA512)) - register_algorithm('ES256', ECAlgorithm(ECAlgorithm.SHA256)) - register_algorithm('ES384', ECAlgorithm(ECAlgorithm.SHA384)) - register_algorithm('ES512', ECAlgorithm(ECAlgorithm.SHA512)) + pyjwt_obj.register_algorithm('ES256', ECAlgorithm(ECAlgorithm.SHA256)) + pyjwt_obj.register_algorithm('ES384', ECAlgorithm(ECAlgorithm.SHA384)) + pyjwt_obj.register_algorithm('ES512', ECAlgorithm(ECAlgorithm.SHA512)) class Algorithm(object): |