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

Support PostgreSQL <9.2 where the pg_stat_activity.pid column was named pg_stat_activity.procpid.

Browse Source
This commit is contained in:
Neal Todd
2014-10-24 16:27:43 +01:00
parent 63ef4407c2
commit 88c11492f5
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
sqlalchemy_utils/functions/database.py
View File

@@ -399,12 +399,14 @@ def drop_database(url):
engine.raw_connection().set_isolation_level(ISOLATION_LEVEL_AUTOCOMMIT)
# Disconnect all users from the database we are dropping.
version = map(int, engine.execute('SHOW server_version;').first()[0].split('.'))
pid_column = 'pid' if (version[0] >= 9 and version[1] >= 2) else 'procpid'
text = '''
SELECT pg_terminate_backend(pg_stat_activity.pid)
SELECT pg_terminate_backend(pg_stat_activity.%(pid_column)s)
FROM pg_stat_activity
WHERE pg_stat_activity.datname = '%s'
AND pid <> pg_backend_pid()
''' % database
WHERE pg_stat_activity.datname = '%(database)s'
AND %(pid_column)s <> pg_backend_pid();
''' % {'pid_column': pid_column, 'database': database}
engine.execute(text)
# Drop the database.