diff options
Diffstat (limited to 'TurtleArt/tautils.py')
-rw-r--r-- | TurtleArt/tautils.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/TurtleArt/tautils.py b/TurtleArt/tautils.py index bd7309f..676c4ca 100644 --- a/TurtleArt/tautils.py +++ b/TurtleArt/tautils.py @@ -657,13 +657,15 @@ def find_group(blk): return group -def find_blk_below(blk, name): +def find_blk_below(blk, namelist): ''' Find a specific block below this block. ''' if blk == None or len(blk.connections) == 0: return + if type(namelist) != list: + namelist = [namelist] group = find_group(blk) for gblk in group: - if gblk.name == name: + if gblk.name in namelist: return gblk return None |