Fix for Python 3 compatibility

This commit is contained in:
Neal Todd
2014-10-24 17:01:02 +01:00
parent 88c11492f5
commit 6293c35ac0

View File

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