Improve classmethod hash

xymy · Mar 16, 2020 · 0dae526 · 0dae526
1 parent 10efac5
commit 0dae526
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/xycrypto/hashes.py b/xycrypto/hashes.py
@@ -52,9 +52,9 @@ def copy(self):
     def hash(cls, data, **kwargs):
         """Return hash of data from byte string or unicode string."""
 
-        ctx = cls(**kwargs)
         if isinstance(data, str):
             data = data.encode('utf-8')
+        ctx = cls(**kwargs)
         ctx.update(data)
         return ctx.finalize()
 

diff --git a/xycrypto/hmac.py b/xycrypto/hmac.py
@@ -68,9 +68,9 @@ def copy(self):
     def hash(cls, hash_cls, key, data, **kwargs):
         """Return hash of data from byte string or unicode string."""
 
-        ctx = cls(hash_cls, key, **kwargs)
         if isinstance(data, str):
             data = data.encode('utf-8')
+        ctx = cls(hash_cls, key, **kwargs)
         ctx.update(data)
         return ctx.finalize()