You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
73 lines
2.4 KiB
73 lines
2.4 KiB
#!/usr/bin/env python
|
|
|
|
import logging
|
|
import json
|
|
import os
|
|
import signal
|
|
|
|
import docopt
|
|
|
|
from qw.client import Client
|
|
from qw import __version__
|
|
|
|
|
|
__doc__ = """Usage:
|
|
qw-client [--host=<host>] [--port=<port>] [--db=<db>] managers
|
|
qw-client [--host=<host>] [--port=<port>] [--db=<db>] workers [<manager>]
|
|
qw-client [--host=<host>] [--port=<port>] [--db=<db>] jobs [<manager>]
|
|
qw-client [--host=<host>] [--port=<port>] [--db=<db>] queue string <data> [<manager>]
|
|
qw-client [--host=<host>] [--port=<port>] [--db=<db>] queue file <file> [<manager>]
|
|
qw-client (--help | --version)
|
|
|
|
Options:
|
|
--help Show this help message
|
|
--version Show version information
|
|
-h --host=<host> Set the redis host to use [default: localhost]
|
|
-p --port=<port> Set the redis port to use [default: 6379]
|
|
-d --db=<db> Set the redis db number to use [default: 0]
|
|
"""
|
|
|
|
arguments = docopt.docopt(__doc__, version="qw-manager %s" % (__version__, ))
|
|
client = Client(host=arguments["--host"], port=arguments["--port"], db=arguments["--db"])
|
|
|
|
if arguments["managers"]:
|
|
print "All Managers:"
|
|
for manager in client.get_all_managers():
|
|
print "\t%s" % (manager, )
|
|
|
|
elif arguments["workers"]:
|
|
managers = []
|
|
if arguments["<manager>"]:
|
|
managers = [arguments["<manager>"]]
|
|
else:
|
|
managers = client.get_all_managers()
|
|
for manager in managers:
|
|
print "%s Workers:" % (manager, )
|
|
for worker in client.get_manager_workers(manager):
|
|
print "\t%s" % (worker, )
|
|
|
|
elif arguments["jobs"]:
|
|
managers = []
|
|
if arguments["<manager>"]:
|
|
managers = [arguments["<manager>"]]
|
|
else:
|
|
managers = client.get_all_managers()
|
|
|
|
for manager in managers:
|
|
print "%s:" % (manager, )
|
|
print "\tQueued Jobs:"
|
|
for job_id in client.get_manager_queued_jobs(manager):
|
|
print "\t %s" % (job_id, )
|
|
print "\tPending Jobs:"
|
|
for worker in client.get_manager_workers(manager):
|
|
for job_id in client.get_worker_pending_jobs(worker):
|
|
print "\t %s - %s" % (worker, job_id)
|
|
|
|
elif arguments["queue"]:
|
|
data = arguments["<data>"]
|
|
if arguments["file"]:
|
|
with open(arguments["<file>"]) as fp:
|
|
data = fp.read()
|
|
data = json.loads(data)
|
|
job_id = client.queue_job(data, manager=arguments["<manager>"])
|
|
print "Queued Job: %s" % (job_id, )
|