From 4bcb8ba27f81bbf8647e1a5bf9a0fb4df2be4cd1 Mon Sep 17 00:00:00 2001 From: Claire Neveu Date: Mon, 19 Oct 2020 15:19:33 -0400 Subject: [PATCH] Correctly handle messages from the pubsub emulator --- src/functions_framework/__init__.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/src/functions_framework/__init__.py b/src/functions_framework/__init__.py index 575c93d9..5a87a0b1 100644 --- a/src/functions_framework/__init__.py +++ b/src/functions_framework/__init__.py @@ -88,6 +88,18 @@ def _run_legacy_event(function, request): event_data = request.get_json() if not event_data: flask.abort(400) + + ############### + # Need to add this code to make pubsub events work. + if 'message' in event_data: + if 'data' not in event_data: + message = event_data['message'] + event_data['data'] = { + 'data': message.get('data'), + 'attributes': message.get('attributes') + } + ############### + event_object = _Event(**event_data) data = event_object.data context = Context(**event_object.context)