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 +++++++++++++++++++++++++++++++++ cloudinit/CloudConfig/cc_puppet.py | 26 +++++++++++--- 2 files changed, 85 insertions(+), 4 deletions(-) create mode 100644 cloudinit/CloudConfig/cc_mcollective.py (limited to 'cloudinit/CloudConfig') 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']) + diff --git a/cloudinit/CloudConfig/cc_puppet.py b/cloudinit/CloudConfig/cc_puppet.py index b498c5f0..c458365d 100644 --- a/cloudinit/CloudConfig/cc_puppet.py +++ b/cloudinit/CloudConfig/cc_puppet.py @@ -19,6 +19,8 @@ import os import pwd import socket import subprocess +import StringIO +import ConfigParser import cloudinit.CloudConfig as cc def handle(name,cfg,cloud,log,args): @@ -31,7 +33,13 @@ def handle(name,cfg,cloud,log,args): # ... and then update the puppet configuration if puppet_cfg.has_key('conf'): # Add all sections from the conf object to puppet.conf - puppet_conf_fh = open('/etc/puppet/puppet.conf', 'a') + puppet_conf_fh = open('/etc/puppet/puppet.conf', 'r') + # Create object for reading puppet.conf values + puppet_config = ConfigParser.ConfigParser() + # Read puppet.conf values from original file in order to be able to mix the rest up + puppet_config.readfp(StringIO.StringIO(''.join(i.lstrip() for i in puppet_conf_fh.readlines()))) + # Close original file, no longer needed + puppet_conf_fh.close() for cfg_name, cfg in puppet_cfg['conf'].iteritems(): # ca_cert configuration is a special case # Dump the puppetmaster ca certificate in the correct place @@ -51,7 +59,12 @@ def handle(name,cfg,cloud,log,args): os.chown('/var/lib/puppet/ssl/certs/ca.pem', pwd.getpwnam('puppet').pw_uid, 0) else: - puppet_conf_fh.write("\n[%s]\n" % (cfg_name)) + #puppet_conf_fh.write("\n[%s]\n" % (cfg_name)) + # If puppet.conf already has this section we don't want to write it again + if puppet_config.has_section(cfg_name) == False + puppet_config.add_section(cfg_name) + # Iterate throug the config items, we'll use ConfigParser.set + # to overwrite or create new items as needed for o, v in cfg.iteritems(): if o == 'certname': # Expand %f as the fqdn @@ -61,8 +74,13 @@ def handle(name,cfg,cloud,log,args): cloud.datasource.get_instance_id()) # certname needs to be downcase v = v.lower() - puppet_conf_fh.write("%s=%s\n" % (o, v)) - puppet_conf_fh.close() + puppet_config.set(cfg_name,o,v) + #puppet_conf_fh.write("%s=%s\n" % (o, v)) + # We got all our config as wanted we'll rename + # the previous puppet.conf and create our new one + os.rename('/etc/puppet/puppet.conf','/etc/puppet/puppet.conf.old') + with open('/etc/puppet/puppet.conf', 'wb') as configfile: + puppet_config.write(configfile) # Set puppet default file to automatically start subprocess.check_call(['sed', '-i', '-e', 's/^START=.*/START=yes/', -- cgit v1.2.3