nac3artiq: filter class definitions

This commit is contained in:
Sebastien Bourdeauducq 2021-09-27 22:25:19 +08:00
parent 6141f01180
commit bfd041d361
1 changed files with 15 additions and 6 deletions

View File

@ -88,13 +88,22 @@ impl Nac3 {
let parser_result = parser::parse_program(&source).map_err(|e| let parser_result = parser::parse_program(&source).map_err(|e|
exceptions::PySyntaxError::new_err(format!("failed to parse host object source: {}", e)))?; exceptions::PySyntaxError::new_err(format!("failed to parse host object source: {}", e)))?;
for stmt in parser_result.into_iter() { for mut stmt in parser_result.into_iter() {
let include = match &stmt.node { let include = match stmt.node {
ast::StmtKind::ClassDef { decorator_list, .. } => { ast::StmtKind::ClassDef { ref decorator_list, ref mut body, .. } => {
let kernels = decorator_list.iter().any(|decorator| if let ast::ExprKind::Name { id, .. } = decorator.node
{ id.to_string() == "kernel" || id.to_string() == "portable" } else { false });
body.retain(|stmt| {
if let ast::StmtKind::FunctionDef { ref decorator_list, .. } = stmt.node {
decorator_list.iter().any(|decorator| if let ast::ExprKind::Name { id, .. } = decorator.node decorator_list.iter().any(|decorator| if let ast::ExprKind::Name { id, .. } = decorator.node
{ id.to_string() == "kernel" || id.to_string() == "portable" } else { false }) { id.to_string() == "kernel" || id.to_string() == "portable" } else { false })
} else {
true
}
});
kernels
}, },
ast::StmtKind::FunctionDef { decorator_list, .. } => { ast::StmtKind::FunctionDef { ref decorator_list, .. } => {
decorator_list.iter().any(|decorator| if let ast::ExprKind::Name { id, .. } = decorator.node decorator_list.iter().any(|decorator| if let ast::ExprKind::Name { id, .. } = decorator.node
{ id.to_string() == "extern" || id.to_string() == "portable" } else { false }) { id.to_string() == "extern" || id.to_string() == "portable" } else { false })
}, },