From 0f0ad73c2288ee967043693298648fc9da1c5492 Mon Sep 17 00:00:00 2001 From: Alex Waygood Date: Mon, 27 Jan 2025 18:39:17 +0000 Subject: [PATCH] [minor] Simplify some `ExprStringLiteral` creation logic --- .../flake8_simplify/rules/split_static_string.rs | 13 +++++-------- .../rules/ruff/rules/assert_with_print_message.rs | 15 ++++++--------- crates/ruff_python_formatter/tests/normalizer.rs | 2 +- 3 files changed, 12 insertions(+), 18 deletions(-) diff --git a/crates/ruff_linter/src/rules/flake8_simplify/rules/split_static_string.rs b/crates/ruff_linter/src/rules/flake8_simplify/rules/split_static_string.rs index 23e85f45a484c..f22e7d8664e6b 100644 --- a/crates/ruff_linter/src/rules/flake8_simplify/rules/split_static_string.rs +++ b/crates/ruff_linter/src/rules/flake8_simplify/rules/split_static_string.rs @@ -3,8 +3,8 @@ use std::cmp::Ordering; use ruff_diagnostics::{Applicability, Diagnostic, Edit, Fix, FixAvailability, Violation}; use ruff_macros::{derive_message_formats, ViolationMetadata}; use ruff_python_ast::{ - Expr, ExprCall, ExprContext, ExprList, ExprStringLiteral, ExprUnaryOp, StringLiteral, - StringLiteralFlags, StringLiteralValue, UnaryOp, + Expr, ExprCall, ExprContext, ExprList, ExprUnaryOp, StringLiteral, StringLiteralFlags, + StringLiteralValue, UnaryOp, }; use ruff_text_size::{Ranged, TextRange}; @@ -120,13 +120,10 @@ fn construct_replacement(elts: &[&str], flags: StringLiteralFlags) -> Expr { elts: elts .iter() .map(|elt| { - Expr::StringLiteral(ExprStringLiteral { - value: StringLiteralValue::single(StringLiteral { - value: Box::from(*elt), - range: TextRange::default(), - flags, - }), + Expr::from(StringLiteral { + value: Box::from(*elt), range: TextRange::default(), + flags, }) }) .collect(), diff --git a/crates/ruff_linter/src/rules/ruff/rules/assert_with_print_message.rs b/crates/ruff_linter/src/rules/ruff/rules/assert_with_print_message.rs index cbd0721da4d65..9e01401cf8022 100644 --- a/crates/ruff_linter/src/rules/ruff/rules/assert_with_print_message.rs +++ b/crates/ruff_linter/src/rules/ruff/rules/assert_with_print_message.rs @@ -90,9 +90,9 @@ pub(crate) fn assert_with_print_message(checker: &mut Checker, stmt: &ast::StmtA mod print_arguments { use itertools::Itertools; use ruff_python_ast::{ - Arguments, ConversionFlag, Expr, ExprFString, ExprStringLiteral, FString, FStringElement, - FStringElements, FStringExpressionElement, FStringFlags, FStringLiteralElement, - FStringValue, StringLiteral, StringLiteralFlags, StringLiteralValue, + Arguments, ConversionFlag, Expr, ExprFString, FString, FStringElement, FStringElements, + FStringExpressionElement, FStringFlags, FStringLiteralElement, FStringValue, StringLiteral, + StringLiteralFlags, }; use ruff_text_size::TextRange; @@ -202,13 +202,10 @@ mod print_arguments { }) .join(&sep_string); - Some(Expr::StringLiteral(ExprStringLiteral { + Some(Expr::from(StringLiteral { + value: combined_string.into(), + flags, range: TextRange::default(), - value: StringLiteralValue::single(StringLiteral { - value: combined_string.into(), - flags, - range: TextRange::default(), - }), })) } diff --git a/crates/ruff_python_formatter/tests/normalizer.rs b/crates/ruff_python_formatter/tests/normalizer.rs index cd745171a29ab..c2b3fb76da812 100644 --- a/crates/ruff_python_formatter/tests/normalizer.rs +++ b/crates/ruff_python_formatter/tests/normalizer.rs @@ -79,7 +79,7 @@ impl Transformer for Normalizer { if can_join { string.value = ast::StringLiteralValue::single(ast::StringLiteral { - value: string.value.to_str().to_string().into_boxed_str(), + value: Box::from(string.value.to_str()), range: string.range, flags: StringLiteralFlags::empty(), });