From 2492df4fbdd54d9c8b3e172db53ee78c1709de84 Mon Sep 17 00:00:00 2001 From: Marc Cluet Date: Tue, 15 Feb 2011 14:50:40 +0000 Subject: Changes to puppet module + mcollective module --- cloudinit/CloudConfig/cc_mcollective.py | 63 +++++++++++++++++++++++++++++++++ 1 file changed, 63 insertions(+) create mode 100644 cloudinit/CloudConfig/cc_mcollective.py (limited to 'cloudinit/CloudConfig/cc_mcollective.py') diff --git a/cloudinit/CloudConfig/cc_mcollective.py b/cloudinit/CloudConfig/cc_mcollective.py new file mode 100644 index 00000000..ce86a2aa --- /dev/null +++ b/cloudinit/CloudConfig/cc_mcollective.py @@ -0,0 +1,63 @@ +# vi: ts=4 expandtab +# +# Copyright (C) 2009-2011 Canonical Ltd. +# +# Author: Marc Cluet +# Based on code by Scott Moser +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License version 3, as +# published by the Free Software Foundation. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see . +import os +import pwd +import socket +import subprocess +import StringIO +import ConfigParser +import cloudinit.CloudConfig as cc + +# Our fake header section +class FakeSecHead(object): + def __init__(self, fp): + self.fp = fp + self.sechead = '[nullsection]\n' + def readline(self): + if self.sechead: + try: return self.sechead + finally: self.sechead = None + else: return self.fp.readline() + +def handle(name,cfg,cloud,log,args): + # If there isn't a mcollective key in the configuration don't do anything + if not cfg.has_key('mcollective'): return + mcollective_cfg = cfg['mcollective'] + # Start by installing the mcollective package ... + cc.install_packages(("mcollective",)) + + # ... and then update the mcollective configuration + if mcollective_cfg.has_key('conf'): + # Create object for reading server.cfg values + mcollective_config = ConfigParser.ConfigParser() + # Read server.cfg values from original file in order to be able to mix the rest up + mcollective_config.readfp(FakeSecHead(open('/etc/mcollective/server.cfg'))) + for cfg_name, cfg in mcollective_cfg['conf'].iteritems(): + # Iterate throug the config items, we'll use ConfigParser.set + # to overwrite or create new items as needed + for o, v in cfg.iteritems(): + mcollective_config.set(cfg_name,o,v) + # We got all our config as wanted we'll rename + # the previous server.cfg and create our new one + os.rename('/etc/mcollective/server.cfg','/etc/mcollective/server.cfg.old') + with open('/etc/mcollective/server.cfg', 'wb') as configfile: + mcollective_config.write(configfile) + # Start mcollective + subprocess.check_call(['service', 'mcollective', 'start']) + -- cgit v1.2.3