Get or create a user and query Note Kfet to get data

users/models.py

@@ -4,10 +4,12 @@
 
 from datetime import datetime
 
+import requests
 from authlib.integrations.django_client import OAuth
 from django.conf import settings
 from django.contrib.auth.models import AbstractUser
 from django.db import models
+from django.db.models import Q
 from django.utils import timezone
 from django.utils.translation import gettext_lazy as _
 from med.settings import MAX_EMPRUNT
@@ -49,6 +51,23 @@ class User(AbstractUser):
         # FIXME Use NK20
         return True
 
+    def update_data(self, data: dict):
+        """
+        Update user data from given dictionary.
+        Useful when we want to update user data from Note Kfet.
+
+        Parameters
+        ----------
+        data : dict
+            Dictionary with user data to update.
+        """
+        self.email = data['email']
+        self.first_name = data['first_name']
+        self.last_name = data['last_name']
+        self.telephone = data['profile']['phone_number']
+        self.address = data['profile']['address']
+        self.comment = data['profile']['section']
+
 
 class AccessToken(models.Model):
     owner = models.ForeignKey(
@@ -103,6 +122,48 @@ class AccessToken(models.Model):
 
         self.save()
 
+    def refresh_if_expired(self):
+        """
+        Refresh the current token if it is invalid.
+        """
+        if self.expires_at < timezone.now():
+            self.refresh()
+
+    def auth_header(self):
+        """
+        Return HTTP header that contains the bearer access token.
+        Refresh the token if needed.
+        """
+        self.refresh_if_expired()
+        return {'Authorization': f'Bearer {self.access_token}'}
+
+    def fetch_user(self, create_if_not_exist: bool = False):
+        """
+        Extract information about the Note Kfet API by using the current
+        access token.
+        """
+        if self.owner:
+            return self.owner
+
+        data = requests.get('https://note-dev.crans.org/api/me/',
+                            headers=self.auth_header()).json()
+        username = data['username']
+        email = data['email']
+        qs = User.objects.filter(Q(username=username) | Q(email=email))
+        if not qs.exists():
+            if create_if_not_exist:
+                user = User.objects.create(username=username, email=email)
+            else:
+                return None
+        else:
+            user = qs.get()
+
+        # Update user data from Note Kfet
+        user.update_data(data)
+        user.save()
+
+        return user
+
     class Meta:
         verbose_name = _('access token')
         verbose_name_plural = _('access tokens')