change User to get_user_model()

This commit is contained in:
danillab 2019-04-15 11:41:33 +07:00 committed by GitHub
parent db65282163
commit f010240d04
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -184,10 +184,10 @@ Note that you'll want to ensure you place this code snippet in an installed `mod
If you've already created some users, you can generate tokens for all existing users like this:
from django.contrib.auth.models import User
from django.contrib.auth import get_user_model
from rest_framework.authtoken.models import Token
for user in User.objects.all():
for user in get_user_model().objects.all():
Token.objects.get_or_create(user=user)
##### By exposing an api endpoint