From c8fbaf6e4b8c62cf7542fd0433dc6010097103e9 Mon Sep 17 00:00:00 2001
From: arifiorino <arifiorino@gmail.com>
Date: Tue, 10 Sep 2019 03:13:01 +0000
Subject: [PATCH] Added if statements to run algorithms

---
 server/website/website/views.py | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/server/website/website/views.py b/server/website/website/views.py
index 82d9e79..65bbe14 100644
--- a/server/website/website/views.py
+++ b/server/website/website/views.py
@@ -532,13 +532,16 @@ def handle_result_files(session, files):
         response = chain(aggregate_target_results.s(result.pk),
                          map_workload.s(),
                          configuration_recommendation.s()).apply_async()
-    elif session.algorithm == AlgorithmType.ALGORITHM1:
+    elif session.algorithm == AlgorithmType.DDPG:
+        response = chain(train_ddpg.s(result.pk),
+                         configuration_recommendation_ddpg.s()).apply_async()
+    elif session.algorithm == AlgorithmType.DNN:
         pass
-    elif session.algorithm == AlgorithmType.ALGORITHM2:
-        pass
-    elif session.algorithm == AlgorithmType.ALGORITHM3:
-        pass
-    taskmeta_ids = [response.parent.parent.id, response.parent.id, response.id]
+    taskmeta_ids = []
+    current_task = response
+    while current_task:
+        taskmeta_ids.append(current_task.id)
+        current_task = current_task.parent
     result.task_ids = ','.join(taskmeta_ids)
     result.save()
     return HttpResponse("Result stored successfully! Running tuner...(status={})  Result ID:{} "