Add @my_context.consumer decorator.

pelme · Aug 18, 2024 · a54c3d5 · a54c3d5
1 parent 54d64bb
commit a54c3d5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/htpy/__init__.py b/htpy/__init__.py
@@ -141,6 +141,9 @@ def provide(self, value: T, children_func: Callable[[], Node]) -> ContextProvide
     def use(self, children_func: Callable[[T], Node]) -> ContextUse[T]:
         return ContextUse(self, children_func)
 
+    def consumer(self, func: Callable[[T], Node]) -> Callable[[], Node]:
+        return lambda: ContextUse(self, func)
+
 
 ContextDict: t.TypeAlias = dict[t.Any, t.Any]