pass test

Author: baoj2010

bali/decorators.py

@@ -158,7 +158,7 @@ def event_handler(event_type):
     # find queue by event_type
     def decorator(func):
         @functools.wraps(func)
-        def wrapper(body, message):
+        def wrapper(handler, body, message):
             try:
                 if isinstance(body, str):
                     body = json.loads(body)
@@ -175,13 +175,15 @@ def wrapper(body, message):
                     ):
                         event = param.annotation(**body)
                         break
-                res = func(HANDLER, event or body)
+                if not handler:
+                    handler = HANDLER
+                res = func(handler, event or body)
                 message.ack()
                 return res
             except:
                 logger.error(traceback.format_exc())
-
-        register_callback(event_type, wrapper)
+        callback = functools.partial(wrapper, None)
+        register_callback(event_type, callback)
         return wrapper
 
     return decorator