- feat: add logout endpoint

Author: koladev32

api/authentication/viewsets/__init__.py

@@ -1,3 +1,4 @@
 from .register import RegisterViewSet
 from .login import LoginViewSet
 from .active_session import ActiveSessionViewSet
+from .logout import LogoutViewSet

api/authentication/viewsets/logout.py

@@ -0,0 +1,20 @@
+from rest_framework import viewsets, mixins
+from rest_framework.response import Response
+from rest_framework import status
+from rest_framework.permissions import IsAuthenticated
+
+from api.authentication.models import ActiveSession
+
+
+class LogoutViewSet(viewsets.GenericViewSet, mixins.CreateModelMixin):
+    permission_classes = (IsAuthenticated,)
+
+    def create(self, request, *args, **kwargs):
+        user = request.user
+
+        session = ActiveSession.objects.get(user=user)
+        session.delete()
+
+        return Response({
+            "success": True
+        }, status=status.HTTP_200_OK)

api/routers.py

@@ -1,4 +1,4 @@
-from api.authentication.viewsets import RegisterViewSet, LoginViewSet, ActiveSessionViewSet
+from api.authentication.viewsets import RegisterViewSet, LoginViewSet, ActiveSessionViewSet, LogoutViewSet
 from rest_framework import routers
 from api.user.viewsets import UserViewSet
 
@@ -12,6 +12,8 @@
 
 router.register(r'checkSession', ActiveSessionViewSet, basename='check-session')
 
+router.register(r'logout', LogoutViewSet, basename='logout')
+
 urlpatterns = [
     *router.urls,
 ]