Mini Shell

# 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)