From 465bc9137e8c330070ba68ae777c1df33921c405 Mon Sep 17 00:00:00 2001
From: Zrief <103112786+Zrief@users.noreply.github.com>
Date: Sun, 9 Mar 2025 18:51:41 +0800
Subject: [PATCH] Update main_info_bar.py
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

简化代码长度
---
 app/core/main_info_bar.py | 51 ++++++++++++---------------------------
 1 file changed, 16 insertions(+), 35 deletions(-)

diff --git a/app/core/main_info_bar.py b/app/core/main_info_bar.py
index c83f1ca..aed883b 100644
--- a/app/core/main_info_bar.py
+++ b/app/core/main_info_bar.py
@@ -42,51 +42,32 @@ class _MainInfoBar:
 
     def push_info_bar(self, mode: str, title: str, content: str, time: int):
         """推送到信息通知栏"""
-
         if self.main_window is None:
             logger.error("信息通知栏未设置父窗口")
             return None
-
-        if mode == "success":
-            InfoBar.success(
+    
+        # 定义模式到 InfoBar 方法的映射
+        mode_mapping = {
+            "success": InfoBar.success,
+            "warning": InfoBar.warning,
+            "error": InfoBar.error,
+            "info": InfoBar.info
+        }
+    
+        # 根据 mode 获取对应的 InfoBar 方法
+        info_bar_method = mode_mapping.get(mode)
+        if info_bar_method:
+            info_bar_method(
                 title=title,
                 content=content,
                 orient=Qt.Horizontal,
                 isClosable=True,
                 position=InfoBarPosition.TOP_RIGHT,
                 duration=time,
-                parent=self.main_window,
-            )
-        elif mode == "warning":
-            InfoBar.warning(
-                title=title,
-                content=content,
-                orient=Qt.Horizontal,
-                isClosable=True,
-                position=InfoBarPosition.TOP_RIGHT,
-                duration=time,
-                parent=self.main_window,
-            )
-        elif mode == "error":
-            InfoBar.error(
-                title=title,
-                content=content,
-                orient=Qt.Horizontal,
-                isClosable=True,
-                position=InfoBarPosition.TOP_RIGHT,
-                duration=time,
-                parent=self.main_window,
-            )
-        elif mode == "info":
-            InfoBar.info(
-                title=title,
-                content=content,
-                orient=Qt.Horizontal,
-                isClosable=True,
-                position=InfoBarPosition.TOP_RIGHT,
-                duration=time,
-                parent=self.main_window,
+                parent=self.main_window
             )
+        else:
+            logger.error(f"未知的通知栏模式: {mode}")
 
 
 MainInfoBar = _MainInfoBar()