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]
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))