registers worker again after a redis flush

Author: guilhermef

pyres/worker.py

@@ -51,6 +51,9 @@ def register_worker(self):
         #self.resq._redis.add("worker:#{self}:started", Time.now.to_s)
         self.started = datetime.datetime.now()
 
+    def is_registered(self):
+        return self.resq.redis.sismember('resque:workers', str(self))
+
     def _set_started(self, dt):
         if dt:
             key = int(time.mktime(dt.timetuple()))
@@ -138,6 +141,9 @@ def work(self, interval=5):
                 logger.info('shutdown scheduled')
                 break
 
+            if not self.is_registered():
+                self.register_worker()
+
             job = self.reserve(interval)
 
             if job:

tests/test_worker.py

@@ -38,6 +38,11 @@ def test_unregister(self):
         assert self.redis.sismember('resque:workers',name)
         worker.unregister_worker()
         assert name not in self.redis.smembers('resque:workers')
+
+    def test_worker_is_registered(self):
+        worker = Worker(['basic'])
+        worker.register_worker()
+        assert worker.is_registered()
 
     def test_working_on(self):
         name = "%s:%s:%s" % (os.uname()[1],os.getpid(),'basic')