#!/usr/bin/env python
# -*- coding:utf-8 -*-

import json
import urllib2
import sys

class rabbitMp_api():
    '''
    初始化变量
    '''
    def __init__(self,host = '127.0.0.1',port = '15672',user = 'guest',protocol='http'):
        self.host = host
        self.port = port
        self.user = user
        self.protocol = protocol
    '''
    获取api信息
    '''
    def post_api(self,path):
        url = '{0}://{1}:{2}/api/{3}'.format(self.protocol, self.host, self.port,path)
        password_mgr = urllib2.HTTPPasswordMgrWithDefaultRealm()
        password_mgr.add_password(None, url, self.user, self.user)
        handler = urllib2.HTTPBasicAuthHandler(password_mgr)
        return json.loads(urllib2.build_opener(handler).open(url).read())
    def list_queues(self):
        queues = []
        quedict = {}
        for queue in self.post_api('queues'):
            queues.append({"{#VHOST_NAME}":queue['vhost']+queue['name']})
        quedict["data"] = queues
        return json.dumps(quedict)
    def check_overview(self,item = sys.argv[1]):
        if item in [ 'connections', 'consumers', 'queues']:
            return int(self.post_api('overview').get('object_totals').get(item, 0))
        if item in ['publish']:
            return int(self.post_api('overview').get('message_stats').get(item,0))

a = rabbitMp_api()
if __name__ == "__main__":
        try:
                item = sys.argv[1]
        except:
                print(sys.argv[0]+" "+ "publish|consumers|queues|connections")
        else:
                if item == "publish" :
                        print a.check_overview('publish')
                elif item  == "consumers":
                        print a.check_overview('consumers')
                elif item  == "queues":
                        print a.check_overview('queues')
                elif item  == "connections":
                        print a.check_overview('connections')
                elif item  == "discover":
                        print a.list_queues()