feat: auto detect extra params for formatters

dooit/ui/api/api_components/formatters/formatter_store.py

@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Any, Callable, List, Optional
+from typing import TYPE_CHECKING, Any, Callable, Dict, List, Optional
 from uuid import uuid4
 from dataclasses import dataclass
 
@@ -80,11 +80,26 @@ def current_formatter(self) -> FormatterFunc:
 
         return enabled_formatters[-1]
 
+    def _get_function_params(self, func: Callable) -> List[str]:
+        return list(func.__code__.co_varnames)
+
     def format_value(self, value: Any, model: ModelType) -> str:
+        params = dict(api=self.api)
+
+        def get_extra_args(func: Callable) -> Dict[str, Any]:
+            func_params = self._get_function_params(func)
+            extra_args = {}
+
+            for param in func_params:
+                if param in params:
+                    extra_args[param] = params[param]
+
+            return extra_args
+
         res = None
 
         for func in reversed(self.formatter_functions):
-            res = func(value, model)
+            res = func(value, model, **get_extra_args(func))
             if res is not None:
                 return res
 

dooit/utils/default_config.py

@@ -8,7 +8,6 @@
 from dooit.ui.events.events import ModeChanged, Startup
 from dooit.ui.widgets.bars import StatusBarWidget
 from rich.text import Text
-from functools import partial
 
 
 @subscribe(ModeChanged)
@@ -173,9 +172,7 @@ def formatter_setup(api: DooitAPI, _):
     api.formatter.todos.status.add(todo_status_formatter)
     api.formatter.todos.description.add(todo_desc_formatter)
     api.formatter.todos.due.add(todo_due_formatter)
-    api.formatter.todos.urgency.add(
-        partial(todo_urgency_formatter, api=api),
-    )
+    api.formatter.todos.urgency.add(todo_urgency_formatter)
 
 
 @subscribe(Startup)