from django.core.management.commands import loaddata
from django.core.management.base import CommandError
from django.db import DatabaseError

from optparse import make_option

from ...models import Schema, template_schema

class Command(loaddata.Command):
    option_list = loaddata.Command.option_list + (
        make_option('--schema', action='store', dest='schema',
            help='Specify which schema to load schema-aware models to',
            default='__template__',
        ),
    )
    
    def handle(self, *app_labels, **options):
        schema_name = options.get('schema')
        if schema_name == '__template__':
            # Hmm, we don't want to accidentally write data to this, so
            # we should raise an exception if we are going to be
            # writing any schema-aware objects.
            schema = None
        else:
            try:
                schema = Schema.objects.get(schema=options.get('schema'))
            except Schema.DoesNotExist:
                raise CommandError('No Schema found named "%s"' % schema_name)
        
            schema.activate()
        
             super(Command, self).handle(*app_labels, **options)

        if schema:
            schema.deactivate()

from django.core.management.commands import loaddata
from django.core.management.base import CommandError
from django.db import DatabaseError

from optparse import make_option

from ...models import Schema, template_schema

class Command(loaddata.Command):
    option_list = loaddata.Command.option_list + (
        make_option('--schema', action='store', dest='schema',
            help='Specify which schema to load schema-aware models to',
            default='__template__',
        ),
    )
    
    def handle(self, *app_labels, **options):
        schema_name = options.get('schema')
        if schema_name == '__template__':
            # Hmm, we don't want to accidentally write data to this, so
            # we should raise an exception if we are going to be
            # writing any schema-aware objects.
            schema = None
        else:
            try:
                schema = Schema.objects.get(schema=options.get('schema'))
            except Schema.DoesNotExist:
                raise CommandError('No Schema found named "%s"' % schema_name)
        
            schema.activate()
        
        super(Command, self).handle(*app_labels, **options)

        if schema:
            schema.deactivate()
        
        
        for schema in Schema.objects.all():
            schema.create_schema()

Fix indenting. Create any schemas that were just loaded.
Fix indenting.
Create any schemas that were just loaded.

  from django.core.management.commands import loaddata
  from django.core.management.base import CommandError
  from django.db import DatabaseError
  
  from optparse import make_option
  
  from ...models import Schema, template_schema
  
  class Command(loaddata.Command):
      option_list = loaddata.Command.option_list + (
          make_option('--schema', action='store', dest='schema',
              help='Specify which schema to load schema-aware models to',
              default='__template__',
          ),
      )
      
      def handle(self, *app_labels, **options):
          schema_name = options.get('schema')
          if schema_name == '__template__':
              # Hmm, we don't want to accidentally write data to this, so
              # we should raise an exception if we are going to be
              # writing any schema-aware objects.
              schema = None
          else:
              try:
                  schema = Schema.objects.get(schema=options.get('schema'))
              except Schema.DoesNotExist:
                  raise CommandError('No Schema found named "%s"' % schema_name)
          
              schema.activate()
          
-              super(Command, self).handle(*app_labels, **options)
+         super(Command, self).handle(*app_labels, **options)
  
          if schema:
              schema.deactivate()
+         
+         
+         for schema in Schema.objects.all():
+             schema.create_schema()
  
Fix indenting. Create any schemas that were just loaded.
## Code Before:
from django.core.management.commands import loaddata
from django.core.management.base import CommandError
from django.db import DatabaseError

from optparse import make_option

from ...models import Schema, template_schema

class Command(loaddata.Command):
    option_list = loaddata.Command.option_list + (
        make_option('--schema', action='store', dest='schema',
            help='Specify which schema to load schema-aware models to',
            default='__template__',
        ),
    )
    
    def handle(self, *app_labels, **options):
        schema_name = options.get('schema')
        if schema_name == '__template__':
            # Hmm, we don't want to accidentally write data to this, so
            # we should raise an exception if we are going to be
            # writing any schema-aware objects.
            schema = None
        else:
            try:
                schema = Schema.objects.get(schema=options.get('schema'))
            except Schema.DoesNotExist:
                raise CommandError('No Schema found named "%s"' % schema_name)
        
            schema.activate()
        
             super(Command, self).handle(*app_labels, **options)

        if schema:
            schema.deactivate()

## Instruction:
Fix indenting. Create any schemas that were just loaded.
## Code After:
from django.core.management.commands import loaddata
from django.core.management.base import CommandError
from django.db import DatabaseError

from optparse import make_option

from ...models import Schema, template_schema

class Command(loaddata.Command):
    option_list = loaddata.Command.option_list + (
        make_option('--schema', action='store', dest='schema',
            help='Specify which schema to load schema-aware models to',
            default='__template__',
        ),
    )
    
    def handle(self, *app_labels, **options):
        schema_name = options.get('schema')
        if schema_name == '__template__':
            # Hmm, we don't want to accidentally write data to this, so
            # we should raise an exception if we are going to be
            # writing any schema-aware objects.
            schema = None
        else:
            try:
                schema = Schema.objects.get(schema=options.get('schema'))
            except Schema.DoesNotExist:
                raise CommandError('No Schema found named "%s"' % schema_name)
        
            schema.activate()
        
        super(Command, self).handle(*app_labels, **options)

        if schema:
            schema.deactivate()
        
        
        for schema in Schema.objects.all():
            schema.create_schema()
