use setproctitle() in pyres_scheduler

Author: joeshaw

pyres/scheduler.py

@@ -2,7 +2,7 @@
 import time
 import logging
 
-from pyres import ResQ
+from pyres import ResQ, __version__
 
 logger = logging.getLogger(__name__)
 
@@ -32,6 +32,7 @@ def schedule_shutdown(self, signal, frame):
         self._shutdown = True
 
     def __call__(self):
+        _setproctitle("Starting")
         logger.info('starting up')
         self.register_signal_handlers()
         #self.load_schedule()
@@ -40,6 +41,7 @@ def __call__(self):
             if self._shutdown:
                 break
             self.handle_delayed_items()
+            _setproctitle("Waiting")
             logger.debug('sleeping')
             time.sleep(5)
         logger.info('shutting down complete')
@@ -63,6 +65,7 @@ def next_item(self, timestamp):
 
     def handle_delayed_items(self):
         for timestamp in self.next_timestamp():
+            _setproctitle('Handling timestamp %s' % timestamp)
             logger.info('handling timestamp: %s' % timestamp)
             for item in self.next_item(timestamp):
                 logger.debug('queueing item %s' % item)
@@ -81,3 +84,11 @@ def run(cls, server, password=None):
         sched()
 
 
+try:
+    from setproctitle import setproctitle
+except ImportError:
+    def setproctitle(name):
+        pass
+
+def _setproctitle(msg):
+    setproctitle("pyres_scheduler-%s: %s" % (__version__, msg))