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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
""" Django database models for all plugins """
import sys
import copy
import logging
import Bcfg2.Options
import Bcfg2.Server.Plugins
from Bcfg2.Compat import walk_packages
from django.db import models
LOGGER = logging.getLogger('Bcfg2.Server.models')
MODELS = []
def load_models(plugins=None, cfile='/etc/bcfg2.conf', quiet=True):
""" load models from plugins specified in the config """
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 = copy.deepcopy(Bcfg2.Options.SERVER_PLUGINS)
all_plugins = []
for submodule in walk_packages(path=Bcfg2.Server.Plugins.__path__,
prefix="Bcfg2.Server.Plugins."):
module = submodule[1].rsplit('.', 1)[-1]
if submodule[1] == "Bcfg2.Server.Plugins.%s" % module:
# we only include direct children of
# Bcfg2.Server.Plugins -- e.g., all_plugins should
# include Bcfg2.Server.Plugins.Cfg, but not
# Bcfg2.Server.Plugins.Cfg.CfgInfoXML
all_plugins.append(module)
plugin_opt.default = all_plugins
setup = Bcfg2.Options.get_option_parser()
setup.add_option("plugins", plugin_opt)
setup.add_option("configfile", Bcfg2.Options.CFILE)
setup.reparse(argv=[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:
err = sys.exc_info()[1]
mod = __import__(plugin)
except: # pylint: disable=W0702
if plugins != plugin_opt.default:
# 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. the error from the first
# attempt to import is probably more accurate than
# the second attempt.
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__:
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)
class InternalDatabaseVersion(models.Model):
""" Object that tell us to which version the database is """
version = models.IntegerField()
updated = models.DateTimeField(auto_now_add=True)
def __str__(self):
return "version %d updated the %s" % (self.version,
self.updated.isoformat())
class Meta: # pylint: disable=C0111,W0232
app_label = "reports"
get_latest_by = "version"
|