From 8769277aed6540a8e795d3c923c8644782c2ee3c Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Tue, 5 Apr 2016 15:13:29 +0200
Subject: [PATCH] [experiments] Fixes experiment reset() method after API
 changes

---
 beat/web/experiments/models.py | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/beat/web/experiments/models.py b/beat/web/experiments/models.py
index 2a80a2665..ebd24c08e 100644
--- a/beat/web/experiments/models.py
+++ b/beat/web/experiments/models.py
@@ -661,13 +661,16 @@ class Experiment(Shareable):
     def reset(self):
         """Resets an experiment so it can be run again"""
 
-        for block in self.blocks.all():
-            block.results.all().delete()
-            block.delete()
+        self.blocks.update(
+            status=Block.NOT_CACHED,
+            start_date=None,
+            end_date=None,
+            runnable_date=None,
+            )
 
         self.start_date = None
-        self.end_date   = None
-        self.status     = self.PENDING
+        self.end_date = None
+        self.status = self.PENDING
 
         # reset sharing state
         self.sharing    = Shareable.PRIVATE
@@ -675,8 +678,7 @@ class Experiment(Shareable):
         self.shared_with_team.clear()
 
         # remove associated attestations
-        if self.has_attestation():
-            self.attestation.all().delete()
+        if self.has_attestation(): self.attestation.all().delete()
 
         self.save()
 
-- 
GitLab