uwsgi_

#!/usr/bin/env python
# -*- python -*-
#
# Define env.addr variable in your settings file, e.g.
#
#/etc/munin/plugin-conf.d/munin-node
#[uwsgi_*]
#env.addr 127.0.0.1:49000
#
# Magic markers
#%# capabilities=autoconf suggest
#%# family=auto

UWSGI_STATS = ''

import os
import os.path
import sys
import json
import socket
from pprint import pformat

UWSGI_STATS = os.environ.get('addr', '') or '127.0.0.1:49000'


if ':' in UWSGI_STATS:
    t = UWSGI_STATS.split(':')
    sfamily, addr = (socket.AF_INET, (t[0], int(t[1])) )
else:
    sfamily, addr = (socket.AF_UNIX, UWSGI_STATS)

NAME, P, MODE = os.path.basename(__file__).partition('_')

modes = {
    'listen_queue': ('Listen queue', 'listen queue slots used'),
    'listen_queue_errors': ('Listen queue errors', 'listen queue errors'),
    'requests': ('Requests', 'requests managed per ${graph_period}'),
    'exceptions': ('Exceptions', 'exceptions per ${graph_period}'),
    'vsz': ('Virtual memory', 'bytes'),
    'tx': ('Transmitted data', 'bytes per ${graph_period}'),
    'avg_rt': ('Average response time', 'time (ms)'),
    'rss': ('Resident Set Size', 'bytes'),
}

def get_uwsgi_stats():
    '''
    Get the json stats from the uwsgi stats socket and return a dict
    '''
    if ':' in UWSGI_STATS:
        host, port = UWSGI_STATS.split(':')
        sfamily, addr = (socket.AF_INET, (host, int(port)))
    else:
        sfamily, addr = (socket.AF_UNIX, UWSGI_STATS)

    raw_js = ''
    try:
        sock = socket.socket(sfamily, socket.SOCK_STREAM)
        sock.settimeout(3)
        sock.connect(addr)

        while True:
            data = sock.recv(4096)
            if len(data) < 1:
                break
            raw_js += data
        data = json.loads(raw_js or '{}')
        return data
    except:
        print("no (unable to get uWSGI stats at %s)" % UWSGI_STATS)

def print_stats():
    '''
    Calculate values and print them out for munin to graph
    '''
    data = get_uwsgi_stats()
    if MODE in ['listen_queue', 'listen_queue_errors']:
        value = data.get(MODE, 0)
        print '{0}.value {1}'.format(MODE, value)
    elif MODE in ['avg_rt']:
        value = sum([x.get(MODE, 0)/1000 for x in data.get('workers',[])]) / len(data.get('workers',[]))
        print '{0}.value {1}'.format(MODE, value)
    else:
        value = sum([x.get(MODE, 0) for x in data.get('workers',[])])
        print '{0}.value {1}'.format(MODE, value)

def print_config():
    '''
    Print the config for munin
    '''
    data = get_uwsgi_stats()
    workers = data.get('workers', [])
    graph = ('graph_title "{desc}"\n'
             'graph_vlabel "{vlabel}"\n'
             'graph_category uWSGI\n').format(desc=modes[MODE][0],
                                              vlabel=modes[MODE][1])
    out = ''
    if MODE in ['tx', 'vsz', 'rss']:
        graph += 'graph_args --base 1024\n'
    if MODE in ['requests', 'exceptions', 'tx']:
        out += '{name}.type DERIVE\n'
        out += '{name}.min 0\n'
    out += "{name}.label {name}"
    out = out.format(name=MODE)
    print graph + out

def main():
    '''
    Process args and do stuff.
    '''
    if len(sys.argv) > 1:
        opt = sys.argv[1]

        if opt == 'suggest':
            print '\n'.join(modes.keys())
        elif opt == 'autoconf':
            if get_data():
                print 'yes'
        elif opt == 'config' and MODE in modes:
            print_config()
    else:
        print_stats()


if __name__ == '__main__':
    main()