mwojcik/nac3
1
0
Fork 0
forked from M-Labs/nac3
Code Pull Requests Activity

nac3artiq: detect decorators with arguments

Browse Source
This commit is contained in:
mwojcik 2024-09-10 17:22:19 +08:00
parent 7204513a9f
commit 987d17b5e8
1 changed files with 19 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
nac3artiq/src/lib.rs
View File

@ -191,14 +191,30 @@ impl Nac3 {
}) })
.unwrap() .unwrap()
}); });
body.retain(|stmt| { body.retain_mut(|stmt| {
if let StmtKind::FunctionDef { ref decorator_list, .. } = stmt.node { if let StmtKind::FunctionDef { ref mut decorator_list, .. } = stmt.node {
decorator_list.iter().any(|decorator| { decorator_list.iter_mut().any(|decorator| {
if let ExprKind::Name { id, .. } = decorator.node { if let ExprKind::Name { id, .. } = decorator.node {
id.to_string() == "kernel" id.to_string() == "kernel"
|| id.to_string() == "portable" || id.to_string() == "portable"
|| id.to_string() == "rpc" || id.to_string() == "rpc"
} else if let ExprKind::Call { func, .. } = &decorator.node {
// decorators with flags (e.g. rpc async) have Call for the node;
// this is to remove the middle
if let ExprKind::Name { id, .. } = func.node {
if id.to_string() == "rpc" {
println!("found rpc: {:?}", func);
println!("decorator node: {:?}", decorator.node);
decorator.node = func.clone().node;
true
} else { } else {
false
}
} else {
false
}
}
else {
false false
} }
}) })