add connectDatasetDPStep & connectDPStepAutoproc & storeDPStep

dataproc/api/views/input_view.py

@@ -15,6 +15,8 @@ from api.models.datacollection_model import DataCollection
 from api.models.construct_model import Construct
 from api.models.user_model import User
 from api.models.ocf_model import OCF
+from api.models.dpstep_model import DPStep
+from api.models.autoPROC_model import autoPROC
 
 # Activate logger
 logger=logging.getLogger('dataproc')
@@ -47,6 +49,7 @@ def storeInput(request):
             storeParseComputationHost(json_data_computationhost)
             storeParseDataCollection(json_data_datacollection)
             storeParseOCF(json_data_ocf)
+            dpstep = storeDPStep()
 
             # Register relationships 
             connectConstructUser(json_data_construct, json_data_user)
@@ -55,6 +58,7 @@ def storeInput(request):
             connectDatasetConstruct(json_data_dataset, json_data_construct)
             connectDatasetStorageHost(json_data_dataset, json_data_storagehost)
             connectDataCollectionDataset(json_data_datacollection, json_data_dataset)
+            connectDatasetDPStep(json_data_dataset, dpstep)
 
             for input_ocf in json_data_ocf:
                 connectConstructOCF(json_data_construct, input_ocf)
@@ -64,6 +68,44 @@ def storeInput(request):
         except :
             return JsonResponse({"STATUS":"ERROR OCCURRED"}, safe=False)
 
+@csrf_exempt
+def connectDatasetDPStep(data1, dpstep):
+    
+    """
+    Create a relationship between a dataset and a construct
+    """
+
+    try:
+        dataset=Dataset.nodes.get(uuid=data1["uuid"])
+        dataset.input_of.connect(dpstep)
+        return dataset
+
+    except:
+        return False
+
+
+@csrf_exempt
+def connectDPStepAutoproc(dpstep):
+    
+    """
+    Create a relationship between a dp step and the autoPROC node
+    """
+    
+    try:
+        autoproc = autoPROC.nodes.first()
+        dpstep.with_autoproc.connect(autoproc)
+        return dpstep
+
+    except:
+        return False
+
+@csrf_exempt
+def storeDPStep():
+    dpstep=DPStep()
+    dpstep.save()
+    dpstep=connectDPStepAutoproc(dpstep)
+    return dpstep
+
 @csrf_exempt
 def storeParseDataset(data):