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

5
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,10 +31,9 @@ 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: