53 lines
1.5 KiB
Python
Executable File
53 lines
1.5 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import sys, re
|
|
from beanstalk import serverconn, protohandler
|
|
|
|
# Temporary workaround until my patch is merged.
|
|
if not protohandler._namematch.match('ABZDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-+/;.$_()'):
|
|
protohandler._namematch = re.compile(r'^[a-zA-Z0-9+\(\)/;.$_][a-zA-Z0-9+\(\)/;.$_-]{0,199}$')
|
|
|
|
STATES = ['ready', 'reserved', 'urgent', 'delayed', 'buried']
|
|
|
|
def connect(host='localhost', port=11300):
|
|
return serverconn.ServerConn(host, port)
|
|
|
|
def config():
|
|
c = connect()
|
|
tubes = c.list_tubes()['data']
|
|
print_config(tubes)
|
|
|
|
def print_config(tubes, graph_title='Beanstalkd jobs', graph_vlabel='count'):
|
|
for tube in tubes:
|
|
print 'multigraph job_count_' + tube
|
|
print 'graph_title %s (%s)' % (graph_title, tube,)
|
|
print 'graph_order ' + ' '.join(STATES)
|
|
print 'graph_vlabel ' + graph_vlabel
|
|
for state in STATES:
|
|
print '%s.label %s' % (state, state,)
|
|
print
|
|
|
|
def run():
|
|
c = connect()
|
|
tubes = c.list_tubes()['data']
|
|
print_values(tubes, c)
|
|
|
|
def print_values(tubes, c):
|
|
for tube in tubes:
|
|
print 'multigraph job_count_' + tube
|
|
stats = c.stats_tube(tube)['data']
|
|
for state in STATES:
|
|
key = 'current-jobs-' + state
|
|
value = stats[key]
|
|
print '%s.value %d' % (state, value,)
|
|
print
|
|
|
|
def main():
|
|
if len(sys.argv) > 1 and sys.argv[1] == "config":
|
|
config()
|
|
else:
|
|
run()
|
|
|
|
if __name__ == "__main__":
|
|
main()
|