Fixbug/swan consoler (#423)

* fix: consoler init

* fix: failed log

* Update console.py

---------

Co-authored-by: KAAANG <[email protected]>

Author: Feudalman

swanlab/log/console.py

@@ -136,14 +136,23 @@ class Consoler(__consoler_class(), LeverCtl):
     # 只有第一次输入时为None
     __previous_message = None
 
-    def __init__(self):
+    __init_status = True
+
+    def __init__(self, *args, **kwargs):
         # 根据环境进行不同的初始化
-        if in_notebook():
-            super().__init__()
-        else:
-            super().__init__(sys.stdout.buffer)
+        try:
+            if in_notebook():
+                super().__init__()
+            else:
+                super().__init__(sys.stdout.buffer)
+        except:
+            self.__init_status = False
         self.original_stdout = sys.stdout  # 保存原始的 sys.stdout
 
+    @property
+    def init_status(self) -> bool:
+        return self.__init_status
+
     def init(self, path, swanlog_level="debug"):
         # 通过当前日期生成日志文件名
         self.now = datetime.now().strftime("%Y-%m-%d")
@@ -153,6 +162,10 @@ def init(self, path, swanlog_level="debug"):
             os.makedirs(path)
         # 日志文件路径
         console_path = os.path.join(path, f"{self.now}.log")
+        # 如果日志系统初始化失败
+        if not self.__init_status:
+            with open(console_path, "w", encoding="utf-8") as f:
+                f.write("Console Recoder Init Failed!")
         # 日志文件
         self.console = open(console_path, "a", encoding="utf-8")
 
@@ -238,7 +251,8 @@ def __init__(self):
 
     def init(self, path):
         self.consoler.init(path)
-        sys.stdout = self.consoler
+        if self.consoler.init_status:
+            sys.stdout = self.consoler
 
     def reset(self):
         """重置输出为原本的样子"""

vue/src/views/experiment/pages/log/LogPage.vue

@@ -122,6 +122,7 @@ const indexWidth = computed(() => {
   if (l) {
     index = lines.value[l - 1].index === '' ? lines.value[l - 2].index : lines.value[l - 1].index
   }
+  if (!index) return '0px'
   return 8 * index.length + 'px'
 })
 
@@ -222,7 +223,9 @@ const computeRange = (e) => {
  * 一般行高是 16px，基本不会出问题，所以目前只在初始化时计算，而没做另外的事件监听
  */
 const computeLineHeight = () => {
-  const line_styles = window.getComputedStyle(document.querySelector('.log-line'))
+  const e = document.querySelector('.log-line')
+  if (!e) return 0
+  const line_styles = window.getComputedStyle(e)
   lineHeight.value = parseFloat(line_styles.getPropertyValue('line-height'))
 }