simplify class handlers

Author: rockem

.gitignore

@@ -1,3 +1,4 @@
 venv
 .pytest_cache
-.idea
+.idea
+__pycache__

README.md

@@ -10,7 +10,7 @@ def handle_user_login(message):
     User.updateLastLoginTime()
 
 # Class handlers
-@app.handler_class('user_logged_in')
+@app.handler('user_logged_in')
 class UserLoginHandler:
     def handle(self, message):
         User.updateLastLoginTime()

eggplant/core.py

@@ -6,15 +6,9 @@ def __init__(self, broker):
         self._broker = broker
         self._handlers = {}
 
-    def handler(self, message_name):
+    def handler(self, topic):
         def decorator_handler(func):
-            self._handlers.update({message_name: func})
-
-        return decorator_handler
-
-    def handler_class(self, topic):
-        def decorator_handler(clazz):
-            self._handlers.update({topic: clazz})
+            self._handlers.update({topic: func})
 
         return decorator_handler
 

tests/test_core.py

@@ -6,7 +6,7 @@ def test_consume_with_class_handler():
     app = Eggplant(broker)
     received_messages = []
 
-    @app.handler_class('password_expired')
+    @app.handler('password_expired')
     class StubHandler:
         def handle(self, message):
             received_messages.append(message)