commit 11e8cc7cf05967f203d5e7084756ccc28e43bdf7
author David Reid <dreid@dreid.org> Wed Nov 13 10:37:47 2013 -0800
committer David Reid <dreid@dreid.org> Wed Nov 13 10:37:47 2013 -0800
tree 00ffb908a4b86f3965f98fc3a5627061b2a5bdc3
parent 6392a9c63ce134c4aceefb8a4eb9da2fa7f4f390

Import AlreadyFinalized instead of exceptions.

cryptography/hazmat/primitives/hashes.py

diff --git a/cryptography/hazmat/primitives/hashes.py b/cryptography/hazmat/primitives/hashes.py
index f85d36a..adf64d3 100644
--- a/cryptography/hazmat/primitives/hashes.py
+++ b/cryptography/hazmat/primitives/hashes.py
@@ -15,7 +15,7 @@
 
 import six
 
-from cryptography import exceptions
+from cryptography.exceptions import AlreadyFinalized
 from cryptography.hazmat.primitives import interfaces
 
 
@@ -39,14 +39,14 @@
 
     def update(self, data):
         if self._ctx is None:
-            raise exceptions.AlreadyFinalized()
+            raise AlreadyFinalized()
         if isinstance(data, six.text_type):
             raise TypeError("Unicode-objects must be encoded before hashing")
         self._ctx.update(data)
 
     def copy(self):
         if self._ctx is None:
-            raise exceptions.AlreadyFinalized()
+            raise AlreadyFinalized()
         return Hash(
             self.algorithm, backend=self._backend, ctx=self._ctx.copy()
         )

tests/hazmat/primitives/test_hashes.py

diff --git a/tests/hazmat/primitives/test_hashes.py b/tests/hazmat/primitives/test_hashes.py
index a5c440b..4c64460 100644
--- a/tests/hazmat/primitives/test_hashes.py
+++ b/tests/hazmat/primitives/test_hashes.py
@@ -19,7 +19,7 @@
 
 import six
 
-from cryptography import exceptions
+from cryptography.exceptions import AlreadyFinalized
 from cryptography.hazmat.bindings import _default_backend
 from cryptography.hazmat.primitives import hashes
 
@@ -56,10 +56,10 @@
         h = hashes.Hash(hashes.SHA1())
         h.finalize()
 
-        with pytest.raises(exceptions.AlreadyFinalized):
+        with pytest.raises(AlreadyFinalized):
             h.update(b"foo")
 
-        with pytest.raises(exceptions.AlreadyFinalized):
+        with pytest.raises(AlreadyFinalized):
             h.copy()