From a8110428b796c8f8da83517841714c8a95a0e54e Mon Sep 17 00:00:00 2001 From: Koen Date: Wed, 14 Feb 2024 13:40:50 +0200 Subject: [PATCH] fix: Check for no microseconds in timestamp for pubsub message handler --- src/firebase_functions/pubsub_fn.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/firebase_functions/pubsub_fn.py b/src/firebase_functions/pubsub_fn.py index 6d833d3..9f08c5f 100644 --- a/src/firebase_functions/pubsub_fn.py +++ b/src/firebase_functions/pubsub_fn.py @@ -105,6 +105,12 @@ def _message_handler( data = event_dict["data"] message_dict = data["message"] + # if no microseconds are present, we should set them to 0 to prevent parsing from failing + if "." not in event_dict["time"]: + event_dict["time"] = event_dict["time"].replace("Z", ".000000Z") + if "." not in message_dict["publish_time"]: + message_dict["publish_time"] = message_dict["publish_time"].replace("Z", ".000000Z") + time = _dt.datetime.strptime( event_dict["time"], "%Y-%m-%dT%H:%M:%S.%f%z",