From 498ad82d2cc8582d3139bf69a0fe333d6b425668 Mon Sep 17 00:00:00 2001
From: gebele <gebele@in-silico.ch>
Date: Thu, 8 Aug 2019 12:44:29 +0000
Subject: ensure dataset parser errors not hidden in a task; check upload by
 first header; remove_task_data, tasks, prediction dataset, training dataset;
 js code refinement

---
 helper.rb | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

(limited to 'helper.rb')

diff --git a/helper.rb b/helper.rb
index cb77ffd..82482d4 100644
--- a/helper.rb
+++ b/helper.rb
@@ -23,4 +23,25 @@ helpers do
     self.match(/^[a-f\d]{24}$/i) ? true : false
   end
 
+  def remove_task_data(pid)
+    task = Task.find_by(:pid => pid)
+    if task and !task.subTasks.blank?
+      task.subTasks.each_with_index do |task_id,idx|
+        t = Task.find task_id
+        predictionDataset = Dataset.find t.dataset_id if t.dataset_id
+        if predictionDataset && idx == 0
+          trainingDataset = Dataset.find predictionDataset.source
+          source = trainingDataset.source
+          trainingDataset.delete
+          File.delete File.join(source) if File.exists? File.join(source)
+          predictionDataset.delete
+        elsif predictionDataset
+          predictionDataset.delete
+        end
+        t.delete
+      end
+    end
+    task.delete if task
+  end
+
 end
-- 
cgit v1.2.3