mongo_ops

#!/usr/bin/env python

## GENERATED FILE - DO NOT EDIT

import urllib2
import sys
import os

try:
    import json
except ImportError:
    import simplejson as json

def getServerStatus():
    host = os.environ.get("host", "127.0.0.1")
    port = 28017
    raw = urllib2.urlopen( "http://%s:%d/_status" % (host, port) ).read()
    return json.loads( raw )["serverStatus"]

def doAutoConf():
    host = os.environ.get("host", "127.0.0.1")
    port = 28017
    try:
        raw = urllib2.urlopen( "http://%s:%d/_status" % (host, port) ).read()
        print "yes"
        return True
    except urllib2.URLError:
        print "no"
        return False

def doData():
    ss = getServerStatus()
    for k,v in ss["opcounters"].iteritems():
        print( str(k) + ".value " + str(v) )

def doConfig():
    print "graph_title MongoDB ops"
    print "graph_args --base 1000 -l 0"
    print "graph_vlabel ops / ${graph_period}"
    print "graph_category MongoDB"
    print "graph_total total"

    for k in getServerStatus()["opcounters"]:
        print k + ".label " + k
        print k + ".min 0"
        print k + ".type COUNTER"
        print k + ".max 500000"
        print k + ".draw LINE1"

if __name__ == "__main__":
    if len(sys.argv) > 1 and sys.argv[1] == "autoconf":
        doAutoConf()
    elif len(sys.argv) > 1 and sys.argv[1] == "config":
        doConfig()
    else:
        doData()