bootstrap-vz/bootstrapvz/plugins/cloud_init/__init__.py
2018-02-25 08:37:24 +00:00

43 lines
1.4 KiB
Python

from bootstrapvz.common.tools import rel_path
assets = rel_path(__file__, 'assets')
def validate_manifest(data, validator, error):
validator(data, rel_path(__file__, 'manifest-schema.yml'))
def resolve_tasks(taskset, manifest):
from . import tasks
import bootstrapvz.providers.ec2.tasks.initd as initd_ec2
from bootstrapvz.common.tasks import apt
from bootstrapvz.common.tasks import initd
from bootstrapvz.common.tasks import ssh
from bootstrapvz.common.releases import wheezy
from bootstrapvz.common.releases import jessie
if manifest.release == wheezy:
taskset.add(apt.AddBackports)
if manifest.release >= jessie:
taskset.add(tasks.SetCloudInitMountOptions)
taskset.update([tasks.SetMetadataSource,
tasks.AddCloudInitPackages,
])
options = manifest.plugins['cloud_init']
if 'username' in options:
taskset.add(tasks.SetUsername)
if 'groups' in options and options['groups']:
taskset.add(tasks.SetGroups)
if 'enable_modules' in options:
taskset.add(tasks.EnableModules)
if 'disable_modules' in options:
taskset.add(tasks.DisableModules)
taskset.discard(initd_ec2.AddEC2InitScripts)
taskset.discard(initd.AddExpandRoot)
taskset.discard(initd.AdjustExpandRootScript)
taskset.discard(initd.AdjustGrowpartWorkaround)
taskset.discard(ssh.AddSSHKeyGeneration)