6cb35e9b0f
After user data handling splits apart all the different content types into there various mime messages it is nice to be able to have each message specify how it should be merged (mainly for cloud-config or cloud-archive) into the single cloud config that is eventually used. This starts to add a plugable merging framework and the needed components to activate said headers and merging.
74 lines
2.6 KiB
Python
74 lines
2.6 KiB
Python
# vi: ts=4 expandtab
|
|
#
|
|
# Copyright (C) 2012 Canonical Ltd.
|
|
# Copyright (C) 2012 Hewlett-Packard Development Company, L.P.
|
|
# Copyright (C) 2012 Yahoo! Inc.
|
|
#
|
|
# Author: Scott Moser <scott.moser@canonical.com>
|
|
# Author: Juerg Haefliger <juerg.haefliger@hp.com>
|
|
# Author: Joshua Harlow <harlowja@yahoo-inc.com>
|
|
#
|
|
# 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 <http://www.gnu.org/licenses/>.
|
|
|
|
import os
|
|
|
|
from cloudinit import handlers
|
|
from cloudinit import log as logging
|
|
from cloudinit import util
|
|
|
|
from cloudinit.settings import (PER_ALWAYS)
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
class BootHookPartHandler(handlers.Handler):
|
|
def __init__(self, paths, datasource, **_kwargs):
|
|
handlers.Handler.__init__(self, PER_ALWAYS)
|
|
self.boothook_dir = paths.get_ipath("boothooks")
|
|
self.instance_id = None
|
|
if datasource:
|
|
self.instance_id = datasource.get_instance_id()
|
|
|
|
def list_types(self):
|
|
return [
|
|
handlers.type_from_starts_with("#cloud-boothook"),
|
|
]
|
|
|
|
def _write_part(self, payload, filename):
|
|
filename = util.clean_filename(filename)
|
|
payload = util.dos2unix(payload)
|
|
prefix = "#cloud-boothook"
|
|
start = 0
|
|
if payload.startswith(prefix):
|
|
start = len(prefix) + 1
|
|
filepath = os.path.join(self.boothook_dir, filename)
|
|
contents = payload[start:]
|
|
util.write_file(filepath, contents, 0700)
|
|
return filepath
|
|
|
|
def handle_part(self, _data, ctype, filename, payload, _frequency):
|
|
if ctype in handlers.CONTENT_SIGNALS:
|
|
return
|
|
|
|
filepath = self._write_part(payload, filename)
|
|
try:
|
|
env = os.environ.copy()
|
|
if self.instance_id is not None:
|
|
env['INSTANCE_ID'] = str(self.instance_id)
|
|
util.subp([filepath], env=env)
|
|
except util.ProcessExecutionError:
|
|
util.logexc(LOG, "Boothooks script %s execution error", filepath)
|
|
except Exception:
|
|
util.logexc(LOG, ("Boothooks unknown "
|
|
"error when running %s"), filepath)
|