From e090ab896c0443f5967301a034ad53735efe83d6 Mon Sep 17 00:00:00 2001 From: Ernesto Rico-Schmidt Date: Thu, 2 Jun 2011 22:23:36 +0200 Subject: added createuser command. 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.' -- cgit v0.10.2