Rust-related/RustPython
1
0
Fork 1
mirror of https://github.com/RustPython/RustPython.git synced 2026-06-02 19:39:49 +09:00
Code Issues Packages Projects Releases Wiki Activity

Improve normalize_encoding_name

Browse Source 
Co-authored-by: Jeong YunWon <youknowone@users.noreply.github.com>
This commit is contained in:
Noah
2021-05-20 08:06:46 -05:00
parent c0f1da8d8f
commit fd2ceeb3e7
1 changed files with 10 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
vm/src/codecs.rs
View File

@@ -311,23 +311,19 @@ impl CodecsRegistry {
}
fn normalize_encoding_name(encoding: &str) -> Cow<'_, str> {
let mut chars = encoding.char_indices();
for (i, c) in &mut chars {
if c == ' ' || c.is_ascii_uppercase() {
let mut out = String::with_capacity(encoding.len());
out.push_str(&encoding[..i]);
out.push(c);
for (_, c) in chars {
let c = match c {
' ' => '-',
c => c.to_ascii_lowercase(),
};
out.push(c as char)
if let Some(i) = encoding.find(|c: char| c == ' ' || c.is_ascii_uppercase()) {
let mut out = encoding.as_bytes().to_owned();
for byte in &mut out[i..] {
if *byte == b' ' {
*byte = b'-';
} else {
byte.make_ascii_lowercase();
}
return out.into();
}
String::from_utf8(out).unwrap().into()
} else {
encoding.into()
}
encoding.into()
}
// TODO: exceptions with custom payloads