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

def validate_manifest(data, validator, error):
    from bootstrapvz.common.tools import rel_path
    validator(data, rel_path(__file__, 'manifest-schema.yml'))


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.CheckPublicKeyFile)
        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,
                    ])