Call complete, rendered, and parent_rendered hooks on render.

Author: adamghill

django_unicorn/components/unicorn_template_response.py

@@ -96,6 +96,7 @@ def render(self):
 
         rendered_template = UnicornTemplateResponse._desoupify(soup)
         rendered_template = mark_safe(rendered_template)
+        self.component.rendered(rendered_template)
 
         response.content = rendered_template
 

django_unicorn/components/unicorn_view.py

@@ -115,6 +115,7 @@ def construct_component(
 
     component.mount()
     component.hydrate()
+    component.complete()
     component._validate_called = False
 
     return component
@@ -235,7 +236,7 @@ def hydrate(self):
 
     def complete(self):
         """
-        Hook that gets called when the component's methods are all called.
+        Hook that gets called after all component methods are executed.
         """
         pass
 
@@ -658,6 +659,8 @@ def _get_component_class(
             return component_class
 
         if use_cache and component_id in constructed_views_cache:
+            # Note that `hydrate()` and `complete` don't need to be called here
+            # because this path only happens for re-rendering from the view
             component = constructed_views_cache[component_id]
             component.setup(request)
             component._validate_called = False

django_unicorn/views.py

@@ -434,6 +434,8 @@ def _process_component_request(
         else:
             raise UnicornViewError(f"Unknown action_type '{action_type}'")
 
+    component.complete()
+
     # Re-load frontend context variables to deal with non-serializable properties
     component_request.data = orjson.loads(component.get_frontend_context_variables())
 
@@ -454,6 +456,8 @@ def _process_component_request(
             component.validate(model_names=list(updated_data.keys()))
 
     rendered_component = component.render()
+    component.rendered(rendered_component)
+
     partial_doms = []
 
     if partials and all(partials):
@@ -540,6 +544,7 @@ def _process_component_request(
 
         if not partial_doms:
             parent_dom = parent_component.render()
+            component.parent_rendered(parent_dom)
 
             parent.update(
                 {