25 lines
831 B
Rust
25 lines
831 B
Rust
use crate::expression::parentheses::{is_expression_parenthesized, Parenthesize};
|
|
use crate::expression::string::StringLayout;
|
|
use crate::prelude::*;
|
|
use crate::FormatNodeRule;
|
|
use rustpython_parser::ast::StmtExpr;
|
|
|
|
#[derive(Default)]
|
|
pub struct FormatStmtExpr;
|
|
|
|
impl FormatNodeRule<StmtExpr> for FormatStmtExpr {
|
|
fn fmt_fields(&self, item: &StmtExpr, f: &mut PyFormatter) -> FormatResult<()> {
|
|
let StmtExpr { value, .. } = item;
|
|
|
|
if let Some(constant) = value.as_constant_expr() {
|
|
if constant.value.is_str()
|
|
&& !is_expression_parenthesized(value.as_ref().into(), f.context().contents())
|
|
{
|
|
return constant.format().with_options(StringLayout::Flat).fmt(f);
|
|
}
|
|
}
|
|
|
|
value.format().with_options(Parenthesize::Optional).fmt(f)
|
|
}
|
|
}
|