[Fixed]-How does one use Django custom management command option?

19đź‘Ť

âś…

You can do it like this:

from optparse import make_option

class Command(BaseCommand):
    option_list = BaseCommand.option_list + (
        make_option('--del',
            action='store_true',
            help='Delete poll'),
        make_option('--close',
            action='store_true',
            help='Close poll'),
    )

    def handle(self, close, *args, **kwargs):
        del_ = kwargs.get('del')

Do note that some keywords in Python are reserved so you can handle those using **kwargs. Otherwise you can use normal arguments (like I did with close)

👤Wolph

7đź‘Ť

A little suggestion about defining the commands (key name, dest) and handling default values (in make_option and in the command):

class Command(BaseCommand):
    option_list = BaseCommand.option_list + (
        make_option('--del',
            action='store_true',
            help='Delete all polls.',
            dest='your_name_for_delete',
            default=False),
        make_option('--close',
            action='store_true',
            help='Close all polls.'),
    )

    def handle(self, close, *args, **options):
        if options.get('your_name_for_delete'):
            Poll.objects.delete()
        if options.get('close', False):
            Poll.objects.update(closed=True)

In Django code you’ll find “keyword arguments” (**kwargs) often named as **options, which is more suggestive (I stick to this naming convention).

The default value can be either specified in make_option, or through the dict.get method,
which allows for a default value.

There’s no reason not u use both defaults, should your Command.handle method be called manually, where the **options dictionary could be missing this entry.

👤Tomasz Gandor

Leave a comment