mirror of https://github.com/astral-sh/ruff
143 lines
5.0 KiB
Rust
143 lines
5.0 KiB
Rust
use memchr::memchr2;
|
|
pub(crate) use normalize::{normalize_string, NormalizedString, StringNormalizer};
|
|
use ruff_python_ast::str::Quote;
|
|
use ruff_python_ast::StringLikePart;
|
|
use ruff_python_ast::{
|
|
self as ast,
|
|
str_prefix::{AnyStringPrefix, StringLiteralPrefix},
|
|
AnyStringFlags, StringFlags,
|
|
};
|
|
use ruff_source_file::LineRanges;
|
|
use ruff_text_size::Ranged;
|
|
|
|
use crate::prelude::*;
|
|
use crate::QuoteStyle;
|
|
|
|
pub(crate) mod docstring;
|
|
pub(crate) mod implicit;
|
|
mod normalize;
|
|
|
|
impl Format<PyFormatContext<'_>> for AnyStringPrefix {
|
|
fn fmt(&self, f: &mut PyFormatter) -> FormatResult<()> {
|
|
// Remove the unicode prefix `u` if any because it is meaningless in Python 3+.
|
|
if !matches!(
|
|
self,
|
|
AnyStringPrefix::Regular(StringLiteralPrefix::Empty | StringLiteralPrefix::Unicode)
|
|
) {
|
|
token(self.as_str()).fmt(f)?;
|
|
}
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
#[derive(Copy, Clone, Debug)]
|
|
pub(crate) struct StringQuotes {
|
|
triple: bool,
|
|
quote_char: Quote,
|
|
}
|
|
|
|
impl Format<PyFormatContext<'_>> for StringQuotes {
|
|
fn fmt(&self, f: &mut PyFormatter) -> FormatResult<()> {
|
|
let quotes = match (self.quote_char, self.triple) {
|
|
(Quote::Single, false) => "'",
|
|
(Quote::Single, true) => "'''",
|
|
(Quote::Double, false) => "\"",
|
|
(Quote::Double, true) => "\"\"\"",
|
|
};
|
|
|
|
token(quotes).fmt(f)
|
|
}
|
|
}
|
|
|
|
impl From<AnyStringFlags> for StringQuotes {
|
|
fn from(value: AnyStringFlags) -> Self {
|
|
Self {
|
|
triple: value.is_triple_quoted(),
|
|
quote_char: value.quote_style(),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl TryFrom<QuoteStyle> for Quote {
|
|
type Error = ();
|
|
|
|
fn try_from(style: QuoteStyle) -> Result<Quote, ()> {
|
|
match style {
|
|
QuoteStyle::Single => Ok(Quote::Single),
|
|
QuoteStyle::Double => Ok(Quote::Double),
|
|
QuoteStyle::Preserve => Err(()),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl From<Quote> for QuoteStyle {
|
|
fn from(value: Quote) -> Self {
|
|
match value {
|
|
Quote::Single => QuoteStyle::Single,
|
|
Quote::Double => QuoteStyle::Double,
|
|
}
|
|
}
|
|
}
|
|
|
|
// Extension trait that adds formatter specific helper methods to `StringLike`.
|
|
pub(crate) trait StringLikeExtensions {
|
|
fn is_multiline(&self, context: &PyFormatContext) -> bool;
|
|
}
|
|
|
|
impl StringLikeExtensions for ast::StringLike<'_> {
|
|
fn is_multiline(&self, context: &PyFormatContext) -> bool {
|
|
self.parts().any(|part| match part {
|
|
StringLikePart::String(_) | StringLikePart::Bytes(_) => {
|
|
part.flags().is_triple_quoted()
|
|
&& context.source().contains_line_break(part.range())
|
|
}
|
|
StringLikePart::FString(f_string) => {
|
|
fn contains_line_break_or_comments(
|
|
elements: &ast::FStringElements,
|
|
context: &PyFormatContext,
|
|
is_triple_quoted: bool,
|
|
) -> bool {
|
|
elements.iter().any(|element| match element {
|
|
ast::FStringElement::Literal(literal) => {
|
|
is_triple_quoted
|
|
&& context.source().contains_line_break(literal.range())
|
|
}
|
|
ast::FStringElement::Expression(expression) => {
|
|
// Expressions containing comments can't be joined.
|
|
//
|
|
// Format specifiers needs to be checked as well. For example, the
|
|
// following should be considered multiline because the literal
|
|
// part of the format specifier contains a newline at the end
|
|
// (`.3f\n`):
|
|
//
|
|
// ```py
|
|
// x = f"hello {a + b + c + d:.3f
|
|
// } world"
|
|
// ```
|
|
context.comments().contains_comments(expression.into())
|
|
|| expression.format_spec.as_deref().is_some_and(|spec| {
|
|
contains_line_break_or_comments(
|
|
&spec.elements,
|
|
context,
|
|
is_triple_quoted,
|
|
)
|
|
})
|
|
|| expression.debug_text.as_ref().is_some_and(|debug_text| {
|
|
memchr2(b'\n', b'\r', debug_text.leading.as_bytes()).is_some()
|
|
|| memchr2(b'\n', b'\r', debug_text.trailing.as_bytes())
|
|
.is_some()
|
|
})
|
|
}
|
|
})
|
|
}
|
|
|
|
contains_line_break_or_comments(
|
|
&f_string.elements,
|
|
context,
|
|
f_string.flags.is_triple_quoted(),
|
|
)
|
|
}
|
|
})
|
|
}
|
|
}
|