diff options
| -rw-r--r-- | program/management/commands/createuser.py | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/program/management/commands/createuser.py b/program/management/commands/createuser.py new file mode 100644 index 0000000..a356bb3 --- /dev/null +++ b/program/management/commands/createuser.py @@ -0,0 +1,25 @@ +from django.contrib.auth.models import User +from django.core.management.base import BaseCommand, CommandError + +from optparse import make_option + +class Command(BaseCommand): + help = 'creates an user' + option_list = BaseCommand.option_list + ( + make_option('--username', dest='username', default=None, help='Specifies the username.'), + make_option('--email', dest='email', default=None, help='Specifies the email address.'), + ) + + def handle(self, *args, **options): + username = options.get('username', None) + email = options.get('email', None) + + if not username or not email: + raise CommandError("You must use --username and --email.") + try: + User.objects.get(username=username) + except User.DoesNotExist: + User.objects.create_user(username=username, email=email) + print 'user created successfully.' + else: + print 'User already exists, no need to create.' |