Range formatting API (#9635)

crates/ruff_python_formatter/src/statement/suite.rs

@@ -2,8 +2,8 @@ use ruff_formatter::{
     write, FormatContext, FormatOwnedWithRule, FormatRefWithRule, FormatRuleWithOptions,
 };
 use ruff_python_ast::helpers::is_compound_statement;
-use ruff_python_ast::AnyNodeRef;
 use ruff_python_ast::{self as ast, Expr, PySourceType, Stmt, Suite};
+use ruff_python_ast::{AnyNodeRef, StmtExpr};
 use ruff_python_trivia::{lines_after, lines_after_ignoring_end_of_line_trivia, lines_before};
 use ruff_text_size::{Ranged, TextRange};
 
@@ -740,6 +740,14 @@ impl<'a> DocstringStmt<'a> {
             _ => None,
         }
     }
+
+    pub(crate) fn is_docstring_statement(stmt: &StmtExpr) -> bool {
+        if let Expr::StringLiteral(ast::ExprStringLiteral { value, .. }) = stmt.value.as_ref() {
+            !value.is_implicit_concatenated()
+        } else {
+            false
+        }
+    }
 }
 
 impl Format<PyFormatContext<'_>> for DocstringStmt<'_> {