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