cleanup

Author: Archmonger

src/django_idom/hooks.py

@@ -112,6 +112,7 @@ def use_query(
 
     Args:
         query: A callable that returns a Django `Model` or `QuerySet`.
+        options: A `QueryOptions` object that can modify how the query is excuted.
         *args: Positional arguments to pass into `query`.
 
     Keyword Args:
@@ -158,7 +159,7 @@ def execute_query() -> None:
 
             # Use a custom postprocessor, if provided
             if query_options.postprocessor:
-                query_options.postprocessor(query_options)
+                query_options.postprocessor(data, query_options)
 
             # Use the default postprocessor
             else:

src/django_idom/types.py

@@ -62,6 +62,6 @@ class QueryOptions:
     )
     """Configuration values usable by the `postprocessor`."""
 
-    postprocessor: Callable[[QueryOptions], None] | None = None
+    postprocessor: Callable[[_Data, QueryOptions], None] | None = None
     """A post processing callable that can read/modify the `QueryOptions` object. If unset, the default fetch
     handler is used to prevent lazy loading of Django fields."""