Skip to content

Commit fc25fff

Browse files
mjrussoMatt George
authored andcommitted
added ability to specify log level via command-line option of pyres_worker
1 parent fdce35e commit fc25fff

File tree

1 file changed

+25
-1
lines changed

1 file changed

+25
-1
lines changed

scripts/pyres_worker

Lines changed: 25 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,41 @@
11
#!/usr/bin/env python
2+
3+
import logging
4+
5+
from optparse import OptionParser
26
from pyres.worker import Worker
7+
8+
LOG_LEVELS = {
9+
'debug': logging.DEBUG,
10+
'info': logging.INFO,
11+
'warning': logging.WARNING,
12+
'error': logging.ERROR,
13+
'critical': logging.CRITICAL
14+
}
15+
316
def main():
4-
from optparse import OptionParser
17+
518
usage = "usage: %prog [options] arg1"
619
parser = OptionParser(usage=usage)
720
#parser.add_option("-q", dest="queue_list")
821
parser.add_option("--host", dest="host", default="localhost")
922
parser.add_option("--port",dest="port",type="int", default=6379)
1023
parser.add_option("-i", '--interval', dest='interval', default=None, help='the default time interval to sleep between runs')
24+
parser.add_option('-l', '--log-level', dest='log_level', default=LOG_LEVELS['info'], help='log level. Valid values are "debug", "info", "warning", "error", "critical", in decreasing order of verbosity. Defaults to "info" if parameter not specified.')
1125
(options,args) = parser.parse_args()
26+
1227
if len(args) != 1:
1328
parser.print_help()
1429
parser.error("Argument must be a comma seperated list of queues")
30+
31+
if options.log_level not in LOG_LEVELS.keys():
32+
parser.print_help()
33+
parser.error("invalid log level specified")
34+
else:
35+
log_level = LOG_LEVELS[options.log_level]
36+
37+
logging.basicConfig(level=log_level, format="%(asctime)s: %(levelname)s: %(message)s")
38+
1539
queues = args[0].split(',')
1640
server = '%s:%s' % (options.host,options.port)
1741
Worker.run(queues, server, options.interval)

0 commit comments

Comments
 (0)