[Feture] 添加ops 页面

apps/ops/api.py

@@ -1,15 +1,42 @@
 # ~*~ coding: utf-8 ~*~
 
 
+from django.shortcuts import get_object_or_404
 from rest_framework import viewsets
 
 from .hands import IsSuperUser
-from .models import AdHoc
-from .serializers import TaskSerializer
+from .models import Task, AdHoc, AdHocRunHistory
+from .serializers import TaskSerializer, AdHocSerializer, AdHocRunHistorySerializer
 
 
 class TaskViewSet(viewsets.ModelViewSet):
-    queryset = AdHoc.objects.all()
+    queryset = Task.objects.all()
     serializer_class = TaskSerializer
     permission_classes = (IsSuperUser,)
 
+
+class AdHocViewSet(viewsets.ModelViewSet):
+    queryset = AdHoc.objects.all()
+    serializer_class = AdHocSerializer
+    permission_classes = (IsSuperUser,)
+
+    def get_queryset(self):
+        task_id = self.request.query_params.get('task')
+        if task_id:
+            task = get_object_or_404(Task, id=task_id)
+            self.queryset = self.queryset.filter(task=task)
+        return self.queryset
+
+
+class AdHocRunHistorySet(viewsets.ModelViewSet):
+    queryset = AdHocRunHistory.objects.all()
+    serializer_class = AdHocRunHistorySerializer
+    permission_classes = (IsSuperUser,)
+
+    def get_queryset(self):
+        task_id = self.request.query_params.get('task')
+        if task_id:
+            task = get_object_or_404(Task, id=task_id)
+            adhocs = task.adhoc.all()
+            self.queryset = self.queryset.filter(adhoc__in=adhocs)
+        return self.queryset