feat: qrcode data checker

apps/api/viewsets.py

@@ -2,18 +2,22 @@
 # SPDX-License-Identifier: GPL-3.0-or-later
 
 import re
+import pyotp
+import base64
+import os
 
 from django.contrib.contenttypes.models import ContentType
 from django_filters.rest_framework import DjangoFilterBackend
 from django.db.models import Q
 from django.conf import settings
 from django.contrib.auth.models import User
-from rest_framework.viewsets import ReadOnlyModelViewSet, ModelViewSet
+from rest_framework.viewsets import ReadOnlyModelViewSet, ModelViewSet, GenericViewSet
+from rest_framework.response import Response
 from permission.backends import PermissionBackend
 from note.models import Alias
 
 from .filters import RegexSafeSearchFilter
-from .serializers import UserSerializer, ContentTypeSerializer
+from .serializers import UserSerializer, ContentTypeSerializer, QRCodeCheckSerializer
 
 
 def is_regex(pattern):
@@ -124,3 +128,17 @@ class ContentTypeViewSet(ReadOnlyModelViewSet):
     filter_backends = [DjangoFilterBackend, RegexSafeSearchFilter]
     filterset_fields = ['id', 'app_label', 'model', ]
     search_fields = ['$app_label', '$model', ]
+
+
+class QRCodeVerificationViewSet(GenericViewSet):
+    serializer_class = QRCodeCheckSerializer
+    queryset = User.objects.none()
+
+    def get_view_name(self):
+        return "Vérification QR Code"
+
+    def create(self, request, *args, **kwargs):
+        serializer = self.get_serializer(data=request.data)
+        serializer.is_valid(raise_exception=True)
+        secret = base64.b32encode(os.getenv("DJANGO_SECRET_KEY").encode())
+        return Response({'valid': pyotp.TOTP(secret, interval=30).verify(serializer.validated_data['data'][-6:])})