diff options
Diffstat (limited to 'TurtleArt/tablock.py')
-rw-r--r-- | TurtleArt/tablock.py | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/TurtleArt/tablock.py b/TurtleArt/tablock.py index 9c7547f..2df89c4 100644 --- a/TurtleArt/tablock.py +++ b/TurtleArt/tablock.py @@ -113,11 +113,16 @@ class Blocks: i += 1 return None - def get_similar_blocks(self, type, name): + def get_similar_blocks(self, block_type, name): block_list = [] - for block in self.list: - if block.type == type and block.name == name: - block_list.append(block) + if type(name) == type(''): + for block in self.list: + if block.type == block_type and block.name == name: + block_list.append(block) + else: + for block in self.list: + if block.type == block_type and block.name in name: + block_list.append(block) return block_list |