Andreas/evennia
1
0
Fork 0
mirror of https://github.com/evennia/evennia.git synced 2026-03-21 23:36:30 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update migration with mysql/postgreql exception

Browse source
This commit is contained in:
Griatch 2017-09-30 15:53:34 +02:00
parent 291fd6fa34
commit 9f6f16fb7d
1 changed files with 20 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

21
evennia/typeclasses/migrations/0010_delete_old_player_tables.py
View file

@ -4,6 +4,7 @@ from __future__ import unicode_literals
from django.db import migrations, connection
_ENGINE = None
def _table_exists(db_cursor, tablename):
"Returns bool if table exists or not"
@ -11,8 +12,26 @@ def _table_exists(db_cursor, tablename):
def _drop_table(db_cursor, table_name):
global _ENGINE
if not _ENGINE:
from django.conf import settings
try:
_ENGINE = settings.DATABASES["default"]["ENGINE"]
except KeyError:
_ENGINE = settings.DATABASE_ENGINE
if _table_exists(db_cursor, table_name):
sql_drop = "DROP TABLE %s;" % table_name
if _ENGINE == "django.db.backends.mysql":
sql_drop = "SET FOREIGN_KEY_CHECKS=0;"\
"DROP TABLE {table};"\
"SET FOREIGN_KEY_CHECKS=1;".format(table=table_name)
elif _ENGINE == "postgresql_psycopg2":
sql_drop = "ALTER TABLE {table} DISABLE TRIGGER ALL;"\
"DROP TABLE {table};"\
"ALTER TABLE {table} ENABLE TRIGGER ALL;".format(table=table_name)
else: # sqlite3, other databases
sql_drop = "DROP TABLE {table};".format(table=table_name)
db_cursor.execute(sql_drop)