Commonize all InstructionTextToken generation

1 files changed, 1 insertions, 11 deletions

diff --git a/python/mediumlevelil.py b/python/mediumlevelil.py
index c88ab6b8..2dae008a 100644
--- a/python/mediumlevelil.py
+++ b/python/mediumlevelil.py
@@ -331,17 +331,7 @@ class MediumLevelILInstruction(object):
 			if not core.BNGetMediumLevelILExprText(self.function.handle, self.function.arch.handle,
 				self.expr_index, tokens, count):
 				return None
-		result = []
-		for i in range(0, count.value):
-			token_type = InstructionTextTokenType(tokens[i].type)
-			text = tokens[i].text
-			value = tokens[i].value
-			size = tokens[i].size
-			operand = tokens[i].operand
-			context = tokens[i].context
-			confidence = tokens[i].confidence
-			address = tokens[i].address
-			result.append(function.InstructionTextToken(token_type, text, value, size, operand, context, address, confidence))
+		result = binaryninja.function.InstructionTextToken.get_instruction_lines(tokens, count.value)
 		core.BNFreeInstructionText(tokens, count.value)
 		return result