Bug fix at 2024-06-27T10:58:00+0800

README.md

@@ -49,12 +49,12 @@ after_generate: Executes operations after generating each data entry but before
 final_generate: Executes operations after generating and processing all data entries. Suitable for final data processing, sending data to message queues, or performing statistical analysis.
 """
 @hook('pre_generate', order=1, hook_set='SET1')
-def pre_generate_first_hook(data):
+def pre_generate_first_hook():
     print("Start Mocking User Data")
 
 
 @hook('pre_generate', order=2, hook_set='SET1')
-def pre_generate_second_hook(data):
+def pre_generate_second_hook():
     """
     Perform some preprocessing operations, such as starting external services.
     """

pistol_magazine/hooks/hook_manager.py

@@ -29,7 +29,10 @@ def register_hook(self, hook_point, func, order=0, hook_set='default'):
 
     def trigger_hooks(self, hook_point, data, hook_set='default'):
         for func, _ in self.hooks.get(hook_set, {}).get(hook_point, []):
-            func(data)
+            if data is not None:
+                func(data)
+            else:
+                func()
         return data