python – 在运行时更改蓝图或重新加载烧瓶应用程序

2023年1月9日 168次阅读

我正在编写一个支持插件架构的Flask应用程序.每个插件都位于一个单独的文件夹中,并且是一个至少有一个类,它是Plugin类的子类.出于安全原因,我不想在最初运行flask应用程序时加载所有插件.相反,用户可以从烧瓶应用程序中启用插件.一旦他这样做,我们会在数据库中存储一份备忘录,该备忘录将应用程序列入白名单以进行加载.但是,我们仍然需要记住哪些插件被禁用并证明了这些插件的视图.我通过为未启用但未加载任何自定义代码的插件创建虚拟类来实现此目的.

每个插件都有自己的蓝图.我们在加载插件时注册.蓝图还定义了启用插件的路径.
整件事看起来像这样：

for plugin_name in os.listdir(plugin_dir):
    plugin_path = os.path.join(plugin_paths, plugin_name)
    module_name = "plugins.{}.__init__".format(plugin_name)
    plugin_enabled = ask_db_whether_plugin_is_enabled(plugin_name)

    if os.path.isdir(plugin_path) and plugin_enabled:
        module = __import__(module_name)
        for plugin in load_plugins_from_module(module):
            app.register_blueprint(plugin.blueprint, url_prefix='/plugins')
    else:
        PluginCls = type(identifier, (Plugin, ), {})
        disabled_plugin = PluginCls()
        app.register_blueprint(disabled_plugin.blueprint, url_prefix='/plugins')

load_plugins_from_module看起来像这样：

def load_plugins_from_module(module):
    def is_plugin(c):
        return inspect.isclass(c) and \
               issubclass(c, Plugin) and \
               c != Plugin

    for name, objects in inspect.getmembers(module, lambda c: inspect.ismodule(c)):
        for name, PluginCls in inspect.getmembers(objects, is_plugin):
            plugin = PluginCls()
            yield plugin

现在问题如下：当我将插件更改为启用时,我基本上想重新运行

module = __import__(module_name)
for plugin in load_plugins_from_module(module):
    app.register_blueprint(plugin.blueprint, url_prefix='/plugins')

对于该插件的模块,以便它变为活动状态并注册已在子类插件中定义的所有路由.这会引发AssertionError,因为我无法在运行时更改蓝图.什么是好的解决方案呢？我可以从应用程序中重新加载应用程序吗？我可以在运行时修改现有的蓝图吗？

谢谢你的帮助！

最佳答案 我不确定你是否需要让这个变得复杂.

您只需为要启用的插件设置配置选项即可.您可以在“Start_app()”方法中根据该配置注册您的蓝图.

您还可以动态设置要从某些文件夹/文件继承的配置选项,以使其更具动态性.

插件通常由开发人员使用,因此配置选项并不繁琐,除非您尝试构建一些每个随机用户都可以修改您的网站的东西 – 这可能会带来巨大的安全问题.

For security reasons, I don’t want to load all the plugins when the
flask app is initially run.

我不确定这一点.不允许用户手动启动插件,更具安全风险(因此,如果用户秘密地能够上传代码,那么现在他可以启用它们).

你可以创建一个像WordPress一样的插件启用CMS,只需不要路由插件的网址,直到用户点击“激活插件”

点赞