Fixbug/jupyter (#401)

* Update console.py

* fix: buffer

* refactor: in notebook

* doc: explain

Author: Feudalman

swanlab/log/console.py

@@ -105,7 +105,30 @@ def getLevel(self):
         return self.__level
 
 
-class Consoler(sys.stdout.__class__, LeverCtl):
+# 检测是否在 notebook 环境中
+def in_notebook():
+    try:
+        # notebook 中会有 __IPYTHON__，而正常环境没有定义，所以 try
+        # 'type: ignore': 可以让 pylance 忽略对变量定义的检查
+        __IPYTHON__  # type: ignore
+        return True
+    except NameError:
+        return False
+
+
+# Consoler 继承的父类
+def __consoler_class():
+    # 如果在 notebook 中，使用 io.StringIO
+    if in_notebook():
+        from io import StringIO
+
+        return StringIO
+    # 正常环境使用标准输出
+    else:
+        return sys.stdout.__class__
+
+
+class Consoler(__consoler_class(), LeverCtl):
     # 记录日志行数
     __sum = 0
 
@@ -114,7 +137,11 @@ class Consoler(sys.stdout.__class__, LeverCtl):
     __previous_message = None
 
     def __init__(self):
-        super().__init__(sys.stdout.buffer)
+        # 根据环境进行不同的初始化
+        if in_notebook():
+            super().__init__()
+        else:
+            super().__init__(sys.stdout.buffer)
         self.original_stdout = sys.stdout  # 保存原始的 sys.stdout
 
     def init(self, path, swanlog_level="debug"):