Move Python whitespace utilities into new `ruff_python_whitespace` crate (#4993)

## Summary

`ruff_newlines` becomes `ruff_python_whitespace`, and includes the
existing "universal newline" handlers alongside the Python
whitespace-specific utilities.
This commit is contained in:
Charlie Marsh 2023-06-09 20:59:57 -04:00 committed by GitHub
parent e86f12a1ec
commit 1d756dc3a7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
57 changed files with 153 additions and 140 deletions

24
Cargo.lock generated
View File

@ -1775,10 +1775,10 @@ dependencies = [
"ruff_cache", "ruff_cache",
"ruff_diagnostics", "ruff_diagnostics",
"ruff_macros", "ruff_macros",
"ruff_newlines",
"ruff_python_ast", "ruff_python_ast",
"ruff_python_semantic", "ruff_python_semantic",
"ruff_python_stdlib", "ruff_python_stdlib",
"ruff_python_whitespace",
"ruff_rustpython", "ruff_rustpython",
"ruff_text_size", "ruff_text_size",
"ruff_textwrap", "ruff_textwrap",
@ -1945,14 +1945,6 @@ dependencies = [
"syn 2.0.18", "syn 2.0.18",
] ]
[[package]]
name = "ruff_newlines"
version = "0.0.0"
dependencies = [
"memchr",
"ruff_text_size",
]
[[package]] [[package]]
name = "ruff_python_ast" name = "ruff_python_ast"
version = "0.0.0" version = "0.0.0"
@ -1967,7 +1959,7 @@ dependencies = [
"num-bigint", "num-bigint",
"num-traits", "num-traits",
"once_cell", "once_cell",
"ruff_newlines", "ruff_python_whitespace",
"ruff_text_size", "ruff_text_size",
"rustc-hash", "rustc-hash",
"rustpython-ast", "rustpython-ast",
@ -1989,8 +1981,8 @@ dependencies = [
"itertools", "itertools",
"once_cell", "once_cell",
"ruff_formatter", "ruff_formatter",
"ruff_newlines",
"ruff_python_ast", "ruff_python_ast",
"ruff_python_whitespace",
"ruff_testing_macros", "ruff_testing_macros",
"ruff_text_size", "ruff_text_size",
"rustc-hash", "rustc-hash",
@ -2024,6 +2016,14 @@ dependencies = [
"rustc-hash", "rustc-hash",
] ]
[[package]]
name = "ruff_python_whitespace"
version = "0.0.0"
dependencies = [
"memchr",
"ruff_text_size",
]
[[package]] [[package]]
name = "ruff_rustpython" name = "ruff_rustpython"
version = "0.0.0" version = "0.0.0"
@ -2055,7 +2055,7 @@ dependencies = [
name = "ruff_textwrap" name = "ruff_textwrap"
version = "0.0.0" version = "0.0.0"
dependencies = [ dependencies = [
"ruff_newlines", "ruff_python_whitespace",
"ruff_text_size", "ruff_text_size",
] ]

View File

@ -17,7 +17,7 @@ name = "ruff"
ruff_cache = { path = "../ruff_cache" } ruff_cache = { path = "../ruff_cache" }
ruff_diagnostics = { path = "../ruff_diagnostics", features = ["serde"] } ruff_diagnostics = { path = "../ruff_diagnostics", features = ["serde"] }
ruff_macros = { path = "../ruff_macros" } ruff_macros = { path = "../ruff_macros" }
ruff_newlines = { path = "../ruff_newlines" } ruff_python_whitespace = { path = "../ruff_python_whitespace" }
ruff_python_ast = { path = "../ruff_python_ast", features = ["serde"] } ruff_python_ast = { path = "../ruff_python_ast", features = ["serde"] }
ruff_python_semantic = { path = "../ruff_python_semantic" } ruff_python_semantic = { path = "../ruff_python_semantic" }
ruff_python_stdlib = { path = "../ruff_python_stdlib" } ruff_python_stdlib = { path = "../ruff_python_stdlib" }

View File

@ -5,9 +5,9 @@ use rustpython_parser::ast::{self, Excepthandler, Expr, Keyword, Ranged, Stmt};
use rustpython_parser::{lexer, Mode, Tok}; use rustpython_parser::{lexer, Mode, Tok};
use ruff_diagnostics::Edit; use ruff_diagnostics::Edit;
use ruff_newlines::NewlineWithTrailingNewline;
use ruff_python_ast::helpers; use ruff_python_ast::helpers;
use ruff_python_ast::source_code::{Indexer, Locator, Stylist}; use ruff_python_ast::source_code::{Indexer, Locator, Stylist};
use ruff_python_whitespace::NewlineWithTrailingNewline;
use crate::autofix::codemods; use crate::autofix::codemods;

View File

@ -4,8 +4,8 @@ use ruff_text_size::TextSize;
use std::path::Path; use std::path::Path;
use ruff_diagnostics::Diagnostic; use ruff_diagnostics::Diagnostic;
use ruff_newlines::StrExt;
use ruff_python_ast::source_code::{Indexer, Locator, Stylist}; use ruff_python_ast::source_code::{Indexer, Locator, Stylist};
use ruff_python_whitespace::UniversalNewlines;
use crate::registry::Rule; use crate::registry::Rule;
use crate::rules::flake8_executable::helpers::{extract_shebang, ShebangDirective}; use crate::rules::flake8_executable::helpers::{extract_shebang, ShebangDirective};

View File

@ -8,9 +8,9 @@ use rustpython_parser::ast::{self, Constant, Expr, Ranged, Stmt, Suite};
use rustpython_parser::lexer::LexResult; use rustpython_parser::lexer::LexResult;
use rustpython_parser::Tok; use rustpython_parser::Tok;
use ruff_newlines::UniversalNewlineIterator;
use ruff_python_ast::source_code::Locator; use ruff_python_ast::source_code::Locator;
use ruff_python_ast::statement_visitor::{walk_stmt, StatementVisitor}; use ruff_python_ast::statement_visitor::{walk_stmt, StatementVisitor};
use ruff_python_whitespace::UniversalNewlineIterator;
/// Extract doc lines (standalone comments) from a token sequence. /// Extract doc lines (standalone comments) from a token sequence.
pub(crate) fn doc_lines_from_tokens(lxr: &[LexResult]) -> DocLines { pub(crate) fn doc_lines_from_tokens(lxr: &[LexResult]) -> DocLines {

View File

@ -1,11 +1,11 @@
use std::fmt::{Debug, Formatter}; use std::fmt::{Debug, Formatter};
use std::iter::FusedIterator; use std::iter::FusedIterator;
use ruff_python_ast::docstrings::{leading_space, leading_words};
use ruff_text_size::{TextLen, TextRange, TextSize}; use ruff_text_size::{TextLen, TextRange, TextSize};
use strum_macros::EnumIter; use strum_macros::EnumIter;
use ruff_newlines::{StrExt, UniversalNewlineIterator}; use ruff_python_whitespace::{UniversalNewlineIterator, UniversalNewlines};
use ruff_python_ast::whitespace;
use crate::docstrings::styles::SectionStyle; use crate::docstrings::styles::SectionStyle;
use crate::docstrings::{Docstring, DocstringBody}; use crate::docstrings::{Docstring, DocstringBody};
@ -154,8 +154,8 @@ impl<'a> SectionContexts<'a> {
} }
if let Some(section_kind) = suspected_as_section(&line, style) { if let Some(section_kind) = suspected_as_section(&line, style) {
let indent = whitespace::leading_space(&line); let indent = leading_space(&line);
let section_name = whitespace::leading_words(&line); let section_name = leading_words(&line);
let section_name_range = TextRange::at(indent.text_len(), section_name.text_len()); let section_name_range = TextRange::at(indent.text_len(), section_name.text_len());
@ -379,7 +379,7 @@ impl Debug for SectionContext<'_> {
} }
fn suspected_as_section(line: &str, style: SectionStyle) -> Option<SectionKind> { fn suspected_as_section(line: &str, style: SectionStyle) -> Option<SectionKind> {
if let Some(kind) = SectionKind::from_str(whitespace::leading_words(line)) { if let Some(kind) = SectionKind::from_str(leading_words(line)) {
if style.sections().contains(&kind) { if style.sections().contains(&kind) {
return Some(kind); return Some(kind);
} }

View File

@ -6,9 +6,9 @@ use rustpython_parser::ast::{Ranged, Stmt};
use rustpython_parser::{lexer, Mode, Tok}; use rustpython_parser::{lexer, Mode, Tok};
use ruff_diagnostics::Edit; use ruff_diagnostics::Edit;
use ruff_newlines::UniversalNewlineIterator;
use ruff_python_ast::helpers::is_docstring_stmt; use ruff_python_ast::helpers::is_docstring_stmt;
use ruff_python_ast::source_code::{Locator, Stylist}; use ruff_python_ast::source_code::{Locator, Stylist};
use ruff_python_whitespace::UniversalNewlineIterator;
use ruff_textwrap::indent; use ruff_textwrap::indent;
#[derive(Debug, Clone, PartialEq, Eq)] #[derive(Debug, Clone, PartialEq, Eq)]
@ -304,8 +304,8 @@ mod tests {
use rustpython_parser::lexer::LexResult; use rustpython_parser::lexer::LexResult;
use rustpython_parser::Parse; use rustpython_parser::Parse;
use ruff_newlines::LineEnding;
use ruff_python_ast::source_code::{Locator, Stylist}; use ruff_python_ast::source_code::{Locator, Stylist};
use ruff_python_whitespace::LineEnding;
use super::Insertion; use super::Insertion;

View File

@ -11,8 +11,8 @@ use regex::Regex;
use ruff_text_size::{TextLen, TextRange, TextSize}; use ruff_text_size::{TextLen, TextRange, TextSize};
use ruff_diagnostics::Diagnostic; use ruff_diagnostics::Diagnostic;
use ruff_newlines::LineEnding;
use ruff_python_ast::source_code::Locator; use ruff_python_ast::source_code::Locator;
use ruff_python_whitespace::LineEnding;
use crate::codes::NoqaCode; use crate::codes::NoqaCode;
use crate::registry::{AsRule, Rule, RuleSet}; use crate::registry::{AsRule, Rule, RuleSet};
@ -514,8 +514,8 @@ mod tests {
use ruff_text_size::{TextRange, TextSize}; use ruff_text_size::{TextRange, TextSize};
use ruff_diagnostics::Diagnostic; use ruff_diagnostics::Diagnostic;
use ruff_newlines::LineEnding;
use ruff_python_ast::source_code::Locator; use ruff_python_ast::source_code::Locator;
use ruff_python_whitespace::LineEnding;
use crate::noqa::{add_noqa_inner, NoqaMapping, NOQA_LINE_REGEX}; use crate::noqa::{add_noqa_inner, NoqaMapping, NOQA_LINE_REGEX};
use crate::rules::pycodestyle::rules::AmbiguousVariableName; use crate::rules::pycodestyle::rules::AmbiguousVariableName;

View File

@ -1,8 +1,8 @@
use ruff_text_size::TextSize; use ruff_text_size::TextSize;
use rustpython_parser::ast::{Expr, Ranged, Stmt}; use rustpython_parser::ast::{Expr, Ranged, Stmt};
use ruff_newlines::StrExt;
use ruff_python_ast::source_code::Locator; use ruff_python_ast::source_code::Locator;
use ruff_python_whitespace::UniversalNewlines;
/// Return `true` if a function's return statement include at least one /// Return `true` if a function's return statement include at least one
/// non-`None` value. /// non-`None` value.

View File

@ -5,12 +5,12 @@ use rustpython_parser::ast::{self, Cmpop, Constant, Expr, ExprContext, Ranged, S
use ruff_diagnostics::{AutofixKind, Diagnostic, Edit, Fix, Violation}; use ruff_diagnostics::{AutofixKind, Diagnostic, Edit, Fix, Violation};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::StrExt;
use ruff_python_ast::comparable::{ComparableConstant, ComparableExpr, ComparableStmt}; use ruff_python_ast::comparable::{ComparableConstant, ComparableExpr, ComparableStmt};
use ruff_python_ast::helpers::{ use ruff_python_ast::helpers::{
any_over_expr, contains_effect, first_colon_range, has_comments, has_comments_in, any_over_expr, contains_effect, first_colon_range, has_comments, has_comments_in,
}; };
use ruff_python_semantic::model::SemanticModel; use ruff_python_semantic::model::SemanticModel;
use ruff_python_whitespace::UniversalNewlines;
use crate::checkers::ast::Checker; use crate::checkers::ast::Checker;
use crate::line_width::LineWidth; use crate::line_width::LineWidth;

View File

@ -5,8 +5,8 @@ use rustpython_parser::ast::{self, Ranged, Stmt, Withitem};
use ruff_diagnostics::{AutofixKind, Violation}; use ruff_diagnostics::{AutofixKind, Violation};
use ruff_diagnostics::{Diagnostic, Fix}; use ruff_diagnostics::{Diagnostic, Fix};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::StrExt;
use ruff_python_ast::helpers::{first_colon_range, has_comments_in}; use ruff_python_ast::helpers::{first_colon_range, has_comments_in};
use ruff_python_whitespace::UniversalNewlines;
use crate::checkers::ast::Checker; use crate::checkers::ast::Checker;
use crate::line_width::LineWidth; use crate::line_width::LineWidth;

View File

@ -1,8 +1,8 @@
use rustpython_parser::ast::{Ranged, Stmt}; use rustpython_parser::ast::{Ranged, Stmt};
use rustpython_parser::{lexer, Mode, Tok}; use rustpython_parser::{lexer, Mode, Tok};
use ruff_newlines::StrExt;
use ruff_python_ast::source_code::Locator; use ruff_python_ast::source_code::Locator;
use ruff_python_whitespace::UniversalNewlines;
use crate::rules::isort::types::TrailingComma; use crate::rules::isort::types::TrailingComma;

View File

@ -6,12 +6,11 @@ use rustpython_parser::ast::{Ranged, Stmt};
use ruff_diagnostics::{AutofixKind, Diagnostic, Edit, Fix, Violation}; use ruff_diagnostics::{AutofixKind, Diagnostic, Edit, Fix, Violation};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::StrExt;
use ruff_python_ast::helpers::{ use ruff_python_ast::helpers::{
followed_by_multi_statement_line, preceded_by_multi_statement_line, trailing_lines_end, followed_by_multi_statement_line, preceded_by_multi_statement_line, trailing_lines_end,
}; };
use ruff_python_ast::source_code::{Indexer, Locator, Stylist}; use ruff_python_ast::source_code::{Indexer, Locator, Stylist};
use ruff_python_ast::whitespace::leading_space; use ruff_python_whitespace::{leading_indentation, UniversalNewlines};
use ruff_textwrap::indent; use ruff_textwrap::indent;
use crate::line_width::LineWidth; use crate::line_width::LineWidth;
@ -89,7 +88,7 @@ pub(crate) fn organize_imports(
package: Option<&Path>, package: Option<&Path>,
) -> Option<Diagnostic> { ) -> Option<Diagnostic> {
let indentation = locator.slice(extract_indentation_range(&block.imports, locator)); let indentation = locator.slice(extract_indentation_range(&block.imports, locator));
let indentation = leading_space(indentation); let indentation = leading_indentation(indentation);
let range = extract_range(&block.imports); let range = extract_range(&block.imports);

View File

@ -2,8 +2,8 @@ use ruff_text_size::{TextLen, TextRange};
use rustpython_parser::ast::{self, Cmpop, Expr}; use rustpython_parser::ast::{self, Cmpop, Expr};
use unicode_width::UnicodeWidthStr; use unicode_width::UnicodeWidthStr;
use ruff_newlines::Line;
use ruff_python_ast::source_code::Generator; use ruff_python_ast::source_code::Generator;
use ruff_python_whitespace::Line;
use crate::line_width::{LineLength, LineWidth, TabSize}; use crate::line_width::{LineLength, LineWidth, TabSize};

View File

@ -1,6 +1,6 @@
use ruff_diagnostics::{Diagnostic, Violation}; use ruff_diagnostics::{Diagnostic, Violation};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::Line; use ruff_python_whitespace::Line;
use crate::rules::pycodestyle::helpers::is_overlong; use crate::rules::pycodestyle::helpers::is_overlong;
use crate::settings::Settings; use crate::settings::Settings;

View File

@ -3,11 +3,10 @@ use rustpython_parser::ast::{self, Arg, Arguments, Constant, Expr, Ranged, Stmt}
use ruff_diagnostics::{AutofixKind, Diagnostic, Edit, Fix, Violation}; use ruff_diagnostics::{AutofixKind, Diagnostic, Edit, Fix, Violation};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::StrExt;
use ruff_python_ast::helpers::{has_leading_content, has_trailing_content}; use ruff_python_ast::helpers::{has_leading_content, has_trailing_content};
use ruff_python_ast::source_code::Generator; use ruff_python_ast::source_code::Generator;
use ruff_python_ast::whitespace::leading_space;
use ruff_python_semantic::model::SemanticModel; use ruff_python_semantic::model::SemanticModel;
use ruff_python_whitespace::{leading_indentation, UniversalNewlines};
use crate::checkers::ast::Checker; use crate::checkers::ast::Checker;
use crate::registry::AsRule; use crate::registry::AsRule;
@ -84,7 +83,7 @@ pub(crate) fn lambda_assignment(
&& !has_trailing_content(stmt, checker.locator) && !has_trailing_content(stmt, checker.locator)
{ {
let first_line = checker.locator.line(stmt.start()); let first_line = checker.locator.line(stmt.start());
let indentation = &leading_space(first_line); let indentation = leading_indentation(first_line);
let mut indented = String::new(); let mut indented = String::new();
for (idx, line) in function( for (idx, line) in function(
checker.semantic_model(), checker.semantic_model(),

View File

@ -1,6 +1,6 @@
use ruff_diagnostics::{Diagnostic, Violation}; use ruff_diagnostics::{Diagnostic, Violation};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::Line; use ruff_python_whitespace::Line;
use crate::rules::pycodestyle::helpers::is_overlong; use crate::rules::pycodestyle::helpers::is_overlong;
use crate::settings::Settings; use crate::settings::Settings;

View File

@ -2,8 +2,7 @@ use ruff_text_size::{TextLen, TextRange};
use ruff_diagnostics::{Diagnostic, Violation}; use ruff_diagnostics::{Diagnostic, Violation};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::Line; use ruff_python_whitespace::{leading_indentation, Line};
use ruff_python_ast::whitespace::leading_space;
/// ## What it does /// ## What it does
/// Checks for mixed tabs and spaces in indentation. /// Checks for mixed tabs and spaces in indentation.
@ -37,7 +36,7 @@ impl Violation for MixedSpacesAndTabs {
/// E101 /// E101
pub(crate) fn mixed_spaces_and_tabs(line: &Line) -> Option<Diagnostic> { pub(crate) fn mixed_spaces_and_tabs(line: &Line) -> Option<Diagnostic> {
let indent = leading_space(line.as_str()); let indent = leading_indentation(line.as_str());
if indent.contains(' ') && indent.contains('\t') { if indent.contains(' ') && indent.contains('\t') {
Some(Diagnostic::new( Some(Diagnostic::new(

View File

@ -2,9 +2,8 @@ use ruff_text_size::{TextLen, TextRange, TextSize};
use ruff_diagnostics::{Diagnostic, Violation}; use ruff_diagnostics::{Diagnostic, Violation};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::Line;
use ruff_python_ast::source_code::Indexer; use ruff_python_ast::source_code::Indexer;
use ruff_python_ast::whitespace::leading_space; use ruff_python_whitespace::{leading_indentation, Line};
#[violation] #[violation]
pub struct TabIndentation; pub struct TabIndentation;
@ -18,7 +17,7 @@ impl Violation for TabIndentation {
/// W191 /// W191
pub(crate) fn tab_indentation(line: &Line, indexer: &Indexer) -> Option<Diagnostic> { pub(crate) fn tab_indentation(line: &Line, indexer: &Indexer) -> Option<Diagnostic> {
let indent = leading_space(line); let indent = leading_indentation(line);
if let Some(tab_index) = indent.find('\t') { if let Some(tab_index) = indent.find('\t') {
// If the tab character is within a multi-line string, abort. // If the tab character is within a multi-line string, abort.
let tab_offset = line.start() + TextSize::try_from(tab_index).unwrap(); let tab_offset = line.start() + TextSize::try_from(tab_index).unwrap();

View File

@ -2,7 +2,7 @@ use ruff_text_size::{TextLen, TextRange, TextSize};
use ruff_diagnostics::{AlwaysAutofixableViolation, Diagnostic, Edit, Fix}; use ruff_diagnostics::{AlwaysAutofixableViolation, Diagnostic, Edit, Fix};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::Line; use ruff_python_whitespace::Line;
use crate::registry::Rule; use crate::registry::Rule;
use crate::settings::Settings; use crate::settings::Settings;

View File

@ -1,12 +1,12 @@
use std::collections::BTreeSet; use std::collections::BTreeSet;
use ruff_newlines::StrExt;
use ruff_python_ast::call_path::from_qualified_name; use ruff_python_ast::call_path::from_qualified_name;
use ruff_python_ast::cast; use ruff_python_ast::cast;
use ruff_python_ast::helpers::map_callable; use ruff_python_ast::helpers::map_callable;
use ruff_python_ast::str::is_implicit_concatenation; use ruff_python_ast::str::is_implicit_concatenation;
use ruff_python_semantic::definition::{Definition, Member, MemberKind}; use ruff_python_semantic::definition::{Definition, Member, MemberKind};
use ruff_python_semantic::model::SemanticModel; use ruff_python_semantic::model::SemanticModel;
use ruff_python_whitespace::UniversalNewlines;
/// Return the index of the first logical line in a string. /// Return the index of the first logical line in a string.
pub(crate) fn logical_line(content: &str) -> Option<usize> { pub(crate) fn logical_line(content: &str) -> Option<usize> {

View File

@ -1,6 +1,6 @@
use ruff_diagnostics::{AutofixKind, Diagnostic, Edit, Fix, Violation}; use ruff_diagnostics::{AutofixKind, Diagnostic, Edit, Fix, Violation};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::{StrExt, UniversalNewlineIterator}; use ruff_python_whitespace::{UniversalNewlineIterator, UniversalNewlines};
use crate::checkers::ast::Checker; use crate::checkers::ast::Checker;
use crate::docstrings::Docstring; use crate::docstrings::Docstring;

View File

@ -3,8 +3,8 @@ use rustpython_parser::ast::Ranged;
use ruff_diagnostics::{AlwaysAutofixableViolation, Diagnostic, Edit, Fix}; use ruff_diagnostics::{AlwaysAutofixableViolation, Diagnostic, Edit, Fix};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::{StrExt, UniversalNewlineIterator};
use ruff_python_semantic::definition::{Definition, Member, MemberKind}; use ruff_python_semantic::definition::{Definition, Member, MemberKind};
use ruff_python_whitespace::{UniversalNewlineIterator, UniversalNewlines};
use crate::checkers::ast::Checker; use crate::checkers::ast::Checker;
use crate::docstrings::Docstring; use crate::docstrings::Docstring;

View File

@ -5,8 +5,8 @@ use rustpython_parser::ast::Ranged;
use ruff_diagnostics::{AlwaysAutofixableViolation, Diagnostic, Edit, Fix}; use ruff_diagnostics::{AlwaysAutofixableViolation, Diagnostic, Edit, Fix};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::{StrExt, UniversalNewlineIterator};
use ruff_python_semantic::definition::{Definition, Member, MemberKind}; use ruff_python_semantic::definition::{Definition, Member, MemberKind};
use ruff_python_whitespace::{UniversalNewlineIterator, UniversalNewlines};
use crate::checkers::ast::Checker; use crate::checkers::ast::Checker;
use crate::docstrings::Docstring; use crate::docstrings::Docstring;

View File

@ -3,7 +3,7 @@ use strum::IntoEnumIterator;
use ruff_diagnostics::{AlwaysAutofixableViolation, Diagnostic, Edit, Fix}; use ruff_diagnostics::{AlwaysAutofixableViolation, Diagnostic, Edit, Fix};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::{StrExt, UniversalNewlineIterator}; use ruff_python_whitespace::{UniversalNewlineIterator, UniversalNewlines};
use crate::checkers::ast::Checker; use crate::checkers::ast::Checker;
use crate::docstrings::sections::SectionKind; use crate::docstrings::sections::SectionKind;

View File

@ -3,7 +3,7 @@ use strum::IntoEnumIterator;
use ruff_diagnostics::{AlwaysAutofixableViolation, Diagnostic, Edit, Fix}; use ruff_diagnostics::{AlwaysAutofixableViolation, Diagnostic, Edit, Fix};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::{StrExt, UniversalNewlineIterator}; use ruff_python_whitespace::{UniversalNewlineIterator, UniversalNewlines};
use crate::checkers::ast::Checker; use crate::checkers::ast::Checker;
use crate::docstrings::sections::SectionKind; use crate::docstrings::sections::SectionKind;

View File

@ -3,8 +3,8 @@ use ruff_text_size::{TextLen, TextRange};
use ruff_diagnostics::{AlwaysAutofixableViolation, Violation}; use ruff_diagnostics::{AlwaysAutofixableViolation, Violation};
use ruff_diagnostics::{Diagnostic, Edit, Fix}; use ruff_diagnostics::{Diagnostic, Edit, Fix};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::NewlineWithTrailingNewline; use ruff_python_ast::docstrings::{clean_space, leading_space};
use ruff_python_ast::whitespace; use ruff_python_whitespace::NewlineWithTrailingNewline;
use crate::checkers::ast::Checker; use crate::checkers::ast::Checker;
use crate::docstrings::Docstring; use crate::docstrings::Docstring;
@ -76,7 +76,7 @@ pub(crate) fn indent(checker: &mut Checker, docstring: &Docstring) {
continue; continue;
} }
let line_indent = whitespace::leading_space(line); let line_indent = leading_space(line);
// We only report tab indentation once, so only check if we haven't seen a tab // We only report tab indentation once, so only check if we haven't seen a tab
// yet. // yet.
@ -93,7 +93,7 @@ pub(crate) fn indent(checker: &mut Checker, docstring: &Docstring) {
if checker.patch(diagnostic.kind.rule()) { if checker.patch(diagnostic.kind.rule()) {
#[allow(deprecated)] #[allow(deprecated)]
diagnostic.set_fix(Fix::unspecified(Edit::range_replacement( diagnostic.set_fix(Fix::unspecified(Edit::range_replacement(
whitespace::clean(docstring.indentation), clean_space(docstring.indentation),
TextRange::at(line.start(), line_indent.text_len()), TextRange::at(line.start(), line_indent.text_len()),
))); )));
} }
@ -133,7 +133,7 @@ pub(crate) fn indent(checker: &mut Checker, docstring: &Docstring) {
let mut diagnostic = let mut diagnostic =
Diagnostic::new(OverIndentation, TextRange::empty(over_indented.start())); Diagnostic::new(OverIndentation, TextRange::empty(over_indented.start()));
if checker.patch(diagnostic.kind.rule()) { if checker.patch(diagnostic.kind.rule()) {
let indent = whitespace::clean(docstring.indentation); let indent = clean_space(docstring.indentation);
let edit = if indent.is_empty() { let edit = if indent.is_empty() {
Edit::range_deletion(over_indented) Edit::range_deletion(over_indented)
} else { } else {
@ -148,12 +148,12 @@ pub(crate) fn indent(checker: &mut Checker, docstring: &Docstring) {
// If the last line is over-indented... // If the last line is over-indented...
if let Some(last) = lines.last() { if let Some(last) = lines.last() {
let line_indent = whitespace::leading_space(last); let line_indent = leading_space(last);
if line_indent.len() > docstring.indentation.len() { if line_indent.len() > docstring.indentation.len() {
let mut diagnostic = let mut diagnostic =
Diagnostic::new(OverIndentation, TextRange::empty(last.start())); Diagnostic::new(OverIndentation, TextRange::empty(last.start()));
if checker.patch(diagnostic.kind.rule()) { if checker.patch(diagnostic.kind.rule()) {
let indent = whitespace::clean(docstring.indentation); let indent = clean_space(docstring.indentation);
let range = TextRange::at(last.start(), line_indent.text_len()); let range = TextRange::at(last.start(), line_indent.text_len());
let edit = if indent.is_empty() { let edit = if indent.is_empty() {
Edit::range_deletion(range) Edit::range_deletion(range)

View File

@ -3,9 +3,9 @@ use rustpython_parser::ast::Ranged;
use ruff_diagnostics::{AlwaysAutofixableViolation, Diagnostic, Edit, Fix}; use ruff_diagnostics::{AlwaysAutofixableViolation, Diagnostic, Edit, Fix};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::{NewlineWithTrailingNewline, UniversalNewlineIterator};
use ruff_python_ast::str::{is_triple_quote, leading_quote}; use ruff_python_ast::str::{is_triple_quote, leading_quote};
use ruff_python_semantic::definition::{Definition, Member}; use ruff_python_semantic::definition::{Definition, Member};
use ruff_python_whitespace::{NewlineWithTrailingNewline, UniversalNewlineIterator};
use crate::checkers::ast::Checker; use crate::checkers::ast::Checker;
use crate::docstrings::Docstring; use crate::docstrings::Docstring;

View File

@ -3,8 +3,8 @@ use rustpython_parser::ast::Ranged;
use ruff_diagnostics::{AlwaysAutofixableViolation, Diagnostic, Edit, Fix}; use ruff_diagnostics::{AlwaysAutofixableViolation, Diagnostic, Edit, Fix};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::{NewlineWithTrailingNewline, StrExt}; use ruff_python_ast::docstrings::clean_space;
use ruff_python_ast::whitespace; use ruff_python_whitespace::{NewlineWithTrailingNewline, UniversalNewlines};
use crate::checkers::ast::Checker; use crate::checkers::ast::Checker;
use crate::docstrings::Docstring; use crate::docstrings::Docstring;
@ -56,7 +56,7 @@ pub(crate) fn newline_after_last_paragraph(checker: &mut Checker, docstring: &Do
let content = format!( let content = format!(
"{}{}", "{}{}",
checker.stylist.line_ending().as_str(), checker.stylist.line_ending().as_str(),
whitespace::clean(docstring.indentation) clean_space(docstring.indentation)
); );
#[allow(deprecated)] #[allow(deprecated)]
diagnostic.set_fix(Fix::unspecified(Edit::replacement( diagnostic.set_fix(Fix::unspecified(Edit::replacement(

View File

@ -2,8 +2,8 @@ use rustpython_parser::ast::{self, Stmt};
use ruff_diagnostics::{Diagnostic, Violation}; use ruff_diagnostics::{Diagnostic, Violation};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::StrExt;
use ruff_python_semantic::definition::{Definition, Member, MemberKind}; use ruff_python_semantic::definition::{Definition, Member, MemberKind};
use ruff_python_whitespace::UniversalNewlines;
use crate::checkers::ast::Checker; use crate::checkers::ast::Checker;
use crate::docstrings::Docstring; use crate::docstrings::Docstring;

View File

@ -2,7 +2,7 @@ use ruff_text_size::{TextLen, TextRange};
use ruff_diagnostics::{AutofixKind, Diagnostic, Edit, Fix, Violation}; use ruff_diagnostics::{AutofixKind, Diagnostic, Edit, Fix, Violation};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::NewlineWithTrailingNewline; use ruff_python_whitespace::NewlineWithTrailingNewline;
use crate::checkers::ast::Checker; use crate::checkers::ast::Checker;
use crate::docstrings::Docstring; use crate::docstrings::Docstring;

View File

@ -5,11 +5,11 @@ use once_cell::sync::Lazy;
use ruff_diagnostics::{Diagnostic, Violation}; use ruff_diagnostics::{Diagnostic, Violation};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::StrExt;
use ruff_python_ast::call_path::{from_qualified_name, CallPath}; use ruff_python_ast::call_path::{from_qualified_name, CallPath};
use ruff_python_ast::cast; use ruff_python_ast::cast;
use ruff_python_semantic::analyze::visibility::{is_property, is_test}; use ruff_python_semantic::analyze::visibility::{is_property, is_test};
use ruff_python_semantic::definition::{Definition, Member, MemberKind}; use ruff_python_semantic::definition::{Definition, Member, MemberKind};
use ruff_python_whitespace::UniversalNewlines;
use crate::checkers::ast::Checker; use crate::checkers::ast::Checker;
use crate::docstrings::Docstring; use crate::docstrings::Docstring;

View File

@ -1,7 +1,7 @@
use ruff_diagnostics::{AutofixKind, Diagnostic, Edit, Fix, Violation}; use ruff_diagnostics::{AutofixKind, Diagnostic, Edit, Fix, Violation};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::NewlineWithTrailingNewline;
use ruff_python_ast::str::{leading_quote, trailing_quote}; use ruff_python_ast::str::{leading_quote, trailing_quote};
use ruff_python_whitespace::NewlineWithTrailingNewline;
use crate::checkers::ast::Checker; use crate::checkers::ast::Checker;
use crate::docstrings::Docstring; use crate::docstrings::Docstring;

View File

@ -8,11 +8,12 @@ use rustpython_parser::ast::{self, Stmt};
use ruff_diagnostics::{AlwaysAutofixableViolation, Violation}; use ruff_diagnostics::{AlwaysAutofixableViolation, Violation};
use ruff_diagnostics::{Diagnostic, Edit, Fix}; use ruff_diagnostics::{Diagnostic, Edit, Fix};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::NewlineWithTrailingNewline; use ruff_python_ast::cast;
use ruff_python_ast::docstrings::{clean_space, leading_space};
use ruff_python_ast::helpers::identifier_range; use ruff_python_ast::helpers::identifier_range;
use ruff_python_ast::{cast, whitespace};
use ruff_python_semantic::analyze::visibility::is_staticmethod; use ruff_python_semantic::analyze::visibility::is_staticmethod;
use ruff_python_semantic::definition::{Definition, Member, MemberKind}; use ruff_python_semantic::definition::{Definition, Member, MemberKind};
use ruff_python_whitespace::NewlineWithTrailingNewline;
use ruff_textwrap::dedent; use ruff_textwrap::dedent;
use crate::checkers::ast::Checker; use crate::checkers::ast::Checker;
@ -400,7 +401,7 @@ fn blanks_and_section_underline(
// Replace the existing underline with a line of the appropriate length. // Replace the existing underline with a line of the appropriate length.
let content = format!( let content = format!(
"{}{}{}", "{}{}{}",
whitespace::clean(docstring.indentation), clean_space(docstring.indentation),
"-".repeat(context.section_name().len()), "-".repeat(context.section_name().len()),
checker.stylist.line_ending().as_str() checker.stylist.line_ending().as_str()
); );
@ -416,7 +417,7 @@ fn blanks_and_section_underline(
} }
if checker.enabled(Rule::SectionUnderlineNotOverIndented) { if checker.enabled(Rule::SectionUnderlineNotOverIndented) {
let leading_space = whitespace::leading_space(&non_blank_line); let leading_space = leading_space(&non_blank_line);
if leading_space.len() > docstring.indentation.len() { if leading_space.len() > docstring.indentation.len() {
let mut diagnostic = Diagnostic::new( let mut diagnostic = Diagnostic::new(
SectionUnderlineNotOverIndented { SectionUnderlineNotOverIndented {
@ -433,7 +434,7 @@ fn blanks_and_section_underline(
// Replace the existing indentation with whitespace of the appropriate length. // Replace the existing indentation with whitespace of the appropriate length.
#[allow(deprecated)] #[allow(deprecated)]
diagnostic.set_fix(Fix::unspecified(Edit::range_replacement( diagnostic.set_fix(Fix::unspecified(Edit::range_replacement(
whitespace::clean(docstring.indentation), clean_space(docstring.indentation),
range, range,
))); )));
}; };
@ -503,7 +504,7 @@ fn blanks_and_section_underline(
let content = format!( let content = format!(
"{}{}{}", "{}{}{}",
checker.stylist.line_ending().as_str(), checker.stylist.line_ending().as_str(),
whitespace::clean(docstring.indentation), clean_space(docstring.indentation),
"-".repeat(context.section_name().len()), "-".repeat(context.section_name().len()),
); );
#[allow(deprecated)] #[allow(deprecated)]
@ -548,7 +549,7 @@ fn blanks_and_section_underline(
let content = format!( let content = format!(
"{}{}{}", "{}{}{}",
checker.stylist.line_ending().as_str(), checker.stylist.line_ending().as_str(),
whitespace::clean(docstring.indentation), clean_space(docstring.indentation),
"-".repeat(context.section_name().len()), "-".repeat(context.section_name().len()),
); );
@ -601,7 +602,7 @@ fn common_section(
} }
if checker.enabled(Rule::SectionNotOverIndented) { if checker.enabled(Rule::SectionNotOverIndented) {
let leading_space = whitespace::leading_space(context.summary_line()); let leading_space = leading_space(context.summary_line());
if leading_space.len() > docstring.indentation.len() { if leading_space.len() > docstring.indentation.len() {
let mut diagnostic = Diagnostic::new( let mut diagnostic = Diagnostic::new(
SectionNotOverIndented { SectionNotOverIndented {
@ -611,7 +612,7 @@ fn common_section(
); );
if checker.patch(diagnostic.kind.rule()) { if checker.patch(diagnostic.kind.rule()) {
// Replace the existing indentation with whitespace of the appropriate length. // Replace the existing indentation with whitespace of the appropriate length.
let content = whitespace::clean(docstring.indentation); let content = clean_space(docstring.indentation);
let fix_range = TextRange::at(context.range().start(), leading_space.text_len()); let fix_range = TextRange::at(context.range().start(), leading_space.text_len());
#[allow(deprecated)] #[allow(deprecated)]
@ -775,7 +776,7 @@ fn args_section(context: &SectionContext) -> FxHashSet<String> {
// Normalize leading whitespace, by removing any lines with less indentation // Normalize leading whitespace, by removing any lines with less indentation
// than the first. // than the first.
let leading_space = whitespace::leading_space(first_line.as_str()); let leading_space = leading_space(first_line.as_str());
let relevant_lines = std::iter::once(first_line) let relevant_lines = std::iter::once(first_line)
.chain(following_lines) .chain(following_lines)
.map(|l| l.as_str()) .map(|l| l.as_str())
@ -819,7 +820,7 @@ fn args_section(context: &SectionContext) -> FxHashSet<String> {
fn parameters_section(checker: &mut Checker, docstring: &Docstring, context: &SectionContext) { fn parameters_section(checker: &mut Checker, docstring: &Docstring, context: &SectionContext) {
// Collect the list of arguments documented in the docstring. // Collect the list of arguments documented in the docstring.
let mut docstring_args: FxHashSet<String> = FxHashSet::default(); let mut docstring_args: FxHashSet<String> = FxHashSet::default();
let section_level_indent = whitespace::leading_space(context.summary_line()); let section_level_indent = leading_space(context.summary_line());
// Join line continuations, then resplit by line. // Join line continuations, then resplit by line.
let adjusted_following_lines = context let adjusted_following_lines = context
@ -830,9 +831,9 @@ fn parameters_section(checker: &mut Checker, docstring: &Docstring, context: &Se
let mut lines = NewlineWithTrailingNewline::from(&adjusted_following_lines); let mut lines = NewlineWithTrailingNewline::from(&adjusted_following_lines);
if let Some(mut current_line) = lines.next() { if let Some(mut current_line) = lines.next() {
for next_line in lines { for next_line in lines {
let current_leading_space = whitespace::leading_space(current_line.as_str()); let current_leading_space = leading_space(current_line.as_str());
if current_leading_space == section_level_indent if current_leading_space == section_level_indent
&& (whitespace::leading_space(&next_line).len() > current_leading_space.len()) && (leading_space(&next_line).len() > current_leading_space.len())
&& !next_line.trim().is_empty() && !next_line.trim().is_empty()
{ {
let parameters = if let Some(semi_index) = current_line.find(':') { let parameters = if let Some(semi_index) = current_line.find(':') {

View File

@ -4,7 +4,7 @@ use ruff_text_size::{TextLen, TextRange, TextSize};
use ruff_diagnostics::{Diagnostic, Violation}; use ruff_diagnostics::{Diagnostic, Violation};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::Line; use ruff_python_whitespace::Line;
/// ## What it does /// ## What it does
/// Check for `noqa` annotations that suppress all diagnostics, as opposed to /// Check for `noqa` annotations that suppress all diagnostics, as opposed to

View File

@ -5,7 +5,7 @@ use ruff_text_size::{TextLen, TextRange, TextSize};
use ruff_diagnostics::{Diagnostic, Violation}; use ruff_diagnostics::{Diagnostic, Violation};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::Line; use ruff_python_whitespace::Line;
/// ## What it does /// ## What it does
/// Check for `type: ignore` annotations that suppress all type warnings, as /// Check for `type: ignore` annotations that suppress all type warnings, as

View File

@ -1,6 +1,6 @@
use ruff_diagnostics::{Diagnostic, Violation}; use ruff_diagnostics::{Diagnostic, Violation};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::Line; use ruff_python_whitespace::Line;
const BIDI_UNICODE: [char; 10] = [ const BIDI_UNICODE: [char; 10] = [
'\u{202A}', //{LEFT-TO-RIGHT EMBEDDING} '\u{202A}', //{LEFT-TO-RIGHT EMBEDDING}

View File

@ -3,7 +3,7 @@ use regex::Regex;
use ruff_diagnostics::{AlwaysAutofixableViolation, Diagnostic, Edit, Fix}; use ruff_diagnostics::{AlwaysAutofixableViolation, Diagnostic, Edit, Fix};
use ruff_macros::{derive_message_formats, violation}; use ruff_macros::{derive_message_formats, violation};
use ruff_newlines::Line; use ruff_python_whitespace::Line;
// TODO: document referencing [PEP 3120]: https://peps.python.org/pep-3120/ // TODO: document referencing [PEP 3120]: https://peps.python.org/pep-3120/
#[violation] #[violation]

View File

@ -8,7 +8,7 @@ rust-version = { workspace = true }
[lib] [lib]
[dependencies] [dependencies]
ruff_newlines = { path = "../ruff_newlines" } ruff_python_whitespace = { path = "../ruff_python_whitespace" }
ruff_text_size = { workspace = true } ruff_text_size = { workspace = true }
anyhow = { workspace = true } anyhow = { workspace = true }

View File

@ -0,0 +1,22 @@
//! Utilities for parsing Python docstrings.
/// Extract the leading words from a line of text within a Python docstring.
pub fn leading_words(line: &str) -> &str {
let line = line.trim();
line.find(|char: char| !char.is_alphanumeric() && !char.is_whitespace())
.map_or(line, |index| &line[..index])
}
/// Extract the leading whitespace from a line of text within a Python docstring.
pub fn leading_space(line: &str) -> &str {
line.find(|char: char| !char.is_whitespace())
.map_or(line, |index| &line[..index])
}
/// Replace any non-whitespace characters from an indentation string within a Python docstring.
pub fn clean_space(indentation: &str) -> String {
indentation
.chars()
.map(|char| if char.is_whitespace() { char } else { ' ' })
.collect()
}

View File

@ -4,7 +4,7 @@ use std::path::Path;
use itertools::Itertools; use itertools::Itertools;
use log::error; use log::error;
use num_traits::Zero; use num_traits::Zero;
use ruff_newlines::UniversalNewlineIterator; use ruff_python_whitespace::UniversalNewlineIterator;
use ruff_text_size::{TextRange, TextSize}; use ruff_text_size::{TextRange, TextSize};
use rustc_hash::{FxHashMap, FxHashSet}; use rustc_hash::{FxHashMap, FxHashSet};
use rustpython_parser::ast::{ use rustpython_parser::ast::{

View File

@ -2,6 +2,7 @@ pub mod all;
pub mod call_path; pub mod call_path;
pub mod cast; pub mod cast;
pub mod comparable; pub mod comparable;
pub mod docstrings;
pub mod function; pub mod function;
pub mod hashable; pub mod hashable;
pub mod helpers; pub mod helpers;

View File

@ -8,7 +8,7 @@ use rustpython_parser::ast::{
Excepthandler, Expr, Identifier, MatchCase, Operator, Pattern, Stmt, Suite, Withitem, Excepthandler, Expr, Identifier, MatchCase, Operator, Pattern, Stmt, Suite, Withitem,
}; };
use ruff_newlines::LineEnding; use ruff_python_whitespace::LineEnding;
use crate::source_code::stylist::{Indentation, Quote, Stylist}; use crate::source_code::stylist::{Indentation, Quote, Stylist};
@ -1459,7 +1459,7 @@ mod tests {
use rustpython_ast::Suite; use rustpython_ast::Suite;
use rustpython_parser::Parse; use rustpython_parser::Parse;
use ruff_newlines::LineEnding; use ruff_python_whitespace::LineEnding;
use crate::source_code::stylist::{Indentation, Quote}; use crate::source_code::stylist::{Indentation, Quote};
use crate::source_code::Generator; use crate::source_code::Generator;

View File

@ -6,7 +6,7 @@ use memchr::{memchr2, memrchr2};
use once_cell::unsync::OnceCell; use once_cell::unsync::OnceCell;
use ruff_text_size::{TextLen, TextRange, TextSize}; use ruff_text_size::{TextLen, TextRange, TextSize};
use ruff_newlines::find_newline; use ruff_python_whitespace::find_newline;
use crate::source_code::{LineIndex, OneIndexed, SourceCode, SourceLocation}; use crate::source_code::{LineIndex, OneIndexed, SourceCode, SourceLocation};

View File

@ -4,7 +4,7 @@ use std::fmt;
use std::ops::Deref; use std::ops::Deref;
use once_cell::unsync::OnceCell; use once_cell::unsync::OnceCell;
use ruff_newlines::{find_newline, LineEnding}; use ruff_python_whitespace::{find_newline, LineEnding};
use rustpython_literal::escape::Quote as StrQuote; use rustpython_literal::escape::Quote as StrQuote;
use rustpython_parser::lexer::LexResult; use rustpython_parser::lexer::LexResult;
use rustpython_parser::Tok; use rustpython_parser::Tok;
@ -166,7 +166,7 @@ mod tests {
use rustpython_parser::lexer::lex; use rustpython_parser::lexer::lex;
use rustpython_parser::Mode; use rustpython_parser::Mode;
use ruff_newlines::{find_newline, LineEnding}; use ruff_python_whitespace::{find_newline, LineEnding};
use crate::source_code::stylist::{Indentation, Quote}; use crate::source_code::stylist::{Indentation, Quote};
use crate::source_code::{Locator, Stylist}; use crate::source_code::{Locator, Stylist};

View File

@ -1,6 +1,8 @@
use ruff_text_size::{TextRange, TextSize}; use ruff_text_size::{TextRange, TextSize};
use rustpython_parser::ast::Ranged; use rustpython_parser::ast::Ranged;
use ruff_python_whitespace::is_python_whitespace;
use crate::source_code::Locator; use crate::source_code::Locator;
/// Extract the leading indentation from a line. /// Extract the leading indentation from a line.
@ -17,41 +19,9 @@ pub fn indentation_at_offset<'a>(locator: &'a Locator, offset: TextSize) -> Opti
let line_start = locator.line_start(offset); let line_start = locator.line_start(offset);
let indentation = &locator.contents()[TextRange::new(line_start, offset)]; let indentation = &locator.contents()[TextRange::new(line_start, offset)];
if indentation.chars().all(char::is_whitespace) { if indentation.chars().all(is_python_whitespace) {
Some(indentation) Some(indentation)
} else { } else {
None None
} }
} }
/// Extract the leading words from a line of text.
pub fn leading_words(line: &str) -> &str {
let line = line.trim();
line.find(|char: char| !char.is_alphanumeric() && !char.is_whitespace())
.map_or(line, |index| &line[..index])
}
/// Extract the leading whitespace from a line of text.
pub fn leading_space(line: &str) -> &str {
line.find(|char: char| !char.is_whitespace())
.map_or(line, |index| &line[..index])
}
/// Replace any non-whitespace characters from an indentation string.
pub fn clean(indentation: &str) -> String {
indentation
.chars()
.map(|char| if char.is_whitespace() { char } else { ' ' })
.collect()
}
/// Returns `true` for [whitespace](https://docs.python.org/3/reference/lexical_analysis.html#whitespace-between-tokens)
/// or new-line characters.
pub const fn is_python_whitespace(c: char) -> bool {
matches!(
c,
' ' | '\n' | '\t' | '\r' |
// Form-feed
'\x0C'
)
}

View File

@ -7,7 +7,7 @@ rust-version = { workspace = true }
[dependencies] [dependencies]
ruff_formatter = { path = "../ruff_formatter" } ruff_formatter = { path = "../ruff_formatter" }
ruff_newlines = { path = "../ruff_newlines" } ruff_python_whitespace = { path = "../ruff_python_whitespace" }
ruff_python_ast = { path = "../ruff_python_ast" } ruff_python_ast = { path = "../ruff_python_ast" }
ruff_text_size = { workspace = true } ruff_text_size = { workspace = true }

View File

@ -1,10 +1,10 @@
use crate::comments::visitor::{CommentPlacement, DecoratedComment}; use crate::comments::visitor::{CommentPlacement, DecoratedComment};
use crate::comments::CommentTextPosition; use crate::comments::CommentTextPosition;
use crate::trivia::{SimpleTokenizer, TokenKind}; use crate::trivia::{SimpleTokenizer, TokenKind};
use ruff_newlines::StrExt;
use ruff_python_ast::node::AnyNodeRef; use ruff_python_ast::node::AnyNodeRef;
use ruff_python_ast::source_code::Locator; use ruff_python_ast::source_code::Locator;
use ruff_python_ast::whitespace; use ruff_python_ast::whitespace;
use ruff_python_whitespace::UniversalNewlines;
use ruff_text_size::{TextRange, TextSize}; use ruff_text_size::{TextRange, TextSize};
use rustpython_parser::ast::Ranged; use rustpython_parser::ast::Ranged;
use std::cmp::Ordering; use std::cmp::Ordering;

View File

@ -9,7 +9,7 @@ use ruff_python_ast::source_code::{CommentRanges, Locator};
// pre-order. // pre-order.
#[allow(clippy::wildcard_imports)] #[allow(clippy::wildcard_imports)]
use ruff_python_ast::visitor::preorder::*; use ruff_python_ast::visitor::preorder::*;
use ruff_python_ast::whitespace::is_python_whitespace; use ruff_python_whitespace::is_python_whitespace;
use ruff_text_size::TextRange; use ruff_text_size::TextRange;
use std::iter::Peekable; use std::iter::Peekable;

View File

@ -1,7 +1,9 @@
use ruff_python_ast::whitespace::is_python_whitespace;
use ruff_text_size::{TextLen, TextRange, TextSize};
use std::str::Chars; use std::str::Chars;
use ruff_text_size::{TextLen, TextRange, TextSize};
use ruff_python_whitespace::is_python_whitespace;
/// Searches for the first non-trivia character in `range`. /// Searches for the first non-trivia character in `range`.
/// ///
/// The search skips over any whitespace and comments. /// The search skips over any whitespace and comments.
@ -528,10 +530,11 @@ impl<'a> Cursor<'a> {
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use crate::trivia::{lines_after, lines_before, SimpleTokenizer, Token};
use insta::assert_debug_snapshot; use insta::assert_debug_snapshot;
use ruff_text_size::{TextLen, TextRange, TextSize}; use ruff_text_size::{TextLen, TextRange, TextSize};
use crate::trivia::{lines_after, lines_before, SimpleTokenizer, Token};
struct TokenizationTestCase { struct TokenizationTestCase {
source: &'static str, source: &'static str,
range: TextRange, range: TextRange,

View File

@ -1,5 +1,5 @@
[package] [package]
name = "ruff_newlines" name = "ruff_python_whitespace"
version = "0.0.0" version = "0.0.0"
publish = false publish = false
edition = { workspace = true } edition = { workspace = true }

View File

@ -0,0 +1,5 @@
mod newlines;
mod whitespace;
pub use newlines::*;
pub use whitespace::*;

View File

@ -5,11 +5,11 @@ use memchr::{memchr2, memrchr2};
use ruff_text_size::{TextLen, TextRange, TextSize}; use ruff_text_size::{TextLen, TextRange, TextSize};
/// Extension trait for [`str`] that provides a [`UniversalNewlineIterator`]. /// Extension trait for [`str`] that provides a [`UniversalNewlineIterator`].
pub trait StrExt { pub trait UniversalNewlines {
fn universal_newlines(&self) -> UniversalNewlineIterator<'_>; fn universal_newlines(&self) -> UniversalNewlineIterator<'_>;
} }
impl StrExt for str { impl UniversalNewlines for str {
fn universal_newlines(&self) -> UniversalNewlineIterator<'_> { fn universal_newlines(&self) -> UniversalNewlineIterator<'_> {
UniversalNewlineIterator::from(self) UniversalNewlineIterator::from(self)
} }
@ -22,7 +22,7 @@ impl StrExt for str {
/// ///
/// ```rust /// ```rust
/// # use ruff_text_size::TextSize; /// # use ruff_text_size::TextSize;
/// # use ruff_newlines::{Line, UniversalNewlineIterator}; /// # use ruff_python_whitespace::{Line, UniversalNewlineIterator};
/// let mut lines = UniversalNewlineIterator::from("foo\nbar\n\r\nbaz\rbop"); /// let mut lines = UniversalNewlineIterator::from("foo\nbar\n\r\nbaz\rbop");
/// ///
/// assert_eq!(lines.next_back(), Some(Line::new("bop", TextSize::from(14)))); /// assert_eq!(lines.next_back(), Some(Line::new("bop", TextSize::from(14))));

View File

@ -0,0 +1,15 @@
/// Returns `true` for [whitespace](https://docs.python.org/3/reference/lexical_analysis.html#whitespace-between-tokens)
/// characters.
pub const fn is_python_whitespace(c: char) -> bool {
matches!(
c,
// Space, tab, or form-feed
' ' | '\t' | '\x0C'
)
}
/// Extract the leading indentation from a line.
pub fn leading_indentation(line: &str) -> &str {
line.find(|char: char| !is_python_whitespace(char))
.map_or(line, |index| &line[..index])
}

View File

@ -6,5 +6,5 @@ edition = { workspace = true }
rust-version = { workspace = true } rust-version = { workspace = true }
[dependencies] [dependencies]
ruff_newlines = { path = "../ruff_newlines" } ruff_python_whitespace = { path = "../ruff_python_whitespace" }
ruff_text_size = { workspace = true } ruff_text_size = { workspace = true }

View File

@ -4,7 +4,7 @@
use std::borrow::Cow; use std::borrow::Cow;
use std::cmp; use std::cmp;
use ruff_newlines::StrExt; use ruff_python_whitespace::UniversalNewlines;
/// Indent each line by the given prefix. /// Indent each line by the given prefix.
/// ///

10
fuzz/Cargo.lock generated
View File

@ -1061,7 +1061,7 @@ dependencies = [
"ruff_cache", "ruff_cache",
"ruff_diagnostics", "ruff_diagnostics",
"ruff_macros", "ruff_macros",
"ruff_newlines", "ruff_python_whitespace",
"ruff_python_ast", "ruff_python_ast",
"ruff_python_semantic", "ruff_python_semantic",
"ruff_python_stdlib", "ruff_python_stdlib",
@ -1149,7 +1149,7 @@ dependencies = [
] ]
[[package]] [[package]]
name = "ruff_newlines" name = "ruff_python_whitespace"
version = "0.0.0" version = "0.0.0"
dependencies = [ dependencies = [
"memchr", "memchr",
@ -1169,7 +1169,7 @@ dependencies = [
"num-bigint", "num-bigint",
"num-traits", "num-traits",
"once_cell", "once_cell",
"ruff_newlines", "ruff_python_whitespace",
"ruff_text_size", "ruff_text_size",
"rustc-hash", "rustc-hash",
"rustpython-ast", "rustpython-ast",
@ -1190,7 +1190,7 @@ dependencies = [
"itertools", "itertools",
"once_cell", "once_cell",
"ruff_formatter", "ruff_formatter",
"ruff_newlines", "ruff_python_whitespace",
"ruff_python_ast", "ruff_python_ast",
"ruff_text_size", "ruff_text_size",
"rustc-hash", "rustc-hash",
@ -1242,7 +1242,7 @@ dependencies = [
name = "ruff_textwrap" name = "ruff_textwrap"
version = "0.0.0" version = "0.0.0"
dependencies = [ dependencies = [
"ruff_newlines", "ruff_python_whitespace",
"ruff_text_size", "ruff_text_size",
] ]