setup_logging.py is now more Jupyter-friendly.

zakv · zakv · commit b7119231cf1b · 2021-11-10T12:15:34.000-05:00
diff --git a/labscript_utils/setup_logging.py b/labscript_utils/setup_logging.py
@@ -11,6 +11,7 @@
 #                                                                   #
 #####################################################################
 import sys, os
+from io import UnsupportedOperation
 import logging, logging.handlers
 from labscript_utils.ls_zprocess import Handler, ensure_connected_to_zlog
 
@@ -52,21 +53,26 @@ def setup_logging(program_name, log_level=logging.DEBUG, terminal_level=logging.
     handler.setFormatter(formatter)
     handler.setLevel(log_level)
     logger.addHandler(handler)
-    if sys.stdout is not None and sys.stdout.fileno() >= 0:
-        stdout_handler = logging.StreamHandler(sys.stdout)
-        stdout_handler.setFormatter(formatter)
-        stdout_handler.setLevel(terminal_level)
-        logger.addHandler(stdout_handler)
-        if sys.stderr is not None and sys.stderr.fileno() >= 0:
-            # Send warnings and greater to stderr instead of stdout:
-            stdout_handler.addFilter(LessThanFilter(logging.WARNING))
-            sterr_handler = logging.StreamHandler(sys.stderr)
-            sterr_handler.setFormatter(formatter)
-            sterr_handler.setLevel(logging.WARNING)
-            logger.addHandler(sterr_handler)
-    else:
-        # Prevent bug on windows where writing to stdout without a command
-        # window causes a crash:
-        sys.stdout = sys.stderr = open(os.devnull, 'w')
+    try:
+        if sys.stdout is not None and sys.stdout.fileno() >= 0:
+            stdout_handler = logging.StreamHandler(sys.stdout)
+            stdout_handler.setFormatter(formatter)
+            stdout_handler.setLevel(terminal_level)
+            logger.addHandler(stdout_handler)
+            if sys.stderr is not None and sys.stderr.fileno() >= 0:
+                # Send warnings and greater to stderr instead of stdout:
+                stdout_handler.addFilter(LessThanFilter(logging.WARNING))
+                sterr_handler = logging.StreamHandler(sys.stderr)
+                sterr_handler.setFormatter(formatter)
+                sterr_handler.setLevel(logging.WARNING)
+                logger.addHandler(sterr_handler)
+        else:
+            # Prevent bug on windows where writing to stdout without a command
+            # window causes a crash:
+            sys.stdout = sys.stderr = open(os.devnull, 'w')
+    except UnsupportedOperation:
+        # Special handling for Jupyter notebook kernels where sys.stdout.fileno is not
+        # callable.
+        pass
     logger.setLevel(logging.DEBUG)
     return logger
