remove_lingering_aws_host_tags.py

# This is a tool for removing AWS host-level tags from your infrastructure in
# datadog. It is intended for users who have removed their EC2 instances from
# their AWS integration and if they no longer want to see AWS tags associated
# with the hosts that still run datadog-agents.


# 3rd Party
import json
import requests


# Input Variables
API_KEY = 'YOUR_API_KEY_HERE'
APPLICATION_KEY = 'YOUR_APPLICATION_KEY_HERE'
TAG_SOURCE = 'Amazon Web Services'  # Can also be set to 'Users', 'Chef', or
                                    # or other source values.
REMOVE_FROM_ALL_HOSTS = False  # change to true to remove from all hosts
                               # be careful not to remove tags you want to keep!
HOSTS = [  # if not all, specify what hosts you want to delete tags from
    'host_name'
]


# functions
def remove_host_tags_by_source(hostname, source, session):
    remove_endpt = 'https://app.datadoghq.com/api/v1/tags/hosts/' + hostname
    session.params['source'] = source
    res = session.request(
        method='DELETE', url=remove_endpt, params=session.params
    )
    print 'removed %s tags from %s with response: %s' % (TAG_SOURCE, hostname, res)


# main
s = requests.session()

s.params = {
    'api_key': API_KEY,
    'application_key': APPLICATION_KEY
}

if REMOVE_FROM_ALL_HOSTS is True:
    perma_link = 'https://app.datadoghq.com/reports/v2/overview'
    perma_content = s.request(
        method='GET', url=perma_link, params=s.params
    ).text
    perma_json = json.loads(perma_content)
    
    for host in perma_json['rows']:
        host_name = host['host_name']
        if TAG_SOURCE in host['tags_by_source']:
            remove_host_tags_by_source(host_name, TAG_SOURCE, s)
        else:
            print "no %s tags in %s, skipping." % (TAG_SOURCE, host_name)
        
else:
    for host in HOSTS:
        remove_host_tags_by_source(host, TAG_SOURCE, s)