review comments

Qiskit · Sep 3, 2024 · 3ebcb12 · 3ebcb12
1 parent 739b194
commit 3ebcb12
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 2 deletions.
diff --git a/gateway/api/serializers.py b/gateway/api/serializers.py
@@ -168,7 +168,6 @@ class JobSerializer(serializers.ModelSerializer):
 
     class Meta:
         model = Job
-        fields = ["id", "result", "status", "program", "created"]
 
 
 class RunProgramSerializer(serializers.Serializer):

diff --git a/gateway/api/v1/views.py b/gateway/api/v1/views.py
@@ -38,6 +38,10 @@ def get_serializer_run_program(*args, **kwargs):
     def get_serializer_run_job(*args, **kwargs):
         return v1_serializers.RunJobSerializer(*args, **kwargs)
 
+    @staticmethod
+    def get_serializer_job(*args, **kwargs):
+        return v1_serializers.JobSerializer(*args, **kwargs)
+
     @swagger_auto_schema(
         operation_description="List author Qiskit Functions",
         responses={status.HTTP_200_OK: v1_serializers.ProgramSerializer(many=True)},

diff --git a/gateway/api/views.py b/gateway/api/views.py
@@ -106,6 +106,14 @@ def get_serializer_run_job(*args, **kwargs):
 
         return RunJobSerializer(*args, **kwargs)
 
+    @staticmethod
+    def get_serializer_job(*args, **kwargs):
+        """
+        This method returns the job serializer
+        """
+
+        return JobSerializer(*args, **kwargs)
+
     def get_serializer_class(self):
         return self.serializer_class
 
@@ -413,7 +421,7 @@ def get_jobs(
                 jobs = Job.objects.filter(program=program)
             else:
                 jobs = Job.objects.filter(program=program, author=request.user)
-            serializer = JobSerializer(jobs, many=True)
+            serializer = self.get_serializer_job(jobs, many=True)
             return Response(serializer.data)