From 9f18ccd65790cbcfe88a9f0b4556cf15e644c2a4 Mon Sep 17 00:00:00 2001 From: yodalee Date: Fri, 13 Jul 2018 00:15:28 +0800 Subject: [PATCH] add pretty print to list and tuple --- vm/src/pyobject.rs | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/vm/src/pyobject.rs b/vm/src/pyobject.rs index 065488cad3..e7e14ac192 100644 --- a/vm/src/pyobject.rs +++ b/vm/src/pyobject.rs @@ -110,8 +110,18 @@ impl PyObject { match self.kind { PyObjectKind::String { ref value } => value.clone(), PyObjectKind::Integer { ref value } => format!("{:?}", value), - PyObjectKind::List { ref elements } => format!("{:?}", elements), - PyObjectKind::Tuple { ref elements } => format!("{:?}", elements), + PyObjectKind::List { ref elements } => { + format!("[{}]", elements.iter() + .map(|elem| elem.borrow_mut().str()) + .collect::>() + .join(", ")) + } + PyObjectKind::Tuple { ref elements } => { + format!("{{{}}}", elements.iter() + .map(|elem| elem.borrow_mut().str()) + .collect::>() + .join(", ")) + } PyObjectKind::None => String::from("None"), _ => { println!("Not impl {:?}", self);