Starting to improve the loging for Heroku (with logging.StreamHandler).

1 files changed, 11 insertions, 7 deletions

diff --git a/src/bootstrap.py b/src/bootstrap.py
index 92399e74..2b94b1b9 100644
--- a/src/bootstrap.py
+++ b/src/bootstrap.py
@@ -10,18 +10,22 @@ import flask_restless
 from urllib.parse import urlsplit
 
 
-def set_logging(log_path, log_level=logging.INFO,
+def set_logging(log_path=None, log_level=logging.INFO, modules=(),
                 log_format='%(asctime)s %(levelname)s %(message)s'):
-    formater = logging.Formatter(log_format)
-    if conf.ON_HEROKU:
-        handler = logging.StreamHandler()
-    else:
+    if not modules:
+        modules = ('root', 'bootstrap', 'runserver',
+                   'web', 'crawler.classic_crawler', 'manager', 'plugins')
+    if log_path:
         handler = logging.FileHandler(log_path)
+    else:
+        handler = logging.StreamHandler()
+    formater = logging.Formatter(log_format)
     handler.setFormatter(formater)
-    for logger_name in ('bootstrap', 'web', 'manager', 'runserver',
-                            'classic_crawler'):
+    for logger_name in modules:
         logger = logging.getLogger(logger_name)
         logger.addHandler(handler)
+        for handler in logger.handlers:
+            handler.setLevel(log_level)
         logger.setLevel(log_level)
 
 from flask import Flask