Add aws_lambda_handler legacy function delegator

bugsnag · Apr 4, 2024 · 4937d84 · 4937d84
1 parent 080152a
commit 4937d84
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git 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
@@ -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)