DatadogからEC2に紐づくセキュリティグループを取り出して、Excelで開く事を前提としたCSVを作ります

host-securitygroup-csv-from-datadog.py

# -*- coding: utf-8 -*-
#
import re
import io
from datadog import initialize, api
import csv

options = {
    'api_key': 'hogehoge',
    'app_key': 'foobar'
}

header = ['name', 'hostname', 'security group']
initialize(**options)

hosts = api.Infrastructure.search(q='hosts:')

with io.open('sample.csv', 'w', encoding='utf-16') as f:
    writer = csv.writer(f)
    writer.writerow(header)

    for i in range(len(hosts['results']['hosts'])):
        body = []
        tags = api.Tag.get(hosts['results']['hosts'][i])
        matching = [ s for s in tags['tags'] if "name:" in s]
        body.append(matching[0])
        body.append(hosts['results']['hosts'][i])
        for n in tags['tags']:
            if n.startswith('security-group:'):
                body.append(n)
        writer.writerows([body])