Creating custom commands in django

Question

I have the following user model,

class User(AbstractBaseUser, PermissionsMixin):
    email = models.EmailField(unique=True, max_length=255)
    mobile = PhoneNumberField(null=True)
    username = models.CharField(null=False, unique=True, max_length=255)
    full_name = models.CharField(max_length=255, blank=True, null=True)
    is_bot = models.BooleanField(default=False)

I want to create a custom command which could work like createsuperuser and creates a bot.

I have created a management package in the relevant app and added a command package inside that and a file createbot.py inside of that.

This is my code inside createbot.py

class Command(BaseCommand):
    def handle(self, email, username=None, password=None):
        user = User.objects.create(email,
                                   username=username,

                                   password=password,
                                   is_staff=True,
                                   is_superuser=True,
                                   is_active=True,
                                   is_bot=True
                                   )
        self.stdout.write(self.style.SUCCESS('Successfully create user bot with id: {}, email: {}'.format(user.id, user.email)))

I want this to work exactly like createsuper user giving me prompts to enter email, name and the works. But when I run it, I get the following,

TypeError: handle() got an unexpected keyword argument 'verbosity'

How can I get this to work?

How did you invoke the command? What did you write on the command line? — willeM_ Van Onsem
– willeM_ Van Onsem, Commented May 25, 2018 at 21:22
Note, you need to be using create_user not just create, otherwise the password won't be hashed. — Daniel Roseman
– Daniel Roseman, Commented May 25, 2018 at 21:35

willeM_ Van Onsem · Accepted Answer · 2018-05-25 21:42:58Z

4

Like is specified in the documentation on creating custom commands:

In addition to being able to add custom command line options, all management commands can accept some default options such as --verbosity and --traceback.

So that means the handle(..) function is invoked with those parameters, even if you are not interested in these.

You can however easily catch those and ignore them, by making using of keyword arguments:

class Command(BaseCommand):

    def handle(self, email, username=None, password=None, **other):
        # ...
        # perform actions
        pass

Here other is a dictionary that maps strings to values: the parameters with which the function is called, but that are not explicitly mentioned in the signature of the function.

The documentation also mentions how to specify the parameters you want to use in the handle, such that helptext can be generated when the user requests how to use the custom command. You can for example write:

class Command(BaseCommand):

    def add_arguments(self, parser):
        # Positional arguments
        parser.add_argument('email', required=True)

        # Named (optional) arguments
        parser.add_argument(
            '--username',
            help='The username for the user',
        )
        parser.add_argument(
            '--password',
            help='The password for the user',
        )

    def handle(self, email, username=None, password=None, **other):
        # ...
        # perform actions
        pass

Note that passwords are hashed in Django, hence you should use create_user(..).

edited May 25, 2018 at 21:42

answered May 25, 2018 at 21:25

willeM_ Van Onsem

482k33 gold badges483 silver badges624 bronze badges

Sign up to request clarification or add additional context in comments.

5 Comments

Melissa Stewart Over a year ago

How do I invoke this on the command line. I can't get it to work.

willeM_ Van Onsem Over a year ago

@MelissaStewart: manage.py your_command email --username foo --password somepwd, note that you need to hash the password.

Melissa Stewart Over a year ago

Van Osem I also have a theoretical question, any help on that will be super appreciated. stackoverflow.com/questions/50536804/…

Melissa Stewart Over a year ago

Also one more question, this creates an user with id : None Why is that happening?

willeM_ Van Onsem Over a year ago

@MelissaStewart: then the database will normally assign a valid id for it.

Collectives™ on Stack Overflow

Creating custom commands in django

1 Answer 1

5 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

5 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related