17 lines
588 B
Rust
17 lines
588 B
Rust
use rustpython_ast::{Expr, Stmt, StmtKind};
|
|
|
|
pub fn name(stmt: &Stmt) -> &str {
|
|
match &stmt.node {
|
|
StmtKind::FunctionDef { name, .. } | StmtKind::AsyncFunctionDef { name, .. } => name,
|
|
_ => panic!("Expected StmtKind::FunctionDef | StmtKind::AsyncFunctionDef"),
|
|
}
|
|
}
|
|
|
|
pub fn decorator_list(stmt: &Stmt) -> &Vec<Expr> {
|
|
match &stmt.node {
|
|
StmtKind::FunctionDef { decorator_list, .. }
|
|
| StmtKind::AsyncFunctionDef { decorator_list, .. } => decorator_list,
|
|
_ => panic!("Expected StmtKind::FunctionDef | StmtKind::AsyncFunctionDef"),
|
|
}
|
|
}
|