def main()

in python/vpce/modify_vpce_service_permissions_sample.py [0:0]

• 14 lines of code
• 1 McCabe index (conditional complexity)

def main():
    """
    Adds/Removes AWS Principal to a VPC Endpoint Service (VPC-E Service)

    Accepts: service_id, principal_arns, add_remove
    --service_id: VPC-E Service id
    --principal_arns: iam_user1_arn
    --remove: remove arns

    Usage:
    python modify_vpce_service_permissions_sample.py \
    --service_id vpce-svc-xxxx \
    --principal_arns iam-user1-arn \
    --add_remove add

    To remove:
    python modify_vpce_service_permissions_sample.py \
    --service_id vpce-svc-xxxx \
    --principal_arns iam-user1-arn \
    --remove
    """
    parser = argparse.ArgumentParser()
    parser.add_argument('--service_id', required=True,
                        help='specify service id', type=str)
    parser.add_argument('--principal_arns', nargs='+', required=True,
                        help='specify iam arns')
    parser.add_argument('--add_remove', required=True,
                        help='add or remove principals', type=str)

    args = parser.parse_args()

    ############################
    # Define script variables:
    ############################
    service_id = args.service_id
    principal_arns = args.principal_arns
    add_remove = args.add_remove
    #############################

    # VPC-E Service:
    service1 = modify_vpce_service_permissions(service_id, principal_arns,
                                               add_remove)