bootstrap-vz/bootstrapvz/plugins/admin_user/__init__.py

def validate_manifest(data, validator, error):
	import os.path
	schema_path = os.path.normpath(os.path.join(os.path.dirname(__file__), 'manifest-schema.yml'))
	validator(data, schema_path)
	pubkey = data['plugins']['admin_user'].get('pubkey', None)
	if pubkey is not None and not os.path.exists(pubkey):
		msg = 'Could not find public key at %s' % pubkey
		error(msg, ['plugins', 'admin_user', 'pubkey'])


def resolve_tasks(taskset, manifest):
	import logging
	import tasks
	from bootstrapvz.common.tasks import ssh

	from bootstrapvz.common.releases import jessie
	if manifest.release < jessie:
		taskset.update([ssh.DisableRootLogin])

	if 'password' in manifest.plugins['admin_user']:
		taskset.discard(ssh.DisableSSHPasswordAuthentication)
		taskset.add(tasks.AdminUserPassword)

	if 'pubkey' in manifest.plugins['admin_user']:
		taskset.add(tasks.AdminUserPublicKey)
	elif manifest.provider['name'] == 'ec2':
		logging.getLogger(__name__).info("The SSH key will be obtained from EC2")
		taskset.add(tasks.AdminUserPublicKeyEC2)
	elif 'password' not in manifest.plugins['admin_user']:
		logging.getLogger(__name__).warn("No SSH key and no password set")

	taskset.update([tasks.AddSudoPackage,
	                tasks.CreateAdminUser,
	                tasks.PasswordlessSudo,
	                ])