From 4937d84b7a2a8a08208266fde9752ae54eb2458a Mon Sep 17 00:00:00 2001 From: Joe Haines Date: Tue, 2 Apr 2024 09:09:39 +0100 Subject: [PATCH] Add aws_lambda_handler legacy function delegator --- bugsnag/__init__.py | 6 ++++-- bugsnag/legacy.py | 9 ++++++++- 2 files changed, 12 insertions(+), 3 deletions(-) diff --git a/bugsnag/__init__.py b/bugsnag/__init__.py index 5a74a852..9c023b2f 100644 --- a/bugsnag/__init__.py +++ b/bugsnag/__init__.py @@ -15,7 +15,8 @@ auto_notify_exc_info, logger, leave_breadcrumb, add_on_breadcrumb, remove_on_breadcrumb, add_feature_flag, add_feature_flags, - clear_feature_flag, clear_feature_flags) + clear_feature_flag, clear_feature_flags, + aws_lambda_handler) __all__ = ('Client', 'Event', 'Configuration', 'RequestConfiguration', 'configuration', 'configure', 'configure_request', @@ -25,4 +26,5 @@ 'BreadcrumbType', 'Breadcrumb', 'Breadcrumbs', 'OnBreadcrumbCallback', 'leave_breadcrumb', 'add_on_breadcrumb', 'remove_on_breadcrumb', 'FeatureFlag', 'add_feature_flag', - 'add_feature_flags', 'clear_feature_flag', 'clear_feature_flags') + 'add_feature_flags', 'clear_feature_flag', 'clear_feature_flags', + 'aws_lambda_handler') diff --git a/bugsnag/legacy.py b/bugsnag/legacy.py index cf9c6247..018e6735 100644 --- a/bugsnag/legacy.py +++ b/bugsnag/legacy.py @@ -1,4 +1,4 @@ -from typing import Dict, Any, Tuple, Type, Optional, Union, List +from typing import Dict, Any, Tuple, Type, Optional, Union, List, Callable import types import sys @@ -176,3 +176,10 @@ def clear_feature_flag(name: Union[str, bytes]) -> None: def clear_feature_flags() -> None: default_client.clear_feature_flags() + + +def aws_lambda_handler( + real_handler: Optional[Callable] = None, + flush_timeout_ms: int = 2000, +) -> Callable: + return default_client.aws_lambda_handler(real_handler, flush_timeout_ms)