Request input from django custom command?

Question

I've made a custom command in django to delete a set account in CMD, but I want to be able to run the python file and have the command line ask me for the account number to delete then have it delete. This is what I have so far.

from django.core.management.base import BaseCommand, CommandError
from accounts.models import client

class Command(BaseCommand):

    args = '<client_id client_id ...>'
    help = 'Closes the specified account.'

    def handle(self, *args, **options):
        for client_id in args:
            try:
                x = client.objects.get(pk=int(client_id))
            except client.DoesNotExist:
                raise CommandError('Client "%s" does no exist' % client_id)

            x.delete()

            self.stdout.write('Successfully closed account "%s"' % client_id)

score 21 · Accepted Answer · edited Aug 04 '21 at 05:18

21

Use the built-in raw_input() function:

def handle(self, *args, **options):
    if args:
        ids = args
    else:
        ids = raw_input('Enter comma-delimited list of ids: ').split(',')
    for client_id in ids:
        ...

Edit: In Python3, use input() instead of raw_input()

edited Aug 04 '21 at 05:18

Anas Imran Tasadduq

584
9
26

answered Feb 03 '15 at 14:36

catavaran

44,703
8
98
85

11

`input()` in Python 3 http://stackoverflow.com/questions/954834/how-do-i-use-raw-input-in-python-3 – Ciro Santilli OurBigBook.com Jun 05 '16 at 09:49
NameError: name 'raw_input' is not defined – Al Mahdi Dec 16 '20 at 19:28

Request input from django custom command?

1 Answers1