mirror of https://github.com/astral-sh/ruff
52 lines
1.8 KiB
Rust
52 lines
1.8 KiB
Rust
use crate::context::PyFormatContext;
|
|
use crate::expression::maybe_parenthesize_expression;
|
|
use crate::expression::parentheses::{NeedsParentheses, OptionalParentheses, Parenthesize};
|
|
use crate::{FormatNodeRule, PyFormatter};
|
|
use ruff_formatter::prelude::{space, text};
|
|
use ruff_formatter::{write, Buffer, FormatResult};
|
|
use ruff_python_ast::node::AnyNodeRef;
|
|
use rustpython_parser::ast::ExprYield;
|
|
|
|
#[derive(Default)]
|
|
pub struct FormatExprYield;
|
|
|
|
impl FormatNodeRule<ExprYield> for FormatExprYield {
|
|
fn fmt_fields(&self, item: &ExprYield, f: &mut PyFormatter) -> FormatResult<()> {
|
|
let ExprYield { range: _, value } = item;
|
|
|
|
if let Some(val) = value {
|
|
write!(
|
|
f,
|
|
[
|
|
text("yield"),
|
|
space(),
|
|
maybe_parenthesize_expression(val, item, Parenthesize::IfRequired)
|
|
]
|
|
)?;
|
|
} else {
|
|
write!(f, [&text("yield")])?;
|
|
}
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
impl NeedsParentheses for ExprYield {
|
|
fn needs_parentheses(
|
|
&self,
|
|
parent: AnyNodeRef,
|
|
_context: &PyFormatContext,
|
|
) -> OptionalParentheses {
|
|
// According to https://docs.python.org/3/reference/grammar.html There are two situations
|
|
// where we do not want to always parenthesize a yield expression:
|
|
// 1. Right hand side of an assignment, e.g. `x = yield y`
|
|
// 2. Yield statement, e.g. `def foo(): yield y`
|
|
// We catch situation 1 below. Situation 2 does not need to be handled here as
|
|
// FormatStmtExpr, does not add parenthesis
|
|
if parent.is_stmt_assign() || parent.is_stmt_ann_assign() || parent.is_stmt_aug_assign() {
|
|
OptionalParentheses::Multiline
|
|
} else {
|
|
OptionalParentheses::Always
|
|
}
|
|
}
|
|
}
|