continue updating

docs/demo/scripts/custom_fwtw_pull.py

@@ -0,0 +1,7 @@
+from wsimod.extensions import register_node_patch
+
+
+@register_node_patch("my_fwtw", "pull_distributed")
+def custom_pull_distributed(self, vqip, *args, **kwargs):
+    """A custom handler function."""
+    return self._patched_pull_distributed(vqip, tag="FWTW")

docs/demo/scripts/custom_reservoir_handler.py

@@ -2,24 +2,24 @@
 
 
 @register_node_patch("my_reservoir", "pull_set_handler", item="FWTW")
-def custom_handler_function(self, vqip, *args, **kwargs):
+def custom_pulls_fwtw(self, vqip, *args, **kwargs):
     """A custom handler function."""
     return self.tank.pull_storage(vqip)
 
 
 @register_node_patch("my_reservoir", "pull_check_handler", item="FWTW")
-def custom_handler_function(self, vqip, *args, **kwargs):
+def custom_pullc_fwtw(self, vqip, *args, **kwargs):
     """A custom handler function."""
     return self.tank.get_avail()
 
 
 @register_node_patch("my_reservoir", "pull_set_handler", item="default")
-def custom_handler_function(self, vqip, *args, **kwargs):
+def custom_pulls_default(self, vqip, *args, **kwargs):
     """A custom handler function."""
-    return self.pull_set_deny()
+    return self.pull_set_deny(vqip)
 
 
 @register_node_patch("my_reservoir", "pull_check_handler", item="default")
-def custom_handler_function(self, vqip, *args, **kwargs):
+def custom_pullc_default(self, vqip, *args, **kwargs):
     """A custom handler function."""
     return self.pull_check_deny()