mirrors/ruff
1
0
Fork 0
mirror of https://github.com/astral-sh/ruff.git synced 2025-07-12 07:35:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Add size hints to string parser (#9413)

Browse source
This commit is contained in:
Charlie Marsh 2024-01-06 15:59:34 -05:00 • committed by GitHub
parent e80b3db10d
commit 1666c7a5cb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

9
crates/ruff_python_parser/src/string.rs
View file

@ -107,6 +107,7 @@ impl<'a> StringParser<'a> {
_ => std::char::from_u32(p).ok_or(unicode_error),
}
}
fn parse_octet(&mut self, o: u8) -> char {
let mut radix_bytes = [o, 0, 0];
let mut len = 1;
@ -203,7 +204,7 @@ impl<'a> StringParser<'a> {
}
fn parse_fstring_middle(&mut self) -> Result<ast::FStringElement, LexicalError> {
let mut value = String::new();
let mut value = String::with_capacity(self.rest.len());
while let Some(ch) = self.next_char() {
match ch {
// We can encounter a `\` as the last character in a `FStringMiddle`
@ -246,7 +247,7 @@ impl<'a> StringParser<'a> {
}
fn parse_bytes(&mut self) -> Result<StringType, LexicalError> {
let mut content = String::new();
let mut content = String::with_capacity(self.rest.len());
while let Some(ch) = self.next_char() {
match ch {
'\\' if !self.kind.is_raw() => {
@ -265,7 +266,6 @@ impl<'a> StringParser<'a> {
}
}
}
Ok(StringType::Bytes(ast::BytesLiteral {
value: content.chars().map(|c| c as u8).collect::<Vec<u8>>(),
range: self.range,
@ -273,8 +273,7 @@ impl<'a> StringParser<'a> {
}
fn parse_string(&mut self) -> Result<StringType, LexicalError> {
let mut value = String::new();
let mut value = String::with_capacity(self.rest.len());
if self.kind.is_raw() {
value.push_str(self.skip_bytes(self.rest.len()));
} else {