Python/ruff
1
0
Fork 0
mirror of https://github.com/astral-sh/ruff synced 2026-01-22 22:10:48 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add support for reformatting byte strings (#3176)

Browse Source
This commit is contained in:
Charlie Marsh
2023-02-23 11:50:24 -05:00
committed by GitHub
parent f967f344fc
commit 1e7233a8eb
20 changed files with 90 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
crates/ruff_python_formatter/src/format/expr.rs
View File

@@ -652,7 +652,7 @@ fn format_constant(
write!(f, [text("False")])?;
}
}
Constant::Str(_) => write!(f, [string_literal(expr)])?,
Constant::Str(_) | Constant::Bytes(_) => write!(f, [string_literal(expr)])?,
_ => write!(f, [literal(Range::from_located(expr))])?,
}
Ok(())

6
crates/ruff_python_formatter/src/format/strings.rs
View File

@@ -37,6 +37,7 @@ impl Format<ASTFormatContext<'_>> for StringLiteralPart {
}
}
// Retain raw prefixes.
let mut is_raw = false;
if leading_quote.contains('r') {
is_raw = true;
@@ -46,6 +47,11 @@ impl Format<ASTFormatContext<'_>> for StringLiteralPart {
f.write_element(FormatElement::StaticText { text: "R" })?;
}
// Normalize bytes literals to use b"...".
if leading_quote.contains('b') || leading_quote.contains('B') {
f.write_element(FormatElement::StaticText { text: "b" })?;
}
if trailing_quote.len() == 1 {
// Single-quoted string.
if dquotes == 0 || squotes > 0 {