From 25a3f57493fb065f7b9b99060ad9e25959c13b51 Mon Sep 17 00:00:00 2001 From: Sergei Azarkin Date: Tue, 12 Sep 2017 14:47:15 +0300 Subject: [PATCH] fixup! Fix authtoken managment command username param --- .../management/commands/drf_create_token.py | 23 +++++++++---------- 1 file changed, 11 insertions(+), 12 deletions(-) diff --git a/rest_framework/authtoken/management/commands/drf_create_token.py b/rest_framework/authtoken/management/commands/drf_create_token.py index 9f1d3563f..da10bfc90 100644 --- a/rest_framework/authtoken/management/commands/drf_create_token.py +++ b/rest_framework/authtoken/management/commands/drf_create_token.py @@ -19,7 +19,7 @@ class Command(BaseCommand): return token[0] def add_arguments(self, parser): - parser.add_argument('username', type=str, nargs='+') + parser.add_argument('username', type=str) parser.add_argument( '-r', @@ -31,16 +31,15 @@ class Command(BaseCommand): ) def handle(self, *args, **options): - usernames = options['username'] + username = options['username'] reset_token = options['reset_token'] - for username in usernames: - try: - token = self.create_user_token(username, reset_token) - except UserModel.DoesNotExist: - 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)) + try: + token = self.create_user_token(username, reset_token) + except UserModel.DoesNotExist: + 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))