openstack/deb-python-sqlalchemy-utils
Code Issues Proposed changes

Ensure isolation level is autocommit for postgres.

Browse Source
This commit is contained in:
Ryan Leckey
2013-11-04 14:26:46 -08:00
parent 7e14156379
commit eb53f7b31e
1 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
sqlalchemy_utils/functions/database.py
View File

@@ -52,6 +52,11 @@ def create_database(url, encoding='utf8'):
engine = sa.create_engine(url) engine = sa.create_engine(url)
if engine.dialect.name == 'postgresql': if engine.dialect.name == 'postgresql':
if engine.driver == 'psycopg2':
from psycopg2.extensions import ISOLATION_LEVEL_AUTOCOMMIT
engine.raw_connection().set_isolation_level(
ISOLATION_LEVEL_AUTOCOMMIT)
text = "CREATE DATABASE %s ENCODING = '%s'" % (database, encoding) text = "CREATE DATABASE %s ENCODING = '%s'" % (database, encoding)
engine.execute(text) engine.execute(text)
@@ -82,6 +87,12 @@ def drop_database(url):
if engine.dialect.name == 'sqlite' and url.database != ':memory:': if engine.dialect.name == 'sqlite' and url.database != ':memory:':
os.remove(url.database) os.remove(url.database)
elif engine.dialect.name == 'postgresql' and engine.driver == 'psycopg2':
from psycopg2.extensions import ISOLATION_LEVEL_AUTOCOMMIT
engine.raw_connection().set_isolation_level(ISOLATION_LEVEL_AUTOCOMMIT)
text = "DROP DATABASE %s" % database
engine.execute(text)
else: else:
text = "DROP DATABASE %s" % database text = "DROP DATABASE %s" % database
engine.execute(text) engine.execute(text)