Handle invalid User situation

Author: andreagrandi

Diff for: rest_framework/authtoken/management/commands/drf_create_token.py

@@ -16,5 +16,11 @@ def add_arguments(self, parser):
 
     def handle(self, *args, **options):
         username = options['username']
-        token = self.create_user_token(username)
+
+        try:
+            token = self.create_user_token(username)
+        except User.DoesNotExist:
+            print('Cannot create the Token: user {0} does not exist'.format(
+                username
+            ))
         print('Generated token {0} for user {1}'.format(token.key, username))

Diff for: tests/test_authtoken.py

@@ -48,3 +48,7 @@ def test_command_create_user_token(self):
         assert token is not None
         token_saved = Token.objects.first()
         assert token.key == token_saved.key
+
+    def test_command_create_user_token_invalid_user(self):
+        with pytest.raises(User.DoesNotExist):
+            AuthTokenCommand().create_user_token('not_existing_user')