From 79e46bb696190a2753ff327b870b7f6b2e8befd7 Mon Sep 17 00:00:00 2001 From: Luigi Marini Date: Wed, 13 Nov 2024 09:19:11 -0600 Subject: [PATCH] Default to None if "owner" doesn't exists in msg --- backend/heartbeat_listener.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/backend/heartbeat_listener.py b/backend/heartbeat_listener.py index 6c82f6cbb..02560d21c 100644 --- a/backend/heartbeat_listener.py +++ b/backend/heartbeat_listener.py @@ -7,12 +7,11 @@ from aio_pika import connect_robust from aio_pika.abc import AbstractIncomingMessage -from packaging import version - from app.config import settings from app.main import startup_beanie from app.models.listeners import EventListenerDB, EventListenerOut, ExtractorInfo from app.routers.listeners import _process_incoming_v1_extractor_info +from packaging import version logging.basicConfig(level=logging.INFO) logger = logging.getLogger(__name__) @@ -31,7 +30,7 @@ async def callback(message: AbstractIncomingMessage): msg = json.loads(message.body.decode("utf-8")) extractor_info = msg["extractor_info"] - owner = msg["owner"] + owner = msg.get("owner") if owner is not None and owner != "": # Extractor name should match queue, which includes secret key with common extractor_info["name"] orig_properties = ExtractorInfo(**extractor_info)