fix: Dont crash during chalice deployment

Author: untitaker

sentry_sdk/integrations/aws_lambda.py

@@ -7,18 +7,27 @@
     AnnotatedValue,
     capture_internal_exceptions,
     event_from_exception,
+    logger,
 )
 from sentry_sdk.integrations import Integration
 from sentry_sdk.integrations._wsgi import _filter_headers
 
-import __main__ as lambda_bootstrap
-import runtime as lambda_runtime
-
 
 class AwsLambdaIntegration(Integration):
     identifier = "aws_lambda"
 
     def install(self):
+        import __main__ as lambda_bootstrap
+
+        if not hasattr(lambda_bootstrap, "make_final_handler"):
+            logger.warn(
+                "Not running in AWS Lambda environment, "
+                "AwsLambdaIntegration disabled"
+            )
+            return
+
+        import runtime as lambda_runtime
+
         old_make_final_handler = lambda_bootstrap.make_final_handler
 
         def sentry_make_final_handler(*args, **kwargs):

tests/integrations/aws_lambda/test_aws.py

@@ -64,10 +64,17 @@ def inner(code, payload):
         tmpdir.ensure_dir("lambda_tmp").remove()
         tmp = tmpdir.ensure_dir("lambda_tmp")
 
-        # https://docs.aws.amazon.com/lambda/latest/dg/lambda-python-how-to-create-deployment-package.html
         tmp.join("test_lambda.py").write(code)
+
+        # Check file for valid syntax first, and that the integration does not
+        # crash when not running in Lambda (but rather a local deployment tool
+        # such as chalice's)
+        subprocess.check_call([sys.executable, str(tmp.join("test_lambda.py"))])
+
         tmp.join("setup.cfg").write("[install]\nprefix=")
         subprocess.check_call([sys.executable, "setup.py", "sdist", "-d", str(tmpdir)])
+
+        # https://docs.aws.amazon.com/lambda/latest/dg/lambda-python-how-to-create-deployment-package.html
         subprocess.check_call("pip install ../*.tar.gz -t .", cwd=str(tmp), shell=True)
         shutil.make_archive(tmpdir.join("ball"), "zip", str(tmp))