From 3a793f91a0c5dcab8de3808281e6b22db0dce62d Mon Sep 17 00:00:00 2001
From: Jorrit Schaap <schaap@astron.nl>
Date: Thu, 3 Dec 2020 16:02:29 +0100
Subject: [PATCH] TMSS-320: added method get_transformed_input_dataproduct
---
SAS/TMSS/src/tmss/tmssapp/models/scheduling.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/SAS/TMSS/src/tmss/tmssapp/models/scheduling.py b/SAS/TMSS/src/tmss/tmssapp/models/scheduling.py
index 38f26a66ab3..af6094a67bd 100644
--- a/SAS/TMSS/src/tmss/tmssapp/models/scheduling.py
+++ b/SAS/TMSS/src/tmss/tmssapp/models/scheduling.py
@@ -225,6 +225,10 @@ class Subtask(BasicCommon):
'''
return Dataproduct.objects.filter(producer__subtask_id=self.id)
+ def get_transformed_input_dataproduct(self, output_dataproduct_id: int) -> 'Dataproduct':
+ '''return the transformed input dataproduct for the given output_dataproduct_id.'''
+ return self.input_dataproducts.get(consumers__output_id=output_dataproduct_id)
+
def get_transformed_output_dataproduct(self, input_dataproduct_id: int) -> 'Dataproduct':
'''return the transformed output dataproduct for the given input_dataproduct_id.'''
return self.output_dataproducts.get(producers__input_id=input_dataproduct_id)
--
GitLab