1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
import sys
import logging
import Bcfg2.Options
import Bcfg2.Server.Plugins
from django.db import models
from Bcfg2.Bcfg2Py3k import ConfigParser
logger = logging.getLogger('Bcfg2.Server.models')
MODELS = []
def load_models(plugins=None, cfile='/etc/bcfg2.conf', quiet=True):
global MODELS
if plugins is None:
# we want to provide a different default plugin list --
# namely, _all_ plugins, so that the database is guaranteed to
# work, even if /etc/bcfg2.conf isn't set up properly
plugin_opt = Bcfg2.Options.SERVER_PLUGINS
plugin_opt.default = Bcfg2.Server.Plugins.__all__
setup = Bcfg2.Options.OptionParser(dict(plugins=plugin_opt,
configfile=Bcfg2.Options.CFILE),
quiet=quiet)
setup.parse([Bcfg2.Options.CFILE.cmd, cfile])
plugins = setup['plugins']
if MODELS:
# load_models() has been called once, so first unload all of
# the models; otherwise we might call load_models() with no
# arguments, end up with _all_ models loaded, and then in a
# subsequent call only load a subset of models
for model in MODELS:
delattr(sys.modules[__name__], model)
MODELS = []
for plugin in plugins:
try:
mod = getattr(__import__("Bcfg2.Server.Plugins.%s" %
plugin).Server.Plugins, plugin)
except ImportError:
try:
mod = __import__(plugin)
except:
if plugins != Bcfg2.Server.Plugins.__all__:
# only produce errors if the default plugin list
# was not used -- i.e., if the config file was set
# up. don't produce errors when trying to load
# all plugins, IOW
err = sys.exc_info()[1]
logger.error("Failed to load plugin %s: %s" % (plugin, err))
continue
for sym in dir(mod):
obj = getattr(mod, sym)
if hasattr(obj, "__bases__") and models.Model in obj.__bases__:
print("Adding %s to models" % sym)
setattr(sys.modules[__name__], sym, obj)
MODELS.append(sym)
# basic invocation to ensure that a default set of models is loaded,
# and thus that this module will always work.
load_models(quiet=True)
|