Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from django.core.management.base import BaseCommand, CommandError
- from django.contrib.auth.models import User
- class Command(BaseCommand):
- help = 'Create a super user'
- def handle(self, *args, **options):
- if len(args) != 2:
- raise CommandError('need exactly two arguments for username and password')
- username, password = args
- u, created = User.objects.get_or_create(username=username)
- if created:
- u.is_superuser = True
- u.is_staff = True
- u.set_password(password)
- u.save()
- else:
- raise CommandError("user '%s' already exist" % username)
- return "Password changed successfully for user '%s'" % u.username
- usage: manage.py createmysuperuser [-h] [--version] [-v {0,1,2,3}]
- [--settings SETTINGS]
- [--pythonpath PYTHONPATH] [--traceback]
- [--no-color]
- manage.py createmysuperuser: error: unrecognized arguments: myuser mypassword
- class Command(BaseCommand):
- ...
- def add_arguments(self, parser):
- parser.add_argument('username')
- parser.add_argument('password')
- def handle(self, *args, **options):
- username = options['username']
- password = options['password']
- ...
- class Command(BaseCommand):
- def add_arguments(self, parser):
- parser.add_argument('args')
- def handle(self, *args, **options):
- if len(args) != 2:
- ...
- from django.core.management.base import BaseCommand
- class Command(BaseCommand):
- def add_arguments(self, parser):
- parser.add_argument('username')
- parser.add_argument('password')
- def handle(self, *args, **options):
- username = options['username']
- password = options['password']
- return u'Username: %s Password: %s' % (username, password)
- def add_arguments(self, parser):
- parser.add_argument('args', nargs='*')
Add Comment
Please, Sign In to add comment