use a utility function to get the name of a stack from the blocks

1 files changed, 20 insertions, 0 deletions

diff --git a/TurtleArt/tautils.py b/TurtleArt/tautils.py
index 244be6d..a8ef384 100644
--- a/TurtleArt/tautils.py
+++ b/TurtleArt/tautils.py
@@ -808,6 +808,26 @@ def find_blk_below(blk, namelist):
     return None
 
 
+def get_stack_name(blk):
+    ''' Return the name of the action stack that the given block belongs to.
+    If the top block of this stack is not a stack-defining block, return
+    None. '''
+    top_block = find_top_block(blk)
+    if top_block.name == 'start':
+        return 'start'
+    elif top_block.name == 'hat1':
+        return 'stack1'
+    elif top_block.name == 'hat2':
+        return 'stack2'
+    elif top_block.name == 'hat':
+        try:
+            return top_block.connections[1].values[0]
+        except (IndexError, AttributeError):
+            return None
+    else:
+        return None
+
+
 def get_hardware():
     ''' Determine whether we are using XO 1.0, 1.5, ... or 'unknown'
     hardware '''