Created serializer + viewset, also fixed up permission settings (Next…

API/API/settings.py

@@ -113,8 +113,11 @@
 
 REST_FRAMEWORK = {
     "DEFAULT_PERMISSION_CLASSES": [
+        "rest_framework.permissions.IsAuthenticated",
+    ],
+    "DEFAULT_AUTHENTICATION_CLASSES": (
         "rest_framework_simplejwt.authentication.JWTAuthentication",
-    ]
+    ),
 }
 # Internationalization
 # https://docs.djangoproject.com/en/5.1/topics/i18n/

API/API/urls.py

@@ -14,12 +14,19 @@
     1. Import the include() function: from django.urls import include, path
     2. Add a URL to urlpatterns:  path('blog/', include('blog.urls'))
 """
+from back_api.views import AgentTypeViewSet
 from django.contrib import admin
 from django.urls import path
+from rest_framework.routers import DefaultRouter
 from rest_framework_simplejwt.views import TokenObtainPairView, TokenRefreshView
 
+router = DefaultRouter()
+router.register(r"agents", AgentTypeViewSet, basename="agent")
+
 urlpatterns = [
     path("admin/", admin.site.urls),
     path("api/token/", TokenObtainPairView.as_view(), name="token_obtain_pair"),
     path("api/token/refresh/", TokenRefreshView.as_view(), name="token_refresh"),
 ]
+
+urlpatterns += router.urls

API/back_api/serializers.py

@@ -0,0 +1,43 @@
+from back_api.models import AgentType, OSINTQuery
+from rest_framework import serializers
+from user_model.models import CustomUserClass
+
+
+class AgentTypeSerializer(serializers.Serializer):
+    vendor_name = serializers.CharField(max_length=30)
+    vendor_description = serializers.CharField(max_length=200)
+    vendor_api = serializers.CharField(max_length=100)
+
+    def create(self, validated_data):
+        return AgentType(**validated_data)
+
+    def update(self, instance, validated_data):
+        instance.name = validated_data.get("vendor_name", instance.name)
+        instance.description = validated_data.get(
+            "vendor_description", instance.description
+        )
+        instance.api = validated_data.get("vendor_api", instance.api)
+        return instance
+
+
+class OSINTQuerySerializer(serializers.Serializer):
+    osint_agent = serializers.DjangoModelField(AgentType)
+    osint_result = serializers.DictField()
+    created_by = serializers.DjangoModelField(CustomUserClass)
+    created_date = serializers.DateTimeField()
+    updated_date = serializers.DateTimeField()
+
+    def create(self, validated_data):
+        return OSINTQuery(**validated_data)
+
+    def update(self, instance, validated_data):
+        instance.agent = validated_data.get("osint_agent", instance.agent)
+        instance.result = validated_data.get("osint_result", instance.result)
+        instance.created_by = validated_data.get("created_by", instance.created_by)
+        instance.created_date = validated_data.get(
+            "created_date", instance.created_date
+        )
+        instance.updated_date = validated_data.get(
+            "updated_date", instance.updated_date
+        )
+        return instance

API/back_api/views.py

@@ -1,3 +1,24 @@
-# from django.shortcuts import render
+from back_api.models import AgentType
+from back_api.serializers import AgentTypeSerializer
+from django.shortcuts import get_object_or_404
+from rest_framework import viewsets
+from rest_framework.response import Response
 
-# Create your views here.
+# from user_model.models import CustomUserClass
+
+
+class AgentTypeViewSet(viewsets.ViewSet):
+    """
+    A ViewSet for listing or retrieving different agents.
+    """
+
+    def list(self, request):
+        queryset = AgentType.objects.all()
+        serializer = AgentTypeSerializer(queryset, many=True)
+        return Response(serializer.data)
+
+    def retrieve(self, request, pk=None):
+        queryset = AgentType.objects.all()
+        agent = get_object_or_404(queryset, pk=pk)
+        serializer = AgentTypeSerializer(agent)
+        return Response(serializer.data)