Add aws profile parameter

epsagon · May 21, 2019 · 78b90aa · 78b90aa
1 parent 7b30835
commit 78b90aa
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/list_lambdas.py b/list_lambdas.py
@@ -54,6 +54,9 @@ def init_boto_client(client_name, region, args):
             aws_secret_access_key=args.token_secret,
             region_name=region
         )
+    elif args.profile:
+        session = boto3.session.Session(profile_name=args.profile)
+        boto_client = session.client(client_name, region_name=region)
     else:
         boto_client = boto3.client(client_name, region_name=region)
 
@@ -278,6 +281,15 @@ def print_lambda_list(args):
         default='region',
         metavar='sort_by'
     )
+    parser.add_argument(
+        '--profile',
+        type=str,
+        help=(
+            'AWS profile. Optional '
+            '(default: "default" from local configuration).'
+        ),
+        metavar='profile'
+    )
 
     arguments = parser.parse_args()
     if arguments.sort_by not in SORT_KEYS: