fix: error on confirmatin

dooit-org · Nov 10, 2024 · ae31610 · ae31610
1 parent a5fa260
commit ae31610
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/dooit/ui/widgets/trees/_decorators.py b/dooit/ui/widgets/trees/_decorators.py
@@ -1,3 +1,4 @@
+from functools import partial
 from typing import Any, Callable, TYPE_CHECKING
 from textual.widgets.option_list import OptionDoesNotExist
 
@@ -48,9 +49,11 @@ def wrapper(self: "ModelTree", *args, **kwargs) -> Any:
 
 def require_confirmation(func: Callable) -> Callable:
     def wrapper(self: "ModelTree", *args, **kwargs) -> Any:
+        function = partial(func, self, *args, **kwargs)
+
         if not self.api.vars.show_confirm:
-            return func()
+            return function()
 
-        self.post_message(ShowConfirm(func, *args, **kwargs))
+        self.post_message(ShowConfirm(function))
 
     return wrapper
diff --git a/dooit/ui/widgets/trees/model_tree.py b/dooit/ui/widgets/trees/model_tree.py
@@ -310,8 +310,8 @@ def add_sibling(self):
         self.highlight_id(node.uuid)
         self.start_edit("description")
 
-    @refresh_tree
     @require_confirmation
+    @refresh_tree
     def _remove_node(self):
         self.current_model.drop()