boto3 session issue layers & subs install (#287)

newrelic · Jun 7, 2024 · b43f6a1 · b43f6a1
1 parent 4062c63
commit b43f6a1
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 5 deletions.
diff --git a/newrelic_lambda_cli/cli/layers.py b/newrelic_lambda_cli/cli/layers.py
@@ -107,7 +107,11 @@ def register(group):
 def install(ctx, **kwargs):
     """Install New Relic AWS Lambda Layers"""
     input = LayerInstall(session=None, verbose=ctx.obj["VERBOSE"], **kwargs)
-
+    input = input._replace(
+        session=boto3.Session(
+            profile_name=input.aws_profile, region_name=input.aws_region
+        )
+    )
     if input.aws_permissions_check:
         permissions.ensure_layer_install_permissions(input)
 
@@ -181,7 +185,11 @@ def install(ctx, **kwargs):
 def uninstall(ctx, **kwargs):
     """Uninstall New Relic AWS Lambda Layers"""
     input = LayerUninstall(session=None, verbose=ctx.obj["VERBOSE"], **kwargs)
-
+    input = input._replace(
+        session=boto3.Session(
+            profile_name=input.aws_profile, region_name=input.aws_region
+        )
+    )
     if input.aws_permissions_check:
         permissions.ensure_layer_uninstall_permissions(input)
 

diff --git a/newrelic_lambda_cli/cli/subscriptions.py b/newrelic_lambda_cli/cli/subscriptions.py
@@ -64,7 +64,11 @@ def register(group):
 def install(**kwargs):
     """Install New Relic AWS Lambda Log Subscriptions"""
     input = SubscriptionInstall(session=None, **kwargs)
-
+    input = input._replace(
+        session=boto3.Session(
+            profile_name=input.aws_profile, region_name=input.aws_region
+        )
+    )
     if input.aws_permissions_check:
         permissions.ensure_subscription_install_permissions(input)
 
@@ -121,7 +125,11 @@ def install(**kwargs):
 def uninstall(**kwargs):
     """Uninstall New Relic AWS Lambda Log Subscriptions"""
     input = SubscriptionUninstall(session=None, **kwargs)
-
+    input = input._replace(
+        session=boto3.Session(
+            profile_name=input.aws_profile, region_name=input.aws_region
+        )
+    )
     if input.aws_permissions_check:
         permissions.ensure_subscription_uninstall_permissions(input)
 

diff --git a/setup.py b/setup.py
@@ -6,7 +6,7 @@
 
 setup(
     name="newrelic-lambda-cli",
-    version="0.9.1",
+    version="0.9.2",
     python_requires=">=3.3",
     description="A CLI to install the New Relic AWS Lambda integration and layers.",
     long_description=README,