diff --git a/users/models.py b/users/models.py index c4b2f37..0751d56 100644 --- a/users/models.py +++ b/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') diff --git a/users/views.py b/users/views.py index 65e91b8..731b15e 100644 --- a/users/views.py +++ b/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')