Skip to content
Draft
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
adding-changes
  • Loading branch information
Gleekzone committed Sep 8, 2020
commit c0209478361eaaa07d4401e9afdde135a733b5c2
19 changes: 12 additions & 7 deletions sentry_sdk/integrations/chalice.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,9 +41,6 @@ def __call__(self, event, context):
reraise(*exc_info)


old_get_view_function_response = Chalice._get_view_function_response


def _get_view_function_response(app, view_function, function_args):
@wraps(view_function)
def wrapped_view_function(**function_args):
Expand Down Expand Up @@ -77,17 +74,25 @@ def wrapped_view_function(**function_args):
hub.flush()
raise

return old_get_view_function_response(
app, wrapped_view_function, function_args
)
return wrapped_view_function


class ChaliceIntegration(Integration):
identifier = "chalice"

@staticmethod
def setup_once():
old_get_view_function_response = Chalice._get_view_function_response

def sentry_event_response(app, view_function, function_args):
wrapped_view_function = _get_view_function_response(
app, view_function, function_args
)

return old_get_view_function_response(
app, wrapped_view_function, function_args
)

Chalice._get_view_function_response = _get_view_function_response
Chalice._get_view_function_response = sentry_event_response
# for everything else (like events)
chalice.app.EventSourceHandler = EventSourceHandler