language-servers/ruff
1
0
Fork 0
mirror of https://github.com/astral-sh/ruff.git synced 2025-09-28 21:05:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Avoid short-circuiting when detecting RET rules (#4002)

Browse source
This commit is contained in:
Charlie Marsh 2023-04-17 22:52:26 -04:00 • committed by GitHub
parent be87a29a9d
commit 79c47e29ee
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 33 additions and 47 deletions
Download patch file Download diff file Expand all files Collapse all files

66
crates/ruff/src/rules/flake8_return/rules.rs
View file

@ -463,34 +463,26 @@ fn superfluous_else_node(checker: &mut Checker, stmt: &Stmt, branch: Branch) ->
}
/// RET505, RET506, RET507, RET508
fn superfluous_elif(checker: &mut Checker, stack: &Stack) -> bool {
fn superfluous_elif(checker: &mut Checker, stack: &Stack) {
for stmt in &stack.elifs {
if superfluous_else_node(checker, stmt, Branch::Elif) {
return true;
superfluous_else_node(checker, stmt, Branch::Elif);
}
}
false
}
/// RET505, RET506, RET507, RET508
fn superfluous_else(checker: &mut Checker, stack: &Stack) -> bool {
fn superfluous_else(checker: &mut Checker, stack: &Stack) {
for stmt in &stack.elses {
if superfluous_else_node(checker, stmt, Branch::Else) {
return true;
superfluous_else_node(checker, stmt, Branch::Else);
}
}
false
}
/// Run all checks from the `flake8-return` plugin.
pub fn function(checker: &mut Checker, body: &[Stmt], returns: Option<&Expr>) {
// Skip empty functions.
if body.is_empty() {
return;
}
// Find the last statement in the function.
let last_stmt = body.last().unwrap();
let Some(last_stmt) = body.last() else {
// Skip empty functions.
return;
};
// Skip functions that consist of a single return statement.
if body.len() == 1 && matches!(last_stmt.node, StmtKind::Return { .. }) {
@ -511,20 +503,14 @@ pub fn function(checker: &mut Checker, body: &[Stmt], returns: Option<&Expr>) {
return;
}
if checker.settings.rules.enabled(Rule::SuperfluousElseReturn)
|| checker.settings.rules.enabled(Rule::SuperfluousElseRaise)
|| checker
.settings
.rules
.enabled(Rule::SuperfluousElseContinue)
|| checker.settings.rules.enabled(Rule::SuperfluousElseBreak)
{
if superfluous_elif(checker, &stack) {
return;
}
if superfluous_else(checker, &stack) {
return;
}
if checker.settings.rules.any_enabled(&[
Rule::SuperfluousElseReturn,
Rule::SuperfluousElseRaise,
Rule::SuperfluousElseContinue,
Rule::SuperfluousElseBreak,
]) {
superfluous_elif(checker, &stack);
superfluous_else(checker, &stack);
}
// Skip any functions without return statements.
@ -532,16 +518,8 @@ pub fn function(checker: &mut Checker, body: &[Stmt], returns: Option<&Expr>) {
return;
}
if !result_exists(&stack.returns) {
if checker.settings.rules.enabled(Rule::UnnecessaryReturnNone) {
// Skip functions that have a return annotation that is not `None`.
if returns.map_or(true, is_const_none) {
unnecessary_return_none(checker, &stack);
}
}
return;
}
// If we have at least one non-`None` return...
if result_exists(&stack.returns) {
if checker.settings.rules.enabled(Rule::ImplicitReturnValue) {
implicit_return_value(checker, &stack);
}
@ -556,4 +534,12 @@ pub fn function(checker: &mut Checker, body: &[Stmt], returns: Option<&Expr>) {
}
}
}
} else {
if checker.settings.rules.enabled(Rule::UnnecessaryReturnNone) {
// Skip functions that have a return annotation that is not `None`.
if returns.map_or(true, is_const_none) {
unnecessary_return_none(checker, &stack);
}
}
}
}