From: Thomas Fillon <thomas@parisson.com>
Date: Thu, 19 Mar 2015 14:26:00 +0000 (+0100)
Subject: Fix missing items from sub-selections in timeside.server.models
X-Git-Url: https://git.parisson.com/?a=commitdiff_plain;h=d96caa0e90c7ed239e330167d24a1dd629598c98;p=timeside.git

Fix missing items from sub-selections in timeside.server.models
---

diff --git a/timeside/server/models.py b/timeside/server/models.py
index 6c08e1f..36226f3 100644
--- a/timeside/server/models.py
+++ b/timeside/server/models.py
@@ -147,6 +147,13 @@ class Selection(DocBaseResource):
         verbose_name = _('selection')
 
 
+    def get_all_items(self):
+        qs_items = self.items.all()
+        for selection in self.selections.all():
+            qs_items |= selection.get_all_items()
+        return qs_items
+
+
 class Item(DocBaseResource):
 
     element_type = 'timeside_item'
@@ -337,8 +344,8 @@ class Task(BaseResource):
 
     def run(self, streaming=False):
         self.status_setter(_RUNNING)
-
-        for item in self.selection.items.all():
+        
+        for item in self.selection. get_all_items():
             item_path = item.get_results_path()
             if not os.path.exists(item_path):
                 os.makedirs(item_path)