Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 7 additions & 1 deletion lucterios/install/lucterios_admin.py
Original file line number Diff line number Diff line change
Expand Up @@ -548,6 +548,12 @@ def write_setting_(self):
def _sql_to_clear(self, only_delete):
from django.db import connection
option = 'CASCADE'
# disable foreign keys
if self.database[0] == 'sqlite':
sql_set_foreign_key = 'PRAGMA foreign_keys = OFF;'
else:
sql_set_foreign_key = 'SET foreign_key_checks = 0;'

if self.database[0] == 'postgresql':
if only_delete:
sql_cmd = 'DELETE FROM "%s" %s;'
Expand All @@ -560,7 +566,7 @@ def _sql_to_clear(self, only_delete):
sql_cmd = 'DROP TABLE IF EXISTS %s %s;'
try:
with connection.cursor() as curs:
curs.execute('SET foreign_key_checks = 0;')
curs.execute(sql_set_foreign_key)
except Exception:
option = 'CASCADE'
return sql_cmd, option
Expand Down