c3mc-ec2create-describe-role-list

#!/usr/bin/env /data/Software/mydan/python3/bin/python3
# -*- coding: utf-8 -*-

import sys
import json
import subprocess

import boto3


def list_roles(access_id, access_key, region):
    """
        查询角色列表
    """
    client = boto3.client(
        "iam",
        aws_access_key_id=access_id,
        aws_secret_access_key=access_key,
        region_name=region
    )

    page_size = 1000

    result = []
    response = client.list_roles(MaxItems=page_size)
    result.extend(response["Roles"])
    marker = response["Marker"] if "Marker" in response else ""
    while marker != "":
        response = client.list_roles(MaxItems=page_size, Marker=marker)
        result.extend(response["Roles"])

        marker = response["Marker"] if "Marker" in response else ""
    return result


def main(params):
    account = params["account"]
    region = params["region"]

    key_info = json.loads(
        subprocess.getoutput(f"c3mc-get-account-info aws {account}")
    )

    role_list = list_roles(
        key_info["access_id"], key_info["access_key"], region)
    for role in role_list:
        print(json.dumps(role, default=str))


if __name__ == '__main__':
    l = list(sys.stdin)
    if not l or len(l) > 1:
        raise type('WrongInputData', (Exception,), {})('数据格式不对, 需要一行json字符串"')

    main(json.loads(l[0]))