test

lib/plugin/create_plugin.py

@@ -0,0 +1,87 @@
+import os, traceback
+from flask import Blueprint
+from framework import F
+from support.base.yaml import SupportYaml
+from . import get_model_setting, Logic, default_route, default_route_single_module
+
+class PluginBase(object):
+    package_name = None
+    logger = None
+    blueprint = None
+    menu = None
+    plugin_info = None
+    ModelSetting = None
+    logic = None
+    module_list = None
+    home_module = None
+
+    def __init__(self, setting):
+        try:
+            is_system = ('system' == os.path.basename(os.path.dirname(setting['filepath'])))
+            self.status = ""
+            self.setting = setting
+
+            info_filepath = os.path.join(os.path.dirname(setting['filepath']), 'info.yaml')
+            if os.path.exists(info_filepath) == False and is_system == False:
+                return
+            if is_system:
+                self.package_name = 'system'
+            else:
+                self.plugin_info = SupportYaml.read_yaml(info_filepath)
+                self.package_name = self.plugin_info['package_name']
+
+            self.logger = F.get_logger(self.package_name)
+            self.blueprint = Blueprint(self.package_name, self.package_name, url_prefix=f'/{self.package_name}', template_folder=os.path.join(os.path.dirname(setting['filepath']), 'templates'), static_folder=os.path.join(os.path.dirname(setting['filepath']), 'static'))
+            self.menu = setting['menu']
+            self.setting_menu = setting['setting_menu']
+
+            self.ModelSetting = None
+            if setting.get('use_db', True):
+                db_path = os.path.join(F.config['path_data'], 'db', f'{self.package_name}.db')
+                F.app.config['SQLALCHEMY_BINDS'][self.package_name] = f"sqlite:///{db_path}"
+            if setting.get('use_default_setting', True):
+                self.ModelSetting = get_model_setting(self.package_name, self.logger)
+            
+            self.module_list = []
+            self.home_module = setting.get('home_module')
+            self.status = "init_success"
+            self.config = {}
+        except Exception as e: 
+            self.logger.error(f'Exception:{str(e)}')
+            self.logger.error(traceback.format_exc())
+            self.status = 'init_fail'
+
+    def set_module_list(self, mod_list):
+        try:
+            for mod in mod_list:
+                mod_ins = mod(self)
+                self.module_list.append(mod_ins)
+            
+        except Exception as e:
+            F.logger.error(f'Exception:{str(e)}')
+            F.logger.error(traceback.format_exc())
+
+            
+        self.logic = Logic(self)
+        route_mode = self.setting.get('default_route', 'normal')
+        if route_mode == 'normal':
+            default_route(self)
+        elif route_mode == 'single':
+            default_route_single_module(self)
+
+
+    def plugin_load(self):
+        self.logic.plugin_load()
+    
+    def plugin_unload(self):
+        self.logic.plugin_unload()
+
+    def get_first_manual_path(self):
+        for __ in self.menu['list']:
+            if __['uri'] == 'manual' and len(__['list']) > 0:
+                return __['list'][0]['uri']
+
+
+def create_plugin_instance(config):
+    ins = PluginBase(config)
+    return ins