优化

core/plugin/plugin_registry.py

@@ -0,0 +1,127 @@
+import os
+import sys
+import time
+import logging
+import importlib
+import threading
+
+
+from pathlib import Path
+from enum import Enum, auto
+from PySide6.QtCore import QObject,Signal
+from dataclasses import dataclass, field, fields
+
+LOCAL_PLUGINS_PATH  = Path("./plugins")
+REMOTE_PLUGINS_PATH = Path('Y:/SE/xufeifei/plugins')
+
+logger = logging.getLogger(__name__)
+
+
+class Event(Enum):
+    Update = auto()
+    Install = auto()
+    Uninstall = auto()
+
+class PluginRegistry(QObject):
+    """插件注册中心。"""
+    plugins_loader_signal = Signal(dict)
+    update_plugins_card_signal = Signal(dict)
+    def __init__(self, parent = None) -> None:
+        super().__init__(parent)
+
+        self.plugins = dict()
+
+        self.start_discover_local(LOCAL_PLUGINS_PATH, self.plugins)
+        self.start_discover_remote(REMOTE_PLUGINS_PATH, self.plugins)
+
+    def start_discover_local(self,plugins_dir: Path, plugins:dict):
+        self.thread_local = threading.Thread(target= self.discover_local_event,args=(plugins_dir, plugins,))
+        self.thread_local.start()
+
+    def discover_local_event(self, plugins_dir: Path, plugins:dict):
+        if not os.path.exists(plugins_dir):
+            os.mkdir(plugins_dir)
+            return
+        
+        sys.path.append(str(plugins_dir))
+        for tool in Path(plugins_dir).glob("*.pyd"):
+            tool_name = tool.stem
+            try:
+                module = importlib.import_module(tool_name)
+                obj = module.create_plugin()
+                name = module.read_plugin_name()
+                version = module.read_plugin_version()
+                if name in plugins.keys():
+                    pass
+                else:
+                    plugins[name] = {}
+                plugins[name]['obj'] = obj
+                plugins[name]['local version'] = version
+            except Exception as e:
+                logger.error(f"插件{tool_name}注册失败，{str(e)}")
+        sys.path.remove(str(LOCAL_PLUGINS_PATH))
+        self.plugins_loader_signal.emit(plugins)
+
+
+    def start_discover_remote(self,plugins_dir: Path, plugins:dict):
+        self.thread_remote = threading.Thread(target= self.discover_remote_event,args=(plugins_dir, plugins,))
+        self.thread_remote.start()
+
+    def discover_remote_event(self, plugins_dir: Path, plugins:dict):
+        if not os.path.exists(plugins_dir):
+            logger.error("服务器链接错误！")
+            return
+        #读取服务器工具信息
+        time.sleep(0.01)
+        sys.path.append(str(plugins_dir))
+        for tool in Path(plugins_dir).glob("*.pyd"):
+            tool_name = tool.stem
+            try:
+                module = importlib.import_module(tool_name)
+
+                name = module.read_plugin_name()
+                version = module.read_plugin_version()
+                description = module.read_plugin_description()
+                if name in plugins.keys():
+                    pass
+                else:
+                    print("discover_remote_event")
+                    plugins[name] = {}
+                    plugins[name]["tool_name"] = tool_name
+                    plugins[name]["local version"] = None
+                    plugins[name]["local description"] = None
+                plugins[name]["remote version"] = version
+                plugins[name]["remote description"] = description
+                plugins[name]['remote path'] = plugins_dir
+            except Exception as e:
+                logger.error(f"服务器{tool_name}无法加载：{str(e)}")
+        sys.path.remove(str(plugins_dir))
+        self.update_plugins_card_signal.emit(plugins)
+
+    def start_plugins_event(self, event:Event, name:str):
+        if event == Event.Install:
+            local_path = str(LOCAL_PLUGINS_PATH) + '\\' + f"{self.plugins[name]['tool_name']}.pyd" 
+            remote_path = str(REMOTE_PLUGINS_PATH) + '\\' + f"{self.plugins[name]['tool_name']}.pyd" 
+
+            self.sub_thread = threading.Thread(target= self.install_plugins_event,args=(local_path,remote_path,))
+            self.sub_thread.start()
+        pass
+
+    def install_plugins_event(self, local_path, remote_path):
+        try:
+            total_size = os.path.getsize(remote_path)
+            copied_size = 0
+
+            with open(remote_path, 'rb') as fsrc, open(local_path, 'wb') as fdst:
+                while True:
+                    buf = fsrc.read(1024*1024)
+                    if not buf:
+                        break
+                    fdst.write(buf)
+                    copied_size += len(buf)
+                    logger.info(f"安装进度: {copied_size/1024/1024:.2f} MB / {total_size/1024/1024:.2f} MB")
+                    time.sleep(0.1)
+                logger.info(f"{local_path}安装成功!!!")
+        except Exception as e:
+            logger.error(f"{remote_path}安装失败{e.args}")
+            return