Cleanup RabbitMQ Virtual Hosts

clean_vhosts.py

#!/usr/bin/env python

import httplib
import base64
import json

auth = base64.encodestring('{0}:{1}'.format('guest','guest'))
headers = {'Content-type':'application/json','Authorization':'Basic {0}'.format(auth)}

def remove_vhost(vhost):
   	print('Deleting vhost: {0}'.format(vhost))
	conn = httplib.HTTPConnection('localhost',15672)
	conn.request('DELETE', '/api/vhosts/{0}'.format(vhost), headers=headers)

def get_connected_vhosts():
	data = get_rest_data('/api/connections')
        return [rmq_conn['vhost'] for rmq_conn in data]

def get_vhosts_with_messages():
	data = get_rest_data('/api/queues')
	return [q['vhost'] for q in data if q['messages'] > 0]

def get_all_vhosts():
	return get_rest_data('/api/vhosts')

def get_rest_data(url):
	conn = httplib.HTTPConnection('localhost',15672)
	conn.request('GET',url,headers=headers)
	data = json.loads(conn.getresponse().read())
	return data

vhosts = get_all_vhosts()
connected_vhosts = get_connected_vhosts()
messaged_vhosts = get_vhosts_with_messages()
vhosts_to_save = connected_vhosts + messaged_vhosts
vhosts_to_save.append('/')
for vhost in vhosts:
	vhost_name = vhost['name']
	if vhost_name not in vhosts_to_save:
		remove_vhost(vhost_name)