From 468d9184a343beeac9cac2e492a1f842a6a3234e Mon Sep 17 00:00:00 2001
From: Galen Williamson <galen@vector35.com>
Date: Wed, 7 Jul 2021 17:11:22 -0400
Subject: Added slicing support to __getitem__ in all basic block classes

---
 python/mediumlevelil.py | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'python/mediumlevelil.py')

diff --git a/python/mediumlevelil.py b/python/mediumlevelil.py
index 8d552b64..d0190f95 100644
--- a/python/mediumlevelil.py
+++ b/python/mediumlevelil.py
@@ -1258,6 +1258,8 @@ class MediumLevelILBasicBlock(basicblock.BasicBlock):
 
 	def __getitem__(self, idx):
 		size = self.end - self.start
+		if isinstance(idx, slice):
+			return [self[index] for index in range(*idx.incides(size))]
 		if idx > size or idx < -size:
 			raise IndexError("list index is out of range")
 		if idx >= 0:
-- 
cgit v1.3.1