def modifyWhileStmt(self, node): statements = ASTModifier.modifyWhileStmt(self, node) if len(statements) == 1: node = statements[0] try: value = evaluate_expression(node.condition) if value is False: return [] except ParseException: pass return [node] else: return flatten([self.modify(stmt) for stmt in statements])
def modifyWhileStmt(self, node): if self.pass_num == 2: node = ASTModifier.modifyWhileStmt(self, node)[0] node.statements = self.fixStatementList(node.statements) return [node]