mirror of
https://github.com/astral-sh/ruff
synced 2026-01-23 14:30:53 -05:00
787e2fd49d
## Summary Format import statements in all their variants. Specifically, this implemented formatting `StmtImport`, `StmtImportFrom` and `Alias`. ## Test Plan I added some custom snapshots, even though this has been covered well by black's tests.
23 lines
636 B
Rust
23 lines
636 B
Rust
use crate::{AsFormat, FormatNodeRule, PyFormatter};
|
|
use ruff_formatter::prelude::{space, text};
|
|
use ruff_formatter::{write, Buffer, Format, FormatResult};
|
|
use rustpython_parser::ast::Alias;
|
|
|
|
#[derive(Default)]
|
|
pub struct FormatAlias;
|
|
|
|
impl FormatNodeRule<Alias> for FormatAlias {
|
|
fn fmt_fields(&self, item: &Alias, f: &mut PyFormatter) -> FormatResult<()> {
|
|
let Alias {
|
|
range: _,
|
|
name,
|
|
asname,
|
|
} = item;
|
|
name.format().fmt(f)?;
|
|
if let Some(asname) = asname {
|
|
write!(f, [space(), text("as"), space(), asname.format()])?;
|
|
}
|
|
Ok(())
|
|
}
|
|
}
|