From 7451e16588d2de6bce18fcc60b3be654ee591ac1 Mon Sep 17 00:00:00 2001 From: Nate Case Date: Fri, 22 Aug 2008 15:58:05 -0500 Subject: [PATCH] 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 Signed-off-by: Jeremy Kerr --- apps/patchwork/models.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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 -- 2.39.2