name : plugin_context.py # 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)