Add HashField.construct() method

To construct a new hash object for the given algorithm. While we're at
it, clean up the hashlib-wrapping code.

Signed-off-by: Jeremy Kerr <jk@ozlabs.org>

1 files changed, 10 insertions, 10 deletions

diff --git a/apps/patchwork/models.py b/apps/patchwork/models.py
index 62ce592..226a69c 100644
--- a/apps/patchwork/models.py
+++ b/apps/patchwork/models.py
@@ -176,19 +176,19 @@ class HashField(models.CharField):
         self.algorithm = algorithm
         try:
             import hashlib
-            self.hashlib = True
+            def _construct(string = ''):
+                return hashlib.new(self.algorithm, string)
+            self.construct = _construct
             self.n_bytes = len(hashlib.new(self.algorithm).hexdigest())
         except ImportError:
-            self.hashlib = False
-            if algorithm == 'sha1':
-                import sha
-                hash_constructor = sha.new
-            elif algorithm == 'md5':
-                import md5
-                hash_constructor = md5.new
-            else:
+            modules = { 'sha1': 'sha', 'md5': 'md5'}
+
+            if algorithm not in modules.keys():
                 raise NameError("Unknown algorithm '%s'" % algorithm)
-            self.n_bytes = len(hash_constructor().hexdigest())
+
+            self.construct = __import__(modules[algorithm]).new
+
+        self.n_bytes = len(self.construct().hexdigest())
 
         kwargs['max_length'] = self.n_bytes
         super(HashField, self).__init__(*args, **kwargs)