@@ -450,7 +450,10 @@ def get_role_name(region, account_id, role):
450
450
return 'arn:{0}:iam::{1}:role/{2}' .format (prefix , account_id , role )
451
451
452
452
453
- def get_account_id (profile_name , aws_access_key_id , aws_secret_access_key , region = None ):
453
+ def get_account_id (
454
+ profile_name , aws_access_key_id , aws_secret_access_key ,
455
+ region = None ,
456
+ ):
454
457
"""Query STS for a users' account_id"""
455
458
client = get_client (
456
459
'sts' , profile_name , aws_access_key_id , aws_secret_access_key ,
@@ -459,7 +462,10 @@ def get_account_id(profile_name, aws_access_key_id, aws_secret_access_key, regio
459
462
return client .get_caller_identity ().get ('Account' )
460
463
461
464
462
- def get_client (client , profile_name , aws_access_key_id , aws_secret_access_key , region = None ):
465
+ def get_client (
466
+ client , profile_name , aws_access_key_id , aws_secret_access_key ,
467
+ region = None ,
468
+ ):
463
469
"""Shortcut for getting an initialized instance of the boto3 client."""
464
470
465
471
boto3 .setup_default_session (
@@ -481,7 +487,9 @@ def create_function(cfg, path_to_zip_file, use_s3=False, s3_file=None):
481
487
aws_secret_access_key = cfg .get ('aws_secret_access_key' )
482
488
483
489
account_id = get_account_id (
484
- profile_name , aws_access_key_id , aws_secret_access_key , cfg .get ('region' ),
490
+ profile_name , aws_access_key_id , aws_secret_access_key , cfg .get (
491
+ 'region' ,
492
+ ),
485
493
)
486
494
role = get_role_name (
487
495
cfg .get ('region' ), account_id ,
@@ -554,7 +562,9 @@ def update_function(cfg, path_to_zip_file, use_s3=False, s3_file=None):
554
562
aws_secret_access_key = cfg .get ('aws_secret_access_key' )
555
563
556
564
account_id = get_account_id (
557
- profile_name , aws_access_key_id , aws_secret_access_key , cfg .get ('region' ),
565
+ profile_name , aws_access_key_id , aws_secret_access_key , cfg .get (
566
+ 'region' ,
567
+ ),
558
568
)
559
569
role = get_role_name (
560
570
cfg .get ('region' ), account_id ,
0 commit comments