From 891be55b874cba8b827874095d66561e126e97e3 Mon Sep 17 00:00:00 2001
From: Vincent Driessen <me@nvie.com>
Date: Wed, 3 Jun 2015 10:05:28 +0200
Subject: [PATCH] Clean up Worker.__init__ logic a little

---
 rq/utils.py  | 16 +++++++++++++++-
 rq/worker.py | 23 +++++++++--------------
 2 files changed, 24 insertions(+), 15 deletions(-)

diff --git a/rq/utils.py b/rq/utils.py
index 3e44a98..f2e1897 100644
--- a/rq/utils.py
+++ b/rq/utils.py
@@ -13,8 +13,9 @@ import datetime
 import importlib
 import logging
 import sys
+from collections import Iterable
 
-from .compat import as_text, is_python_version
+from .compat import as_text, is_python_version, string_types
 
 
 class _Colorizer(object):
@@ -205,6 +206,19 @@ def first(iterable, default=None, key=None):
     return default
 
 
+def is_nonstring_iterable(obj):
+    """Returns whether the obj is an iterable, but not a string"""
+    return isinstance(obj, Iterable) and not isinstance(obj, string_types)
+
+
+def ensure_list(obj):
+    """
+    When passed an iterable of objects, does nothing, otherwise, it returns
+    a list with just that object in it.
+    """
+    return obj if is_nonstring_iterable(obj) else [obj]
+
+
 def current_timestamp():
     """Returns current UTC timestamp"""
     return calendar.timegm(datetime.datetime.utcnow().utctimetuple())
diff --git a/rq/worker.py b/rq/worker.py
index ee048c3..dc9ce00 100644
--- a/rq/worker.py
+++ b/rq/worker.py
@@ -2,7 +2,6 @@
 from __future__ import (absolute_import, division, print_function,
                         unicode_literals)
 
-from datetime import timedelta
 import errno
 import logging
 import os
@@ -13,6 +12,7 @@ import sys
 import time
 import traceback
 import warnings
+from datetime import timedelta
 
 from rq.compat import as_text, string_types, text_type
 
@@ -20,11 +20,12 @@ from .connections import get_current_connection
 from .exceptions import DequeueTimeout
 from .job import Job, JobStatus
 from .logutils import setup_loghandlers
-from .queue import get_failed_queue, Queue
-from .registry import clean_registries, FinishedJobRegistry, StartedJobRegistry
+from .queue import Queue, get_failed_queue
+from .registry import FinishedJobRegistry, StartedJobRegistry, clean_registries
 from .suspension import is_suspended
 from .timeouts import UnixSignalDeathPenalty
-from .utils import enum, import_attribute, make_colorizer, utcformat, utcnow, utcparse
+from .utils import (ensure_list, enum, import_attribute, make_colorizer,
+                    utcformat, utcnow, utcparse)
 from .version import VERSION
 
 try:
@@ -126,7 +127,9 @@ class Worker(object):
         if connection is None:
             connection = get_current_connection()
         self.connection = connection
-        queues = self.process_queue_args(queues)
+
+        queues = [self.queue_class(name=q) if isinstance(q, text_type) else q
+                  for q in ensure_list(queues)]
         self._name = name
         self.queues = queues
         self.validate_queues()
@@ -163,15 +166,7 @@ class Worker(object):
         """Sanity check for the given queues."""
         for queue in self.queues:
             if not isinstance(queue, self.queue_class):
-                raise TypeError('{0} is not a Queue or a string'.format(queue))
-
-    def process_queue_args(self, queue_args):
-        """Allow for a string, a queue an iterable of strings or an iterable of queues"""
-        if isinstance(queue_args, text_type):
-            return self.queue_class(name=queue_args)
-        else:
-            return [self.queue_class(name=queue_arg) if isinstance(queue_arg, text_type) else queue_arg
-                    for queue_arg in queue_args]
+                raise TypeError('{0} is not of type {1} or text type'.format(queue, self.queue_class))
 
     def queue_names(self):
         """Returns the queue names of this worker's queues."""