Add command to create missing tenant schemas

refs #5106
This commit is contained in:
Benjamin Dauvergne 2014-08-20 14:57:02 +02:00
parent ccec1ff41f
commit 3afe385b85
1 changed files with 20 additions and 0 deletions

View File

@ -0,0 +1,20 @@
from django.core.management.base import BaseCommand
from entrouvert.djommon.multitenant.middleware import TenantMiddleware
from django.db import connection
class Command(BaseCommand):
help = "Create schemas"
def handle(self, *args, **options):
verbosity = int(options.get('verbosity'))
connection.set_schema_to_public()
all_tenants = TenantMiddleware.get_tenants()
for tenant in all_tenants:
if verbosity >= 1:
print
print self.style.NOTICE("=== Creating schema ") \
+ self.style.SQL_TABLE(tenant.schema_name)
if not tenant.create_schema(check_if_exists=True):
print self.style.ERROR(' Nothing to do')