diff options
Diffstat (limited to 'src/lib/Server/Plugins/Decisions.py')
-rw-r--r-- | src/lib/Server/Plugins/Decisions.py | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/src/lib/Server/Plugins/Decisions.py b/src/lib/Server/Plugins/Decisions.py new file mode 100644 index 000000000..4b758cdd4 --- /dev/null +++ b/src/lib/Server/Plugins/Decisions.py @@ -0,0 +1,51 @@ +import Bcfg2.Server.Plugin, lxml.etree + +class DecisionFile(Bcfg2.Server.Plugin.SpecificData): + def handle_event(self, event): + Bcfg2.Server.Plugin.SpecificData.handle_event(self, event) + self.contents = lxml.etree.XML(self.data) + + def get_decisions(self): + return [(x.get('type'), x.get('name')) for x in self.contents.xpath('.//Decision')] + +class DecisionSet(Bcfg2.Server.Plugin.EntrySet): + def __init__(self, path, fam, encoding): + """Container for decision specification files + + Arguments: + - `path`: repository path + - `fam`: reference to the file monitor + - `encoding`: XML character encoding + """ + pattern = '(white|black)list' + Bcfg2.Server.Plugin.EntrySet.__init__(self, pattern, path, False, \ + DecisionFile, encoding) + fam.AddMonitor(path, self) + + def HandleEvent(self, event): + if event.filename != self.path: + return self.handle_event(event) + + def GetDecisions(self, metadata, mode): + ret = [] + candidates = [c for c in self.get_matching(metadata) + if c.name.split('/')[-1].startswith(mode)] + for c in candidates: + ret += c.get_decisions() + return ret + +class Decisions(DecisionSet, Bcfg2.Server.Plugin.DecisionPlugin): + __name__ = 'Decisions' + __version__ = '$Id: $' + __author__ = 'bcfg-dev@mcs.anl.gov' + + def __init__(self, core, datastore): + """Decisions plugins + + Arguments: + - `core`: Bcfg2.Core instance + - `datastore`: File repository location + """ + Bcfg2.Server.Plugin.DecisionPlugin.__init__(self, core, datastore) + DecisionSet.__init__(self, self.data, core.fam, core.encoding) + |