django_check_current_users.py

#!/usr/bin/python

import argparse
import json
import sys
import urllib2

CRITICAL = 5
WARNING = 2

parser = argparse.ArgumentParser(description='Checks logged in users on remote host.')

parser.add_argument('url', help='URL of remote endpoint to check.')

args = parser.parse_args()

response = urllib2.urlopen(args.url)
data = json.load(response)   

if 'count' in data:
    if data['count'] > CRITICAL:
        print 'USERS CRITICAL: ' + str(data['count'])
        sys.exit(2)
    elif data['count'] > WARNING:
        print 'USERS WARNING: ' + str(data['count'])
        sys.exit(1)
    else:
        print 'USERS OK: ' + str(data['count'])
        sys.exit(0)
else:
    print 'USERS UNKNOWN: ' + args.url
    sys.exit(3)