|
@@ -68,8 +68,13 @@ class Worker(object):
|
|
self.include = self.include.split(",")
|
|
self.include = self.include.split(",")
|
|
|
|
|
|
if not isinstance(self.loglevel, int):
|
|
if not isinstance(self.loglevel, int):
|
|
- self.loglevel = LOG_LEVELS[self.loglevel.upper()]
|
|
|
|
-
|
|
|
|
|
|
+ try:
|
|
|
|
+ self.loglevel = LOG_LEVELS[self.loglevel.upper()]
|
|
|
|
+ except KeyError:
|
|
|
|
+ self.die("Unknown level %r. Please use one of %s." % (
|
|
|
|
+ self.loglevel,
|
|
|
|
+ "|".join(l for l in LOG_LEVELS.keys()
|
|
|
|
+ if isinstance(l, basestring))))
|
|
def run(self):
|
|
def run(self):
|
|
self.init_loader()
|
|
self.init_loader()
|
|
self.init_queues()
|
|
self.init_queues()
|
|
@@ -211,6 +216,10 @@ class Worker(object):
|
|
info=info,
|
|
info=info,
|
|
hostname=self.hostname)
|
|
hostname=self.hostname)
|
|
|
|
|
|
|
|
+ def die(self, msg, exitcode=1):
|
|
|
|
+ sys.stderr.write("Error: %s\n" % (msg, ))
|
|
|
|
+ sys.exit(exitcode)
|
|
|
|
+
|
|
|
|
|
|
def install_worker_int_handler(worker):
|
|
def install_worker_int_handler(worker):
|
|
|
|
|