File tree Expand file tree Collapse file tree 1 file changed +5
-6
lines changed
polytope_server/common/queue Expand file tree Collapse file tree 1 file changed +5
-6
lines changed Original file line number Diff line number Diff line change 1818# does it submit to any jurisdiction.
1919#
2020
21- import importlib
2221from abc import ABC , abstractmethod
2322
23+ from polytope_server .common .queue .rabbitmq_queue import RabbitmqQueue
24+ from polytope_server .common .queue .sqs_queue import SQSQueue
25+
2426
2527class Message :
2628 def __init__ (self , body , context = None ):
@@ -70,12 +72,9 @@ def get_type(self) -> str:
7072 """Get the implementation type"""
7173
7274
73- queue_dict = {"rabbitmq" : " RabbitmqQueue" , "sqs" : " SQSQueue" }
75+ queue_types = {"rabbitmq" : RabbitmqQueue , "sqs" : SQSQueue }
7476
7577
7678def create_queue (queue_config ) -> Queue :
77-
7879 queue_type = next (iter (queue_config .keys ()), "rabbitmq" )
79-
80- QueueClass = importlib .import_module ("polytope_server.common.queue." + queue_type + "_queue" )
81- return getattr (QueueClass , queue_dict [queue_type ])(queue_config [queue_type ])
80+ return queue_types [queue_type ](queue_config [queue_type ])
You can’t perform that action at this time.
0 commit comments