From d3bcf2cdd3f4e448c219d5935ff7de3dd85298b3 Mon Sep 17 00:00:00 2001
From: Jakub Trllo <jakub.trllo@gmail.com>
Date: Wed, 10 Apr 2019 11:34:48 +0200
Subject: [PATCH] changed get children method

---
 pype/ftrack/actions/action_create_folders.py | 22 ++++++++++++--------
 1 file changed, 13 insertions(+), 9 deletions(-)

diff --git a/pype/ftrack/actions/action_create_folders.py b/pype/ftrack/actions/action_create_folders.py
index 3af05a21e8..515d6e08d0 100644
--- a/pype/ftrack/actions/action_create_folders.py
+++ b/pype/ftrack/actions/action_create_folders.py
@@ -35,15 +35,6 @@ class CreateFolders(BaseAction):
             return False
         return True
 
-    def getShotAsset(self, entity):
-        if entity not in self.importable:
-            if entity['object_type']['name'] != 'Task':
-                self.importable.add(entity)
-
-        if entity['children']:
-            children = entity['children']
-            for child in children:
-                self.getShotAsset(child)
 
     def launch(self, session, entities, event):
         '''Callback method for custom action.'''
@@ -134,6 +125,19 @@ class CreateFolders(BaseAction):
             'message': 'Created Folders Successfully!'
         }
 
+    def get_notask_children(self, entity):
+        output = []
+        if entity.get('object_type', {}).get(
+            'name', entity.entity_type
+        ).lower() == 'task':
+            return output
+        else:
+            output.append(entity)
+        if entity['children']:
+            for child in entity['children']:
+                output.extend(self.get_notask_children(child))
+        return output
+
     def get_presets(self):
         fpath_items = [pypelib.get_presets_path(), 'tools', 'sw_folders.json']
         filepath = os.path.normpath(os.path.sep.join(fpath_items))