modify: 封装webview通用模块

app/view/web_view/entrance.py

@@ -7,59 +7,17 @@ import base64
 import tempfile
 from pathlib import Path
 
-from PySide6.QtCore import QObject, Slot, QUrl
-from PySide6.QtWebChannel import QWebChannel
+from PySide6.QtCore import Slot, QUrl
 from PySide6.QtWebEngineCore import QWebEngineSettings
-from qframelesswindow.webengine import FramelessWebEngineView
-from queue_sqlite.model import MessageItem
 
-from ...scheduler_manager import scheduler
-from ...signal import global_signals
+from ...utils.base_webview_bridge import BaseWebViewBridge
 
 
-class WebViewQueueSqliteOperations(QObject):
-    """
-    用于前端调用QueueSqlite的队列操作
-    """
-
-    def __init__(self, webview: FramelessWebEngineView, parent=None):
-        super().__init__(parent)
-        self.webview = webview
-        self.page = self.webview.page()
-        global_signals.update_queue_state.connect(self.update_queue_state)
-
-    @Slot(str, str, result=str)
-    def send_message(self, message: str, store_key: str):
-        def callback(result: MessageItem):
-            print(f"send_message callback: {result}")
-            global_signals.update_queue_state.emit(result.to_dict(), store_key)
-
-        scheduler.send_message(MessageItem.from_dict(json.loads(message)), callback)
-
-    def update_queue_state(self, result: dict, store_key: str):
-        json_str = json.dumps(result)
-        print(f"result: {result}")
-        print(f"json_str: {json_str}")
-        js_code = f"""
-        try {{
-            if (typeof window.queueStore.updateQueueData === 'function') {{
-                window.queueStore.updateQueueData('{store_key}', {json_str});
-            }} else {{
-                console.log('updateQueueData function not found');
-            }}
-        }} catch (error) {{
-            console.error('Error in updateQueueData:', error);
-        }}
-        """
-        print(f"执行的JavaScript代码: {js_code}")
-        self.page.runJavaScript(js_code)
-
-
-class WebviewBridge(QObject):
+class Entrance(BaseWebViewBridge):
     """Webview桥接类，用于处理前端和Python后端之间的通信"""
 
     def __init__(self, webview, parent=None):
-        super().__init__(parent)
+        super().__init__(webview, parent)
         self.webview = webview
 
         # 启用本地文件访问
@@ -362,17 +320,3 @@ class WebviewBridge(QObject):
         except Exception as e:
             # 确保任何异常都返回有效的JSON
             return json.dumps({"success": False, "error": str(e)})
-
-
-class WebviewController(QObject):
-    """Webview控制器类，负责管理Webview的逻辑"""
-
-    def __init__(self, web_engine_page, web_webview, parent=None):
-        super().__init__(parent)
-        self.page = web_engine_page
-        self.bridge = WebviewBridge(web_webview)
-        self.queue_sqlite_operations = WebViewQueueSqliteOperations(web_webview)
-        self.channel = QWebChannel()
-        self.channel.registerObject("entrance", self.bridge)
-        self.channel.registerObject("queueSqlite", self.queue_sqlite_operations)
-        self.page.setWebChannel(self.channel)

app/view/web_view/web_view_interface.py

@@ -15,8 +15,10 @@ from PySide6.QtWidgets import (
 from PySide6.QtCore import QPoint, Qt, QUrl
 from qframelesswindow.webengine import FramelessWebEngineView
 
-from .entrance import WebviewController
 from ...config import config
+from ...utils.webview_controller import WebviewController
+from .entrance import Entrance
+from ...scheduler_manager import scheduler
 
 
 class DevToolsWindow(FramelessWebEngineView):
@@ -60,7 +62,9 @@ class WebViewInterface(QFrame):
 
         # 创建WebviewController实例
         self.page = self.webView.page()
-        self.controller = WebviewController(self.page, self.webView)
+        self.controller = WebviewController(
+            self.page, [Entrance], self.webView, scheduler
+        )
 
         # 加载初始页面
         self.load_current_page()