From: Nate Case <ncase@xes-inc.com>
Date: Fri, 22 Aug 2008 20:58:05 +0000 (-0500)
Subject: Support postgresql_psycopg2 DATABASE_ENGINE
X-Git-Url: https://git.ozlabs.org/?p=patchwork;a=commitdiff_plain;h=7451e16588d2de6bce18fcc60b3be654ee591ac1

Support postgresql_psycopg2 DATABASE_ENGINE

Check for "postgresql*" rather than just "postgresql" in
HashField.db_type() so that postgresql_psycopg2 will work.

While we're here, raise an exception if the database engine is
unknown rather than silently not returning anything.

Signed-off-by: Nate Case <ncase@xes-inc.com>
Signed-off-by: Jeremy Kerr <jk@ozlabs.org>
---

diff --git a/apps/patchwork/models.py b/apps/patchwork/models.py
index 6d331f3..11457d7 100644
--- a/apps/patchwork/models.py
+++ b/apps/patchwork/models.py
@@ -194,10 +194,13 @@ class HashField(models.Field):
             n_bytes = len(hashlib.new(self.algorithm).digest())
         else:
             n_bytes = len(self.hash_constructor().digest())
-        if settings.DATABASE_ENGINE == 'postgresql':
+        if settings.DATABASE_ENGINE.startswith('postgresql'):
             return 'bytea'
         elif settings.DATABASE_ENGINE == 'mysql':
             return 'binary(%d)' % n_bytes
+        else:
+            raise Exception("Unknown database engine '%s'" % \
+                            settings.DATABASE_ENGINE)
 
     def to_python(self, value):
         return value