From 85576d64c928c6b825f99bd100fd0ddcafd15c09 Mon Sep 17 00:00:00 2001
From: Mattia Mancini <mancini@astron.nl>
Date: Fri, 6 Jul 2018 12:15:48 +0000
Subject: [PATCH] Story LSMR-13: adding viewsets and urls for
 WorkRequest/WorkRequestBlueprint

---
 SAS/LSMR/src/lsmr/lsmrapp/viewsets/specification.py | 10 ++++++++++
 SAS/LSMR/src/lsmr/urls.py                           |  2 ++
 2 files changed, 12 insertions(+)

diff --git a/SAS/LSMR/src/lsmr/lsmrapp/viewsets/specification.py b/SAS/LSMR/src/lsmr/lsmrapp/viewsets/specification.py
index 21ff677625b..39373de61f0 100644
--- a/SAS/LSMR/src/lsmr/lsmrapp/viewsets/specification.py
+++ b/SAS/LSMR/src/lsmr/lsmrapp/viewsets/specification.py
@@ -61,6 +61,16 @@ class RunViewSet(viewsets.ModelViewSet):
     serializer_class = serializers.RunSerializer
 
 
+class WorkRequestBlueprintViewSet(viewsets.ModelViewSet):
+    queryset = models.WorkRequestBlueprint.objects.all()
+    serializer_class = serializers.WorkRequestBlueprintSerializer
+
+
+class WorkRequestViewSet(viewsets.ModelViewSet):
+    queryset = models.WorkRequest.objects.all()
+    serializer_class = serializers.WorkRequestSerializer
+
+
 class WorkRequestRelationBlueprintViewSet(viewsets.ModelViewSet):
     queryset = models.WorkRequestRelationBlueprint.objects.all()
     serializer_class = serializers.WorkRequestRelationBlueprintSerializer
diff --git a/SAS/LSMR/src/lsmr/urls.py b/SAS/LSMR/src/lsmr/urls.py
index 033fc7f376b..2420d5e95fa 100644
--- a/SAS/LSMR/src/lsmr/urls.py
+++ b/SAS/LSMR/src/lsmr/urls.py
@@ -49,6 +49,8 @@ router.register(r'work_io_roles', viewsets.WorkIORolesViewSet)
 router.register(r'cycle', viewsets.CycleViewSet)
 router.register(r'project', viewsets.ProjectViewSet)
 router.register(r'run_set', viewsets.RunSetViewSet)
+router.register(r'work_request_blueprint', viewsets.WorkRequestBlueprintViewSet)
+router.register(r'work_request', viewsets.WorkRequestViewSet)
 router.register(r'work_request_relation_blueprint', viewsets.WorkRequestRelationBlueprintViewSet)
 router.register(r'work_request_relation', viewsets.WorkRequestRelationViewSet)
 
-- 
GitLab