From 4556f39c34b9f6dd3ad9d9f7d6c27f25cca30ad6 Mon Sep 17 00:00:00 2001
From: liangsijian <liangsijian@foxmail.com>
Date: Sun, 28 May 2017 23:31:51 +0800
Subject: [PATCH] use enum WorkerStatus instead of str

---
 rq/worker.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/rq/worker.py b/rq/worker.py
index f87bc69..c63c7ba 100644
--- a/rq/worker.py
+++ b/rq/worker.py
@@ -394,7 +394,7 @@ class Worker(object):
 
         # If shutdown is requested in the middle of a job, wait until
         # finish before shutting down and save the request in redis
-        if self.get_state() == 'busy':
+        if self.get_state() == WorkerStatus.BUSY:
             self._stop_requested = True
             self.set_shutdown_requested_date()
             self.log.debug('Stopping after current horse is finished. '
@@ -583,10 +583,10 @@ class Worker(object):
         within the given timeout bounds, or will end the work horse with
         SIGALRM.
         """
-        self.set_state('busy')
+        self.set_state(WorkerStatus.BUSY)
         self.fork_work_horse(job, queue)
         self.monitor_work_horse(job)
-        self.set_state('idle')
+        self.set_state(WorkerStatus.IDLE)
 
     def main_work_horse(self, job, queue):
         """This is the entry point of the newly spawned work horse."""