encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-08-04 12:30:11 +03:00
Code Issues Packages Projects Releases Wiki Activity

fixup! Fix authtoken managment command username param

Browse Source
This commit is contained in:
Sergei Azarkin 2017-09-12 14:47:15 +03:00
parent f98641ef97
commit 25a3f57493
1 changed files with 11 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
rest_framework/authtoken/management/commands/drf_create_token.py
View File

@ -19,7 +19,7 @@ class Command(BaseCommand):
return token[0] return token[0]
def add_arguments(self, parser): def add_arguments(self, parser):
parser.add_argument('username', type=str, nargs='+') parser.add_argument('username', type=str)
parser.add_argument( parser.add_argument(
'-r', '-r',
@ -31,16 +31,15 @@ class Command(BaseCommand):
) )
def handle(self, *args, **options): def handle(self, *args, **options):
usernames = options['username'] username = options['username']
reset_token = options['reset_token'] reset_token = options['reset_token']
for username in usernames: try:
try: token = self.create_user_token(username, reset_token)
token = self.create_user_token(username, reset_token) except UserModel.DoesNotExist:
except UserModel.DoesNotExist: raise CommandError(
raise CommandError( 'Cannot create the Token: user {0} does not exist'.format(
'Cannot create the Token: user {0} does not exist'.format( username)
username) )
) self.stdout.write(
self.stdout.write( 'Generated token {0} for user {1}'.format(token.key, username))
'Generated token {0} for user {1}'.format(token.key, username))