Warn if thread/socket depended celery modules loaded before eventlet patched

ask · ask · commit d87c21cc7936 · 2012-08-05T15:43:11.000+01:00
diff --git a/celery/concurrency/eventlet.py b/celery/concurrency/eventlet.py
@@ -10,8 +10,20 @@
 
 import os
 
-EVENTLET_NOPATCH = int(os.environ.get('EVENTLET_NOPATCH', 0))
+EVENTLET_NOPATCH = os.environ.get('EVENTLET_NOPATCH', False)
 EVENTLET_DBLOCK = int(os.environ.get('EVENTLET_NOBLOCK', 0))
+W_RACE = """\
+Celery module with %s imported before eventlet patched\
+"""
+
+def _racedetect():
+    import sys
+    for mod in (mod for mod in sys.modules if mod.startswith('celery.')):
+            for side in ('thread', 'threading', 'socket'):
+                if getattr(mod, side, None):
+                    warnings.warn(RuntimeWarning(W_RACE % side))
+_racedetect()
+
 
 PATCHED = [0]
 if not EVENTLET_NOPATCH and not PATCHED[0]:
