# -*- mode: python; coding: utf-8 -*- # Copyright (C) 2017-2019 by BDE ENS Paris-Saclay # SPDX-License-Identifier: GPL-3.0-or-later from datetime import datetime from authlib.integrations.django_client import OAuth from django.contrib.auth.models import Group from django.urls import reverse from django.utils import timezone from django.views.generic import RedirectView from rest_framework import viewsets from users.models import User, AccessToken from .serializers import GroupSerializer, UserSerializer class LoginView(RedirectView): def get_redirect_url(self, *args, **kwargs): oauth = OAuth() oauth.register('notekfet') redirect_url = self.request.build_absolute_uri(reverse('users:auth')) return oauth.notekfet.authorize_redirect(self.request, redirect_url).url class AuthorizeView(RedirectView): def get_redirect_url(self, *args, **kwargs): oauth = OAuth() oauth.register('notekfet') token = oauth.notekfet.authorize_access_token(self.request) token_obj = AccessToken.objects.create( access_token=token['access_token'], expires_in=token['expires_in'], scopes=token['scope'], refresh_token=token['refresh_token'], expires_at=timezone.utc.fromutc( datetime.fromtimestamp(token['expires_at'])), ) # TODO Log in or create user return '/' class UserViewSet(viewsets.ModelViewSet): """ API endpoint that allows users to be viewed or edited. """ queryset = User.objects.all() serializer_class = UserSerializer class GroupViewSet(viewsets.ModelViewSet): """ API endpoint that allows groups to be viewed or edited. """ queryset = Group.objects.all() serializer_class = GroupSerializer