--- /dev/null
+#part-handler
+
+# Licensed under the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License. You may obtain
+# a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations
+# under the License.
+
+import datetime
+import errno
+import os
+
+
+def list_types():
+ return(["text/x-cfninitdata"])
+
+
+def handle_part(data, ctype, filename, payload):
+ if ctype == "__begin__":
+ try:
+ os.makedirs('/var/lib/heat-cfntools', 0o700)
+ except OSError as e:
+ if e.errno != errno.EEXIST:
+ raise
+ return
+
+ if ctype == "__end__":
+ return
+
+ with open('/var/log/part-handler.log', 'a') as log:
+ timestamp = datetime.datetime.now()
+ log.write('%s filename:%s, ctype:%s\n' % (timestamp, filename, ctype))
+
+ if ctype == 'text/x-cfninitdata':
+ with open('/var/lib/heat-cfntools/%s' % filename, 'w') as f:
+ f.write(payload)
+
+ # TODO(sdake) hopefully temporary until users move to heat-cfntools-1.3
+ with open('/var/lib/cloud/data/%s' % filename, 'w') as f:
+ f.write(payload)
import os
import pkgutil
from urlparse import urlparse
-import yaml
from oslo.config import cfg
filename=filename)
return msg
- def read_cfg_file():
- rd = pkgutil.get_data('heat', 'cloudinit/config')
- rd = rd.replace('@INSTANCE_USER@',
- cfg.CONF.instance_user)
- rd = yaml.safe_load(rd)
- return rd
-
- def add_data_file(fn, data):
- rd = [{'owner': 'root:root',
- 'path': '/var/lib/heat-cfntools/%s' % fn,
- 'permissions': '0700',
- 'content': '%s' % data}]
- return rd
-
- def yaml_add(yd, fn, data):
- rd = '%s%s' % (yd, yaml.safe_dump(add_data_file(fn, data)))
- return rd
-
- def read_data_file(fn):
+ def read_cloudinit_file(fn):
data = pkgutil.get_data('heat', 'cloudinit/%s' % fn)
- rd = add_data_file(fn, data)
- return rd
+ data = data.replace('@INSTANCE_USER@',
+ cfg.CONF.instance_user)
+ return data
- def read_cloudinit_file(fn):
- rd = pkgutil.get_data('heat', 'cloudinit/%s' % fn)
- rd = rd.replace('@INSTANCE_USER@', cfg.CONF.instance_user)
- return rd
+ attachments = [(read_cloudinit_file('config'), 'cloud-config'),
+ (read_cloudinit_file('boothook.sh'), 'boothook.sh',
+ 'cloud-boothook'),
+ (read_cloudinit_file('part_handler.py'),
+ 'part-handler.py'),
+ (userdata, 'cfn-userdata', 'x-cfninitdata'),
+ (read_cloudinit_file('loguserdata.py'),
+ 'loguserdata.py', 'x-shellscript')]
+
+ if 'Metadata' in self.t:
+ attachments.append((json.dumps(self.metadata),
+ 'cfn-init-data', 'x-cfninitdata'))
+
+ attachments.append((cfg.CONF.heat_watch_server_url,
+ 'cfn-watch-server', 'x-cfninitdata'))
+
+ attachments.append((cfg.CONF.heat_metadata_server_url,
+ 'cfn-metadata-server', 'x-cfninitdata'))
# Create a boto config which the cfntools on the host use to know
# where the cfn and cw API's are to be accessed
"cloudwatch_region_name = heat",
"cloudwatch_region_endpoint = %s" %
cw_url.hostname])
+ attachments.append((boto_cfg,
+ 'cfn-boto-cfg', 'x-cfninitdata'))
- # Build yaml write_files section
- yamld = ''
- yamld = yaml_add(yamld, 'cfn-watch-server',
- cfg.CONF.heat_watch_server_url)
- yamld = yaml_add(yamld, 'cfn-metadata-server',
- cfg.CONF.heat_metadata_server_url)
- yamld = yaml_add(yamld, 'cfn-boto-cfg',
- boto_cfg)
- yamld = yaml_add(yamld, 'cfn-userdata',
- userdata)
-
- if 'Metadata' in self.t:
- yamld = yaml_add(yamld, 'cfn-init-data',
- json.dumps(self.metadata))
-
- yamld = "%swrite_files:\n%s" % (yaml.safe_dump(read_cfg_file()),
- yamld)
-
- attachments = [(yamld, 'cloud-config'),
- (read_cloudinit_file('boothook.sh'), 'boothook.sh',
- 'cloud-boothook'),
- (read_cloudinit_file('loguserdata.py'),
- 'x-shellscript')]
subparts = [make_subpart(*args) for args in attachments]
mime_blob = MIMEMultipart(_subparts=subparts)
super(LoguserdataTest, self).setUp()
self.m.StubOutWithMock(pkg_resources, 'get_distribution')
self.m.StubOutWithMock(subprocess, 'Popen')
+ self.m.StubOutWithMock(os, 'chmod')
def test_ci_version(self):
# too old versions
pkg_resources.get_distribution('cloud-init').AndReturn(
FakeCiVersion('0.7.0'))
+ os.chmod('/var/lib/heat-cfntools/cfn-userdata', 0o700).AndReturn(None)
subprocess.Popen(
['/var/lib/heat-cfntools/cfn-userdata'],
stderr=mox.IgnoreArg(),
pkg_resources.get_distribution('cloud-init').AndReturn(
FakeCiVersion('0.7.0'))
+ os.chmod('/var/lib/heat-cfntools/cfn-userdata', 0o700).AndReturn(None)
subprocess.Popen(
['/var/lib/heat-cfntools/cfn-userdata'],
stderr=mox.IgnoreArg(),
pkg_resources.get_distribution('cloud-init').AndReturn(
FakeCiVersion('0.7.0'))
+ os.chmod('/var/lib/heat-cfntools/cfn-userdata', 0o700).AndReturn(None)
subprocess.Popen(
['/var/lib/heat-cfntools/cfn-userdata'],
stderr=mox.IgnoreArg(),
def test_main_error_other(self):
pkg_resources.get_distribution('cloud-init').AndReturn(
FakeCiVersion('0.7.0'))
+ os.chmod('/var/lib/heat-cfntools/cfn-userdata', 0o700).AndReturn(None)
subprocess.Popen(
['/var/lib/heat-cfntools/cfn-userdata'],
stderr=mox.IgnoreArg(),
pkg_resources.get_distribution('cloud-init').AndReturn(
FakeCiVersion('0.7.0'))
+ os.chmod('/var/lib/heat-cfntools/cfn-userdata', 0o700).AndReturn(None)
subprocess.Popen(
['/var/lib/heat-cfntools/cfn-userdata'],
stderr=mox.IgnoreArg(),