def run_agent():
def print_help():
print(usage_msg)
sys.exit(7)
usage_msg = """
{prog} agent [OPTIONS] COMMANDS
Examples:
{prog} agent version
{prog} agent metric-list
{prog} agent metric-get <metric key>
""".format(prog=sys.argv[0])
parser = optparse.OptionParser(
usage=usage_msg,
version='%prog agent {0}'.format(__version__))
parser.add_option(
'-c', '--config', dest='config',
help=optparse.SUPPRESS_HELP)
args, commands = parser.parse_args()
cfg = Config(args.config)
host = cfg.fetch('agent', 'host')
port = cfg.fetch('agent', 'port', int)
url, key = None, None
if len(commands) == 0:
return print_help()
if commands[0] == 'version':
if len(commands) >= 2:
return print_help()
url = 'http://{0}:{1}/version'.format(host, port)
elif commands[0] == 'metric-list':
if len(commands) >= 2:
return print_help()
url = 'http://{0}:{1}/list'.format(host, port)
elif commands[0] == 'metric-get':
if not len(commands) == 2:
return print_help()
key = commands[1]
url = 'http://{0}:{1}/get?key={2}'.format(host, port, key)
else:
return print_help()
request = urllib2.Request(url)
try:
response = urllib2.urlopen(request)
except Exception as e:
sys.stderr.write('Open url: {0}, error: {1}\n'.format(
url, e))
sys.exit(9)
if not response.code == 200:
sys.stderr.write('Bad response from url {0}, code: {1}\n'.format(
url,
response.code))
sys.exit(8)
else:
try:
body = response.read()
except Exception as e:
sys.stderr.write('Read url: {0}, error: {1}\n'.format(
url, e))
sys.exit(9)
if body == API.UNKNOWN_VERSION:
sys.stderr.write('Unknown API version\n')
sys.exit(9)
elif body == API.METRIC_NOT_FOUND and key is not None:
sys.stderr.write('Metric \'{0}\' not found\n'.format(key))
sys.exit(9)
elif body == API.METRIC_NOT_FOUND and key is None:
sys.stderr.write('Unknown API version\n')
sys.exit(9)
else:
print(body)