mschoi/RustPython
1
0
Fork 0
forked from Rust-related/RustPython
Code Pull Requests Actions Activity

Merge pull request #3346 from youknowone/set-repr

Browse Source 
set repr push to buffer directly
This commit is contained in:
Jim Fasarakis-Hilliard
2021-10-20 07:54:08 +03:00
committed by GitHub
parent 68379da4c9 96dffd6b13
commit a4ff86190b
1 changed files with 4 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
vm/src/builtins/set.rs
View File

@@ -209,39 +209,30 @@ impl PySetInner {
}
fn repr(&self, class_name: Option<&str>, vm: &VirtualMachine) -> PyResult<String> {
let mut repr_len = class_name.map_or(0, |name| name.len() + 2);
let mut parts = Vec::with_capacity(self.content.len());
for key in self.elements() {
let part = vm.to_repr(&key)?;
repr_len += part.as_str().len() + 2;
parts.push(part);
}
let (parts, repr_len) = (parts, repr_len);
let mut repr = String::with_capacity(repr_len);
let mut repr = String::new();
if let Some(name) = class_name {
repr.push_str(name);
repr.push('(');
}
repr.push('{');
{
let mut parts_iter = parts.into_iter();
let mut parts_iter = self.elements().into_iter().map(|o| vm.to_repr(&o));
repr.push_str(
parts_iter
.next()
.transpose()?
.expect("this is not called for empty set")
.as_str(),
);
for part in parts_iter {
repr.push_str(", ");
repr.push_str(part.as_str());
repr.push_str(part?.as_str());
}
}
repr.push('}');
if class_name.is_some() {
repr.push(')');
}
debug_assert_eq!(repr.len(), repr_len);
Ok(repr)
}