From 34c38e0cfe5e880e678704c4d473f082787fca64 Mon Sep 17 00:00:00 2001 From: Andrea Grandi Date: Sat, 3 Jun 2017 11:06:14 +0100 Subject: [PATCH] Use self.sdtout and CommandError to print output --- .../management/commands/drf_create_token.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/rest_framework/authtoken/management/commands/drf_create_token.py b/rest_framework/authtoken/management/commands/drf_create_token.py index 2dc333a6b..8fe47c44a 100644 --- a/rest_framework/authtoken/management/commands/drf_create_token.py +++ b/rest_framework/authtoken/management/commands/drf_create_token.py @@ -1,5 +1,5 @@ from django.contrib.auth import get_user_model -from django.core.management.base import BaseCommand +from django.core.management.base import BaseCommand, CommandError from rest_framework.authtoken.models import Token @@ -23,7 +23,9 @@ class Command(BaseCommand): try: token = self.create_user_token(username) except UserModel.DoesNotExist: - print('Cannot create the Token: user {0} does not exist'.format( - username - )) - print('Generated token {0} for user {1}'.format(token.key, username)) + raise CommandError( + 'Cannot create the Token: user {0} does not exist'.format( + username) + ) + self.stdout.write( + 'Generated token {0} for user {1}'.format(token.key, username))