entrouvert
/
debian-django-tenant-schemas
17
0
Fork 0
Code Issues Pull Requests Releases Activity

#424: Added IF EXISTS to DROP SCHEMA queries

This commit is contained in:
Alexander van Eck 2017-01-03 17:34:06 +01:00 committed by Gary Reynolds
parent 93a0780c34
commit 3579735cb3
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tenant_schemas/models.py
View File

@ -67,7 +67,7 @@ class TenantMixin(models.Model):
if schema_exists(self.schema_name) and (self.auto_drop_schema or force_drop):
cursor = connection.cursor()
cursor.execute('DROP SCHEMA %s CASCADE' % self.schema_name)
cursor.execute('DROP SCHEMA IF EXISTS %s CASCADE' % self.schema_name)
super(TenantMixin, self).delete(*args, **kwargs)

2
tenant_schemas/test/cases.py
View File

@ -22,7 +22,7 @@ class TenantTestCase(TestCase):
cls.tenant.delete()
cursor = connection.cursor()
cursor.execute('DROP SCHEMA test CASCADE')
cursor.execute('DROP SCHEMA IF EXISTS test CASCADE')
@classmethod
def sync_shared(cls):

2
tenant_schemas/tests/testcases.py
View File

@ -25,7 +25,7 @@ class BaseTestCase(TestCase):
# a blank public schema for this set of tests.
connection.set_schema_to_public()
cursor = connection.cursor()
cursor.execute('DROP SCHEMA %s CASCADE; CREATE SCHEMA %s;'
cursor.execute('DROP SCHEMA IF EXISTS %s CASCADE; CREATE SCHEMA %s;'
% (get_public_schema_name(), get_public_schema_name()))
super(BaseTestCase, cls).setUpClass()