django rest framework - aggiunta a views.obtain_auth_token

Question

Ho implementato l'autenticazione token con il framework django rest e posso inserire nome utente e password a /api-token-auth/ e ottenere il token.

url(r'^api-token-auth/', token_views.obtain_auth_token) 

Oltre al token, voglio ottenere l'oggetto User relativi al token restituito.

Come posso eseguire l'override/aggiunta a questa vista e anche restituire l'oggetto Utente effettivo?

Answer 1

È possibile trovare la relativa vista qui:

https://github.com/tomchristie/django-rest-framework/blob/master/rest_framework/authtoken/views.py#L21

Supponendo che hai creato una sorta di serializzatore utente già, si può sostanzialmente prendere l'istanza user lì e spingere nella tua UserSerializer. quindi aggiungilo alla risposta, qualcosa come il seguente.

... 
user_serializer = UserSerializer(user) 
return Response({'token': token.key, 'user': user_serializer.data})