mirror of https://github.com/astral-sh/ruff
20 lines
683 B
Rust
20 lines
683 B
Rust
use rustpython_parser::ast::{self, Expr, Stmt};
|
|
|
|
pub fn name(stmt: &Stmt) -> &str {
|
|
match stmt {
|
|
Stmt::FunctionDef(ast::StmtFunctionDef { name, .. })
|
|
| Stmt::AsyncFunctionDef(ast::StmtAsyncFunctionDef { name, .. }) => name.as_str(),
|
|
_ => panic!("Expected Stmt::FunctionDef | Stmt::AsyncFunctionDef"),
|
|
}
|
|
}
|
|
|
|
pub fn decorator_list(stmt: &Stmt) -> &[Expr] {
|
|
match stmt {
|
|
Stmt::FunctionDef(ast::StmtFunctionDef { decorator_list, .. })
|
|
| Stmt::AsyncFunctionDef(ast::StmtAsyncFunctionDef { decorator_list, .. }) => {
|
|
decorator_list
|
|
}
|
|
_ => panic!("Expected Stmt::FunctionDef | Stmt::AsyncFunctionDef"),
|
|
}
|
|
}
|