From 178993d9f1da3b1bfade0b3fca076bd069936115 Mon Sep 17 00:00:00 2001 From: Ryan Leckey Date: Mon, 4 Nov 2013 13:54:48 -0800 Subject: [PATCH] Fix name of postgresql dialect. --- sqlalchemy_utils/functions/database.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/sqlalchemy_utils/functions/database.py b/sqlalchemy_utils/functions/database.py index be2ebe6..7871e4b 100644 --- a/sqlalchemy_utils/functions/database.py +++ b/sqlalchemy_utils/functions/database.py @@ -1,6 +1,6 @@ from sqlalchemy.engine.url import make_url import sqlalchemy as sa -from sqlalchemy.exc import ProgrammingError +from sqlalchemy.exc import ProgrammingError, OperationalError import os @@ -14,7 +14,7 @@ def database_exists(url): engine = sa.create_engine(url) - if engine.dialect.name == 'postgres': + if engine.dialect.name == 'postgresql': text = "SELECT 1 FROM pg_database WHERE datname='%s'" % database return bool(engine.execute(text).scalar()) @@ -34,7 +34,7 @@ def database_exists(url): engine.execute(text) return True - except ProgrammingError: + except (ProgrammingError, OperationalError): return False @@ -50,7 +50,7 @@ def create_database(url, encoding='utf8'): engine = sa.create_engine(url) - if engine.dialect.name == 'postgres': + if engine.dialect.name == 'postgresql': text = "CREATE DATABASE %s ENCODING = '%s'" % (database, encoding) engine.execute(text)