X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=osm_policy_module%2Fcommon%2Fmessage_bus_client.py;h=4073d0fd025d0fb53b6d70651fbec79057f12c65;hb=4584f8e86a492d67d120bfea1195eff1475c0a65;hp=ea5095de6b7cedbb7a799ea0646331233944c5de;hpb=e8ee171063d81a02033729273793f5ba2c2b71cc;p=osm%2FPOL.git diff --git a/osm_policy_module/common/message_bus_client.py b/osm_policy_module/common/message_bus_client.py index ea5095d..4073d0f 100644 --- a/osm_policy_module/common/message_bus_client.py +++ b/osm_policy_module/common/message_bus_client.py @@ -31,13 +31,15 @@ from osm_policy_module.core.config import Config class MessageBusClient: def __init__(self, config: Config, loop=None): - if config.get('message', 'driver') == "local": + if config.get("message", "driver") == "local": self.msg_bus = msglocal.MsgLocal() - elif config.get('message', 'driver') == "kafka": + elif config.get("message", "driver") == "kafka": self.msg_bus = msgkafka.MsgKafka() else: - raise Exception("Unknown message bug driver {}".format(config.get('section', 'driver'))) - self.msg_bus.connect(config.get('message')) + raise Exception( + "Unknown message bug driver {}".format(config.get("section", "driver")) + ) + self.msg_bus.connect(config.get("message")) if not loop: loop = asyncio.get_event_loop() self.loop = loop