Skip to content

Commit 79e46bb

Browse files
committed
Default to None if "owner" doesn't exists in msg
1 parent 96a61d1 commit 79e46bb

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

backend/heartbeat_listener.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,12 +7,11 @@
77

88
from aio_pika import connect_robust
99
from aio_pika.abc import AbstractIncomingMessage
10-
from packaging import version
11-
1210
from app.config import settings
1311
from app.main import startup_beanie
1412
from app.models.listeners import EventListenerDB, EventListenerOut, ExtractorInfo
1513
from app.routers.listeners import _process_incoming_v1_extractor_info
14+
from packaging import version
1615

1716
logging.basicConfig(level=logging.INFO)
1817
logger = logging.getLogger(__name__)
@@ -31,7 +30,7 @@ async def callback(message: AbstractIncomingMessage):
3130
msg = json.loads(message.body.decode("utf-8"))
3231

3332
extractor_info = msg["extractor_info"]
34-
owner = msg["owner"]
33+
owner = msg.get("owner")
3534
if owner is not None and owner != "":
3635
# Extractor name should match queue, which includes secret key with common extractor_info["name"]
3736
orig_properties = ExtractorInfo(**extractor_info)

0 commit comments

Comments
 (0)