Server : Apache System : Linux indy02.toastserver.com 3.10.0-962.3.2.lve1.5.85.el7.x86_64 #1 SMP Thu Apr 18 15:18:36 UTC 2024 x86_64 User : palandch ( 1163) PHP Version : 7.1.33 Disable Function : NONE Directory :  /opt/cloudlinux/venv/lib/python3.11/site-packages/lvestats/core/

Upload File : current_dir [ Writeable ] document_root [ Writeable ]

# coding=utf-8
#
# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2019 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENSE.TXT

from lvestats.lib import config as lvestats_config, dbengine

__author__ = 'shaman'


class PluginContext(object):
    def __init__(self, config):
        self.config = config
        self.db_engine = dbengine.make_db_engine(config, debug=False)
        self._cache = {}

    def get_instance(self, plugin_class):
        if plugin_class in self._cache:
            return self._cache[plugin_class]

        plugin = plugin_class()
        if hasattr(plugin, 'set_config'):
            plugin_config = lvestats_config.read_plugin_config(self.config, plugin=plugin_class.__name__)
            plugin.set_config(plugin_config)

        if hasattr(plugin, 'set_db_engine') and self.db_engine:
            plugin.set_db_engine(self.db_engine)

        self._cache[plugin_class] = plugin
        return plugin

    def recover_db(self):
        dbengine.recreate_db(self.db_engine)