from django.core.management.base import BaseCommand

from login.tests.model_maker import make_superuser
from login.tests.model_maker import make_user


class Command(BaseCommand):

    help = "Create demo data for 'login'"

    def handle(self, *args, **options):
        make_superuser('admin', 'admin')
        make_user('staff', 'staff', is_staff=True)
        make_user('web', 'web')
        print("Created 'login' demo data...")

from django.core.management.base import BaseCommand

from login.tests.scenario import (
    user_contractor,
    user_default,
)


class Command(BaseCommand):

    help = "Create demo data for 'login'"

    def handle(self, *args, **options):
        user_contractor()
        user_default()
        print("Created 'login' demo data...")

Use the standard scenario for demo data
Use the standard scenario for demo data

  from django.core.management.base import BaseCommand
  
- from login.tests.model_maker import make_superuser
- from login.tests.model_maker import make_user
+ from login.tests.scenario import (
+     user_contractor,
+     user_default,
+ )
  
  
  class Command(BaseCommand):
  
      help = "Create demo data for 'login'"
  
      def handle(self, *args, **options):
+         user_contractor()
+         user_default()
-         make_superuser('admin', 'admin')
-         make_user('staff', 'staff', is_staff=True)
-         make_user('web', 'web')
          print("Created 'login' demo data...")
  
Use the standard scenario for demo data
## Code Before:
from django.core.management.base import BaseCommand

from login.tests.model_maker import make_superuser
from login.tests.model_maker import make_user


class Command(BaseCommand):

    help = "Create demo data for 'login'"

    def handle(self, *args, **options):
        make_superuser('admin', 'admin')
        make_user('staff', 'staff', is_staff=True)
        make_user('web', 'web')
        print("Created 'login' demo data...")

## Instruction:
Use the standard scenario for demo data
## Code After:
from django.core.management.base import BaseCommand

from login.tests.scenario import (
    user_contractor,
    user_default,
)


class Command(BaseCommand):

    help = "Create demo data for 'login'"

    def handle(self, *args, **options):
        user_contractor()
        user_default()
        print("Created 'login' demo data...")
