Add a constructor

This commit is contained in:
Charles Marsh 2022-08-13 17:03:52 -04:00
parent 290b0091c5
commit 52afc02023
1 changed files with 8 additions and 1 deletions

View File

@ -9,6 +9,12 @@ struct Checker {
checks: Vec<Check>, checks: Vec<Check>,
} }
impl Checker {
fn new() -> Self {
Checker { checks: vec![] }
}
}
impl Visitor for Checker { impl Visitor for Checker {
fn visit_stmt(&mut self, stmt: &Stmt) { fn visit_stmt(&mut self, stmt: &Stmt) {
match &stmt.node { match &stmt.node {
@ -32,6 +38,7 @@ impl Visitor for Checker {
} }
_ => {} _ => {}
} }
walk_stmt(self, stmt); walk_stmt(self, stmt);
} }
@ -72,7 +79,7 @@ pub fn check_ast(python_ast: &Suite) -> Vec<Check> {
python_ast python_ast
.iter() .iter()
.flat_map(|stmt| { .flat_map(|stmt| {
let mut checker = Checker { checks: vec![] }; let mut checker = Checker::new();
checker.visit_stmt(stmt); checker.visit_stmt(stmt);
checker.checks checker.checks
}) })