Add importerror if gevent specified but not installed.

Refs #759
coleifer · Aug 4, 2023 · 2dedbb3 · 2dedbb3
1 parent a8f85f5
commit 2dedbb3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/huey/consumer.py b/huey/consumer.py
@@ -277,6 +277,8 @@ def __init__(self, huey, workers=1, periodic=True, initial_delay=0.1,
                                      'of 60, e.g. 1, 2, 3, 4, 5, 6, 10, 12...')
 
         if worker_type == 'gevent': worker_type = WORKER_GREENLET
+        if worker_type == WORKER_GREENLET and Greenlet is None:
+            raise ImportError('Could not import gevent - is it installed?')
         self.worker_type = worker_type  # What process model are we using?
 
         # Configure health-check and consumer main-loop attributes.