Store access token id in session data

users/models.py

@@ -202,6 +202,10 @@ class AccessToken(models.Model):
 
         return user
 
+    @classmethod
+    def get_token(cls, request):
+        return AccessToken.objects.get(pk=request.session['access_token_id'])
+
     class Meta:
         verbose_name = _('access token')
         verbose_name_plural = _('access tokens')

users/views.py

@@ -37,6 +37,8 @@ class AuthorizeView(RedirectView):
                 datetime.fromtimestamp(token['expires_at'])),
         )
         user = token_obj.fetch_user(True)
+        self.request.session['access_token_id'] = token_obj.id
+        self.request.session.save()
         login(self.request, user)
         return reverse('index')