Asking the User for Confirmation¶
Suppose you want to confirm an action before actually executing it. Add the following to your command:
def handle(self): if not self.confirm('Continue with this action?', False): return
In this case, the user will be asked “Continue with this action?”.
If the user answers with
y it returns
False if they answer with
The second argument to
confirm() is the default value to return if the user doesn’t enter any valid input.
If the second argument is not provided,
True is assumed.
You can customize the regex used to check if the answer means “yes” in the third argument of the
For instance, to allow anything that starts with either
j, you would set it to:
self.confirm('Continue with this action?', False, '(?i)^(y|j)')
The regex defaults to
Asking the User for Information¶
You can also ask a question with more than a simple yes/no answer. For instance, if you want to know a user name, you can add this to your command:
def handle(self): name = self.ask('Please enter your name', 'John Doe')
The user will be asked “Please enter your name”.
They can type some name which will be returned by the
If they leave it empty, the default value (
John Doe here) is returned.
Let the User Choose from a List of Answers¶
If you have a predefined set of answers the user can choose from,
you could use a
ChoiceQuestion or the
choice() method which
makes sure that the user can only enter a valid string from a predefined list:
def handle(self): color = self.choice( 'Please select your favorite color (defaults to red)', ['red', 'blue', 'yellow'], 0 ) self.line('You have just selected: %s' % color)
The option which should be selected by default is provided with the third argument.
The default is
None, which means that no option is the default one.
If the user enters an invalid string, an error message is shown
and the user is asked to provide the answer another time,
until they enter a valid string or reach the maximum number of attempts.
The default value for the maximum number of attempts is
None, which means infinite number of attempts.
Sometimes, multiple answers can be given. The
provides this feature using comma separated values.
This is disabled by default, to enable this use the
multiple keyword if using the
multiselect attribute if using the
def handle(self): colors = self.choice( 'Please select your favorite color (defaults to red and blue), ['red', 'blue', 'yellow'], '0,1' multiple=True ) self.line('You have just selected: %s' % ', '.join(colors))
Now, when the user enters
1,2, the result will be:
You have just selected: blue, yellow.
If the user does not enter anything, the result will be:
You have just selected: red, blue.
You can also specify an array of potential answers for a given question. These will be autocompleted as the user types:
def handle(self): names = ['John', 'Jane', 'Paul'] question = self.create_question('Please enter a name', default='John') question.set_autocomplete_values(names) name = self.ask(question)
Hiding the User’s Response¶
You can also ask a question and hide the response. This is particularly convenient for passwords:
def handle(self): password = self.secret('What is the database password?')
Validating the Answer¶
You can even validate the answer. For instance, you might only accept integers:
def handle(self): question = self.create_question('Choose a number') question.set_validator(int) question.set_max_attempts(2) number = self.ask(question)
validator a callback which handles the validation.
It should throw an exception if there is something wrong.
The exception message is displayed in the console, so it is a good practice to put some useful information in it.
The validator or the callback function should also return the value of the user’s input if the validation was successful.
You can set the max number of times to ask with the
If you reach this max number it will use the default value.
None means the amount of attempts is infinite.
The user will be asked as long as they provide an invalid answer
and will only be able to proceed if their input is valid.
Testing a Command that Expects Input¶
If you want to write a unit test for a command which expects some kind of input from the command line, you need to set the helper input stream:
def test_execute_command(self): command_tester = CommandTester(command) # Equals to a user inputting "Test" and hitting ENTER # If you need to enter a confirmation, "yes\n" will work command_tester.execute(inputs="Test\n")