Rearranged the way the SHA hash functions were laid out.

1 files changed, 11 insertions, 6 deletions

diff --git a/jwt/algorithms.py b/jwt/algorithms.py
index d3bc11d..720d675 100644
--- a/jwt/algorithms.py
+++ b/jwt/algorithms.py
@@ -83,11 +83,13 @@ class HMACAlgorithm(Algorithm):
     Performs signing and verification operations using HMAC
     and the specified hash function.
     """
+    SHA256 = hashlib.sha256
+    SHA384 = hashlib.sha384
+    SHA512 = hashlib.sha512
+
     def __init__(self, hash_alg):
         self.hash_alg = hash_alg
 
-    SHA256, SHA384, SHA512 = hashlib.sha256, hashlib.sha384, hashlib.sha512
-
     def prepare_key(self, key):
         if not isinstance(key, string_types) and not isinstance(key, bytes):
             raise TypeError('Expecting a string- or bytes-formatted key.')
@@ -110,12 +112,13 @@ if has_crypto:
         Performs signing and verification operations using
         RSASSA-PKCS-v1_5 and the specified hash function.
         """
+        SHA256 = hashes.SHA256
+        SHA384 = hashes.SHA384
+        SHA512 = hashes.SHA512
 
         def __init__(self, hash_alg):
             self.hash_alg = hash_alg()
 
-        SHA256, SHA384, SHA512 = hashes.SHA256, hashes.SHA384, hashes.SHA512
-
         def prepare_key(self, key):
             if isinstance(key, interfaces.RSAPrivateKey) or \
                isinstance(key, interfaces.RSAPublicKey):
@@ -166,11 +169,13 @@ if has_crypto:
         Performs signing and verification operations using
         ECDSA and the specified hash function
         """
+        SHA256 = hashes.SHA256
+        SHA384 = hashes.SHA384
+        SHA512 = hashes.SHA512
+
         def __init__(self, hash_alg):
             self.hash_alg = hash_alg()
 
-        SHA256, SHA384, SHA512 = hashes.SHA256, hashes.SHA384, hashes.SHA512
-
         def prepare_key(self, key):
             if isinstance(key, interfaces.EllipticCurvePrivateKey) or \
                isinstance(key, interfaces.EllipticCurvePublicKey):