[IMP] server: refactor hooks to handle function evaluations

odoo · Oct 1, 2024 · 61d7343 · 61d7343
1 parent 5cd5df0
commit 61d7343
Show file tree

Hide file tree

Showing 4 changed files with 310 additions and 271 deletions.
diff --git a/server/Cargo.toml b/server/Cargo.toml
@@ -28,5 +28,6 @@ tracing-appender = "0.2.3"
 tracing-panic = "0.1.2"
 winapi = { version = "0.3.9", features = ["winbase", "processthreadsapi", "synchapi", "handleapi"] }
 ctrlc = "3.4.4"
+once_cell = "1.20.1"
 [target.'cfg(target_os = "linux")'.dependencies]
 nix = { version = "0.29.0", features = ["process"] }
diff --git a/server/src/core/python_arch_eval.rs b/server/src/core/python_arch_eval.rs
@@ -110,6 +110,7 @@ impl PythonArchEval {
         } else {
             //then Symbol must be a function
             symbol.borrow_mut().as_func_mut().replace_diagnostics(BuildSteps::ARCH_EVAL, self.diagnostics.clone());
+            PythonArchEvalHooks::on_function_eval(session.sync_odoo, self.sym_stack.first().unwrap().clone());
         }
         let mut symbol = self.sym_stack.first().unwrap().borrow_mut();
         symbol.set_build_status(BuildSteps::ARCH_EVAL, BuildStatus::DONE);
@@ -337,9 +338,6 @@ impl PythonArchEval {
     fn _visit_assign(&mut self, session: &mut SessionInfo, assign_stmt: &StmtAssign) {
         let assigns = python_utils::unpack_assign(&assign_stmt.targets, None, Some(&assign_stmt.value));
         for assign in assigns.iter() {
-            if assign.target.id.to_string() == "AbstractModel" {
-                println!("here");
-            }
             let variable = self.sym_stack.last().unwrap().borrow_mut().get_positioned_symbol(&assign.target.id.to_string(), &assign.target.range);
             if let Some(variable_rc) = variable {
                 let parent = variable_rc.borrow().parent().as_ref().unwrap().upgrade().unwrap().clone();