mumble-django/pyweb/mumble/management/update_schema.py

# -*- coding: utf-8 -*-

"""
 *  Copyright © 2010, Michael "Svedrin" Ziegler <diese-addy@funzt-halt.net>
 *
 *  Mumble-Django is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This package is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
"""

import os
from os.path		import join

from django.db		import connection, transaction
from django.db.models	import signals
from django.conf	import settings

from mumble		import models
from mumble.management.server_detect import find_existing_instances


def update_schema( **kwargs ):
	scriptpath = join(
		settings.MUMBLE_DJANGO_ROOT, "pyweb", "mumble", "conversionsql", {
			'postgresql_psycopg2': 'pgsql',
			'postgresql': 'pgsql',
			'mysql':      'mysql',
			'sqlite3':    'sqlite',
			}[settings.DATABASE_ENGINE]
		)
	
	scripts = [ filename for filename in os.listdir( scriptpath ) if filename.endswith( ".sql" ) ]
	scripts.sort()
	
	for filename in scripts:
		cursor = connection.cursor()
		
		scriptfile = open( os.path.join( scriptpath, filename ), "r" )
		try:
			stmt = scriptfile.read()
			cursor.execute( stmt )
			if "postgresql" in settings.DATABASE_ENGINE:
				transaction.commit()
		
		except IOError, err:
			print "Error reading file '%s':" % filename
			print err
		
		except cursor.db.connection.Error, err:
			print "Error executing file '%s':" % filename
			print err
			if "postgresql" in settings.DATABASE_ENGINE:
				transaction.rollback()
		
		finally:
			scriptfile.close()
			cursor.close()
	
	find_existing_instances( **kwargs )