Forked mumble-django project from https://bitbucket.org/Svedrin/mumble-django
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
65 lines
1.8 KiB
65 lines
1.8 KiB
# -*- 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 )
|