Преглед на файлове

Only show multiprocessing logs if envvar MP_LOG is set

Ask Solem преди 13 години
родител
ревизия
9cfd49c44a
променени са 1 файла, в които са добавени 4 реда и са изтрити 1 реда
  1. 4 1
      celery/utils/log.py

+ 4 - 1
celery/utils/log.py

@@ -2,6 +2,7 @@
 from __future__ import absolute_import
 
 import logging
+import os
 import sys
 import threading
 import traceback
@@ -15,6 +16,8 @@ from .term import colored
 _process_aware = False
 is_py3k = sys.version_info[0] == 3
 
+MP_LOG = os.environ.get("MP_LOG", False)
+
 
 # Sets up our logging hierarchy.
 #
@@ -196,7 +199,7 @@ def ensure_process_aware_logger():
 
 
 def get_multiprocessing_logger():
-    return None; #mputil.get_logger() if mputil else None
+    return mputil.get_logger() if mputil and MP_LOG else None
 
 
 def reset_multiprocessing_logger():