mirror of
https://github.com/astral-sh/ruff.git
synced 2025-09-29 13:25:17 +00:00
Simplify iteration idioms (#13834)
Remove unnecessary uses of `.as_ref()`, `.iter()`, `&**` and similar, mostly in situations when iterating over variables. Many of these changes are only possible following #13826, when we bumped our MSRV to 1.80: several useful implementations on `&Box<[T]>` were only stabilised in Rust 1.80. Some of these changes we could have done earlier, however.
This commit is contained in:
parent
7fd8e30eed
commit
72adb09bf3
36 changed files with 72 additions and 73 deletions
|
@ -732,7 +732,7 @@ impl<'a> Visitor<'a> for Checker<'a> {
|
|||
|
||||
// The default values of the parameters needs to be evaluated in the enclosing
|
||||
// scope.
|
||||
for parameter in &**parameters {
|
||||
for parameter in parameters {
|
||||
if let Some(expr) = parameter.default() {
|
||||
self.visit_expr(expr);
|
||||
}
|
||||
|
@ -744,7 +744,7 @@ impl<'a> Visitor<'a> for Checker<'a> {
|
|||
self.visit_type_params(type_params);
|
||||
}
|
||||
|
||||
for parameter in &**parameters {
|
||||
for parameter in parameters {
|
||||
if let Some(expr) = parameter.annotation() {
|
||||
if singledispatch && !parameter.is_variadic() {
|
||||
self.visit_runtime_required_annotation(expr);
|
||||
|
|
|
@ -141,7 +141,7 @@ pub(crate) fn fastapi_unused_path_parameter(
|
|||
.parameters
|
||||
.args
|
||||
.iter()
|
||||
.chain(function_def.parameters.kwonlyargs.iter())
|
||||
.chain(&function_def.parameters.kwonlyargs)
|
||||
.map(|ParameterWithDefault { parameter, .. }| {
|
||||
parameter_alias(parameter, checker.semantic())
|
||||
.unwrap_or_else(|| parameter.name.as_str())
|
||||
|
|
|
@ -116,7 +116,7 @@ fn is_infinite_iterator(arg: &Expr, semantic: &SemanticModel) -> bool {
|
|||
}
|
||||
|
||||
// Ex) `iterools.repeat(1, times=None)`
|
||||
for keyword in &**keywords {
|
||||
for keyword in keywords {
|
||||
if keyword.arg.as_ref().is_some_and(|name| name == "times") {
|
||||
if keyword.value.is_none_literal_expr() {
|
||||
return true;
|
||||
|
|
|
@ -110,7 +110,7 @@ fn check_log_record_attr_clash(checker: &mut Checker, extra: &Keyword) {
|
|||
..
|
||||
}) => {
|
||||
if checker.semantic().match_builtin_expr(func, "dict") {
|
||||
for keyword in &**keywords {
|
||||
for keyword in keywords {
|
||||
if let Some(attr) = &keyword.arg {
|
||||
if is_reserved_attr(attr) {
|
||||
checker.diagnostics.push(Diagnostic::new(
|
||||
|
|
|
@ -78,7 +78,7 @@ pub(crate) fn custom_type_var_return_type(
|
|||
let Some(self_or_cls_annotation) = args
|
||||
.posonlyargs
|
||||
.iter()
|
||||
.chain(args.args.iter())
|
||||
.chain(&args.args)
|
||||
.next()
|
||||
.and_then(|parameter_with_default| parameter_with_default.parameter.annotation.as_ref())
|
||||
else {
|
||||
|
|
|
@ -140,7 +140,7 @@ pub(crate) fn bad_exit_annotation(checker: &mut Checker, function: &StmtFunction
|
|||
let non_self_positional_args: SmallVec<[&ParameterWithDefault; 3]> = parameters
|
||||
.posonlyargs
|
||||
.iter()
|
||||
.chain(parameters.args.iter())
|
||||
.chain(¶meters.args)
|
||||
.skip(1)
|
||||
.collect();
|
||||
|
||||
|
|
|
@ -252,7 +252,7 @@ impl UnittestAssert {
|
|||
FxHashMap::with_capacity_and_hasher(args.len() + keywords.len(), FxBuildHasher);
|
||||
|
||||
// Process positional arguments.
|
||||
for (arg_name, value) in arg_spec.iter().zip(args.iter()) {
|
||||
for (arg_name, value) in arg_spec.iter().zip(args) {
|
||||
args_map.insert(arg_name, value);
|
||||
}
|
||||
|
||||
|
|
|
@ -65,7 +65,7 @@ pub(crate) fn if_with_same_arms(checker: &mut Checker, stmt_if: &ast::StmtIf) {
|
|||
if !current_branch
|
||||
.body
|
||||
.iter()
|
||||
.zip(following_branch.body.iter())
|
||||
.zip(following_branch.body)
|
||||
.all(|(stmt1, stmt2)| ComparableStmt::from(stmt1) == ComparableStmt::from(stmt2))
|
||||
{
|
||||
continue;
|
||||
|
|
|
@ -248,12 +248,7 @@ pub(crate) fn literal_comparisons(checker: &mut Checker, compare: &ast::ExprComp
|
|||
}
|
||||
|
||||
// Check each comparator in order.
|
||||
for (index, (op, next)) in compare
|
||||
.ops
|
||||
.iter()
|
||||
.zip(compare.comparators.iter())
|
||||
.enumerate()
|
||||
{
|
||||
for (index, (op, next)) in compare.ops.iter().zip(&compare.comparators).enumerate() {
|
||||
if helpers::is_constant_non_singleton(comparator) {
|
||||
comparator = next;
|
||||
continue;
|
||||
|
|
|
@ -62,10 +62,10 @@ impl Violation for TypeComparison {
|
|||
|
||||
/// E721
|
||||
pub(crate) fn type_comparison(checker: &mut Checker, compare: &ast::ExprCompare) {
|
||||
for (left, right) in std::iter::once(compare.left.as_ref())
|
||||
.chain(compare.comparators.iter())
|
||||
for (left, right) in std::iter::once(&*compare.left)
|
||||
.chain(&compare.comparators)
|
||||
.tuple_windows()
|
||||
.zip(compare.ops.iter())
|
||||
.zip(&compare.ops)
|
||||
.filter(|(_, op)| matches!(op, CmpOp::Eq | CmpOp::NotEq))
|
||||
.map(|((left, right), _)| (left, right))
|
||||
{
|
||||
|
|
|
@ -870,7 +870,7 @@ pub(crate) fn string_dot_format_missing_argument(
|
|||
let missing: Vec<String> = summary
|
||||
.autos
|
||||
.iter()
|
||||
.chain(summary.indices.iter())
|
||||
.chain(&summary.indices)
|
||||
.filter(|&&i| i >= args.len())
|
||||
.map(ToString::to_string)
|
||||
.chain(
|
||||
|
|
|
@ -76,7 +76,7 @@ pub(crate) fn compare_to_empty_string(
|
|||
|
||||
let mut first = true;
|
||||
for ((lhs, rhs), op) in std::iter::once(left)
|
||||
.chain(comparators.iter())
|
||||
.chain(comparators)
|
||||
.tuple_windows::<(&Expr, &Expr)>()
|
||||
.zip(ops)
|
||||
{
|
||||
|
|
|
@ -57,7 +57,7 @@ pub(crate) fn comparison_of_constant(
|
|||
comparators: &[Expr],
|
||||
) {
|
||||
for ((left, right), op) in std::iter::once(left)
|
||||
.chain(comparators.iter())
|
||||
.chain(comparators)
|
||||
.tuple_windows()
|
||||
.zip(ops)
|
||||
{
|
||||
|
|
|
@ -55,7 +55,7 @@ pub(crate) fn comparison_with_itself(
|
|||
comparators: &[Expr],
|
||||
) {
|
||||
for ((left, right), op) in std::iter::once(left)
|
||||
.chain(comparators.iter())
|
||||
.chain(comparators)
|
||||
.tuple_windows()
|
||||
.zip(ops)
|
||||
{
|
||||
|
|
|
@ -64,7 +64,7 @@ pub(crate) fn duplicate_bases(checker: &mut Checker, name: &str, arguments: Opti
|
|||
let bases = &arguments.args;
|
||||
|
||||
let mut seen: FxHashSet<&str> = FxHashSet::with_capacity_and_hasher(bases.len(), FxBuildHasher);
|
||||
for base in &**bases {
|
||||
for base in bases {
|
||||
if let Expr::Name(ast::ExprName { id, .. }) = base {
|
||||
if !seen.insert(id) {
|
||||
let mut diagnostic = Diagnostic::new(
|
||||
|
|
|
@ -97,10 +97,7 @@ fn is_magic_value(literal_expr: LiteralExpressionRef, allowed_types: &[ConstantT
|
|||
|
||||
/// PLR2004
|
||||
pub(crate) fn magic_value_comparison(checker: &mut Checker, left: &Expr, comparators: &[Expr]) {
|
||||
for (left, right) in std::iter::once(left)
|
||||
.chain(comparators.iter())
|
||||
.tuple_windows()
|
||||
{
|
||||
for (left, right) in std::iter::once(left).chain(comparators).tuple_windows() {
|
||||
// If both of the comparators are literals, skip rule for the whole expression.
|
||||
// R0133: comparison-of-constants
|
||||
if as_literal(left).is_some() && as_literal(right).is_some() {
|
||||
|
@ -108,7 +105,7 @@ pub(crate) fn magic_value_comparison(checker: &mut Checker, left: &Expr, compara
|
|||
}
|
||||
}
|
||||
|
||||
for comparison_expr in std::iter::once(left).chain(comparators.iter()) {
|
||||
for comparison_expr in std::iter::once(left).chain(comparators) {
|
||||
if let Some(value) = as_literal(comparison_expr) {
|
||||
if is_magic_value(value, &checker.settings.pylint.allow_magic_value_types) {
|
||||
checker.diagnostics.push(Diagnostic::new(
|
||||
|
|
|
@ -49,7 +49,7 @@ impl Violation for NanComparison {
|
|||
|
||||
/// PLW0177
|
||||
pub(crate) fn nan_comparison(checker: &mut Checker, left: &Expr, comparators: &[Expr]) {
|
||||
for expr in std::iter::once(left).chain(comparators.iter()) {
|
||||
for expr in std::iter::once(left).chain(comparators) {
|
||||
if let Some(qualified_name) = checker.semantic().resolve_qualified_name(expr) {
|
||||
match qualified_name.segments() {
|
||||
["numpy", "nan" | "NAN" | "NaN"] => {
|
||||
|
|
|
@ -108,7 +108,7 @@ pub(crate) fn repeated_equality_comparison(checker: &mut Checker, bool_op: &ast:
|
|||
// Break into sequences of consecutive comparisons.
|
||||
let mut sequences: Vec<(Vec<usize>, Vec<&Expr>)> = Vec::new();
|
||||
let mut last = None;
|
||||
for (index, comparator) in indices.iter().zip(comparators.iter()) {
|
||||
for (index, comparator) in indices.iter().zip(comparators) {
|
||||
if last.is_some_and(|last| last + 1 == *index) {
|
||||
let (indices, comparators) = sequences.last_mut().unwrap();
|
||||
indices.push(*index);
|
||||
|
|
|
@ -179,7 +179,7 @@ pub(crate) fn unnecessary_lambda(checker: &mut Checker, lambda: &ExprLambda) {
|
|||
if call_posargs.len() != lambda_posargs.len() {
|
||||
return;
|
||||
}
|
||||
for (param, arg) in lambda_posargs.iter().zip(call_posargs.iter()) {
|
||||
for (param, arg) in lambda_posargs.iter().zip(call_posargs) {
|
||||
let Expr::Name(ast::ExprName { id, .. }) = arg else {
|
||||
return;
|
||||
};
|
||||
|
|
|
@ -143,10 +143,10 @@ fn concatenate_expressions(expr: &Expr) -> Option<(Expr, Type)> {
|
|||
}
|
||||
// If the splat element is itself a list/tuple, insert them in the other list/tuple.
|
||||
Expr::List(ast::ExprList { elts, .. }) if matches!(type_, Type::List) => {
|
||||
other_elements.iter().chain(elts.iter()).cloned().collect()
|
||||
other_elements.iter().chain(elts).cloned().collect()
|
||||
}
|
||||
Expr::Tuple(ast::ExprTuple { elts, .. }) if matches!(type_, Type::Tuple) => {
|
||||
other_elements.iter().chain(elts.iter()).cloned().collect()
|
||||
other_elements.iter().chain(elts).cloned().collect()
|
||||
}
|
||||
_ => return None,
|
||||
};
|
||||
|
|
|
@ -189,12 +189,12 @@ fn should_be_fstring(
|
|||
.filter_map(ast::Expr::as_call_expr)
|
||||
{
|
||||
let ast::Arguments { keywords, args, .. } = &expr.arguments;
|
||||
for keyword in &**keywords {
|
||||
for keyword in keywords {
|
||||
if let Some(ident) = keyword.arg.as_ref() {
|
||||
arg_names.insert(&ident.id);
|
||||
}
|
||||
}
|
||||
for arg in &**args {
|
||||
for arg in args {
|
||||
if let ast::Expr::Name(ast::ExprName { id, .. }) = arg {
|
||||
arg_names.insert(id);
|
||||
}
|
||||
|
|
|
@ -100,7 +100,7 @@ impl<'a> StatementVisitor<'a> for RaiseStatementVisitor<'a> {
|
|||
Stmt::Try(ast::StmtTry {
|
||||
body, finalbody, ..
|
||||
}) => {
|
||||
for stmt in body.iter().chain(finalbody.iter()) {
|
||||
for stmt in body.iter().chain(finalbody) {
|
||||
walk_stmt(self, stmt);
|
||||
}
|
||||
}
|
||||
|
|
|
@ -135,9 +135,8 @@ impl std::fmt::Display for SourceKindDiff<'_> {
|
|||
}
|
||||
DiffKind::IpyNotebook(original, modified) => {
|
||||
// Cell indices are 1-based.
|
||||
for ((idx, src_cell), dst_cell) in (1u32..)
|
||||
.zip(original.cells().iter())
|
||||
.zip(modified.cells().iter())
|
||||
for ((idx, src_cell), dst_cell) in
|
||||
(1u32..).zip(original.cells()).zip(modified.cells())
|
||||
{
|
||||
let (Cell::Code(src_cell), Cell::Code(dst_cell)) = (src_cell, dst_cell) else {
|
||||
continue;
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue