mirror of https://github.com/astral-sh/ruff
119 lines
3.4 KiB
Rust
119 lines
3.4 KiB
Rust
use ruff_formatter::{write, FormatOwnedWithRule, FormatRefWithRule};
|
|
use ruff_python_ast::node::AnyNodeRef;
|
|
use ruff_python_ast::{CmpOp, ExprCompare};
|
|
|
|
use crate::comments::{leading_comments, SourceComment};
|
|
use crate::expression::parentheses::{
|
|
in_parentheses_only_group, in_parentheses_only_soft_line_break_or_space, NeedsParentheses,
|
|
OptionalParentheses,
|
|
};
|
|
use crate::prelude::*;
|
|
|
|
#[derive(Default)]
|
|
pub struct FormatExprCompare;
|
|
|
|
impl FormatNodeRule<ExprCompare> for FormatExprCompare {
|
|
fn fmt_fields(&self, item: &ExprCompare, f: &mut PyFormatter) -> FormatResult<()> {
|
|
let ExprCompare {
|
|
range: _,
|
|
left,
|
|
ops,
|
|
comparators,
|
|
} = item;
|
|
|
|
let comments = f.context().comments().clone();
|
|
|
|
let inner = format_with(|f| {
|
|
write!(f, [in_parentheses_only_group(&left.format())])?;
|
|
|
|
assert_eq!(comparators.len(), ops.len());
|
|
|
|
for (operator, comparator) in ops.iter().zip(comparators) {
|
|
let leading_comparator_comments = comments.leading(comparator);
|
|
if leading_comparator_comments.is_empty() {
|
|
write!(f, [in_parentheses_only_soft_line_break_or_space()])?;
|
|
} else {
|
|
// Format the expressions leading comments **before** the operator
|
|
write!(
|
|
f,
|
|
[
|
|
hard_line_break(),
|
|
leading_comments(leading_comparator_comments)
|
|
]
|
|
)?;
|
|
}
|
|
|
|
write!(
|
|
f,
|
|
[
|
|
operator.format(),
|
|
space(),
|
|
in_parentheses_only_group(&comparator.format())
|
|
]
|
|
)?;
|
|
}
|
|
|
|
Ok(())
|
|
});
|
|
|
|
in_parentheses_only_group(&inner).fmt(f)
|
|
}
|
|
|
|
fn fmt_dangling_comments(
|
|
&self,
|
|
_dangling_comments: &[SourceComment],
|
|
_f: &mut PyFormatter,
|
|
) -> FormatResult<()> {
|
|
// Node can not have dangling comments
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
impl NeedsParentheses for ExprCompare {
|
|
fn needs_parentheses(
|
|
&self,
|
|
_parent: AnyNodeRef,
|
|
_context: &PyFormatContext,
|
|
) -> OptionalParentheses {
|
|
OptionalParentheses::Multiline
|
|
}
|
|
}
|
|
|
|
#[derive(Copy, Clone)]
|
|
pub struct FormatCmpOp;
|
|
|
|
impl<'ast> AsFormat<PyFormatContext<'ast>> for CmpOp {
|
|
type Format<'a> = FormatRefWithRule<'a, CmpOp, FormatCmpOp, PyFormatContext<'ast>>;
|
|
|
|
fn format(&self) -> Self::Format<'_> {
|
|
FormatRefWithRule::new(self, FormatCmpOp)
|
|
}
|
|
}
|
|
|
|
impl<'ast> IntoFormat<PyFormatContext<'ast>> for CmpOp {
|
|
type Format = FormatOwnedWithRule<CmpOp, FormatCmpOp, PyFormatContext<'ast>>;
|
|
|
|
fn into_format(self) -> Self::Format {
|
|
FormatOwnedWithRule::new(self, FormatCmpOp)
|
|
}
|
|
}
|
|
|
|
impl FormatRule<CmpOp, PyFormatContext<'_>> for FormatCmpOp {
|
|
fn fmt(&self, item: &CmpOp, f: &mut PyFormatter) -> FormatResult<()> {
|
|
let operator = match item {
|
|
CmpOp::Eq => "==",
|
|
CmpOp::NotEq => "!=",
|
|
CmpOp::Lt => "<",
|
|
CmpOp::LtE => "<=",
|
|
CmpOp::Gt => ">",
|
|
CmpOp::GtE => ">=",
|
|
CmpOp::Is => "is",
|
|
CmpOp::IsNot => "is not",
|
|
CmpOp::In => "in",
|
|
CmpOp::NotIn => "not in",
|
|
};
|
|
|
|
token(operator).fmt(f)
|
|
}
|
|
}
|