check_balancer.py

#!/usr/bin/env python
# encoding: utf-8
"""
check_balancer.py

Created by Wuvist on 2012-12-10.
Copyright (c) 2011 . All rights reserved.
"""

import urllib
import sys

def get_x_server(url):
    f = urllib.urlopen(url)
    f.close()
    return f.headers["X-Server"]

def check(url):
	servers = {}
	for i in range(0, 10):
		x_server = get_x_server(url)
		if servers.has_key(x_server):
			servers[x_server] += 1
		else:
			servers[x_server] = 1
	print servers

if __name__ == '__main__':
	check(sys.argv[1])