Update src/smpp_gateway/router.py

Co-authored-by: Tobias McNulty <tobias@caktusgroup.com>
caktus · Nov 14, 2024 · f7236b1 · f7236b1
1 parent 5f7133d
commit f7236b1
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/smpp_gateway/router.py b/src/smpp_gateway/router.py
@@ -32,10 +32,12 @@ class PriorityBlockingRouter(BlockingRouter):
     outgoing_message_class = PriorityOutgoingMessage
 
     def new_incoming_message(
-        self, text, connections, class_=incoming_message_class, **kwargs
+        self, text, connections, class_=None, **kwargs
     ):
-        return super().new_incoming_message(
-            text, connections, class_=PriorityIncomingMessage, **kwargs
+        if class_ is None:
+            class_ = self.incoming_message_class
+        self.return super().new_incoming_message(
+            text, connections, class_=class_, **kwargs
         )
 
     def new_outgoing_message(