mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-24 15:36:27 +00:00
69 lines
2.3 KiB
Python
69 lines
2.3 KiB
Python
from bootstrapvz.common import task_groups
|
|
import tasks.apt
|
|
import tasks.boot
|
|
import tasks.configuration
|
|
import tasks.image
|
|
import tasks.initd
|
|
import tasks.host
|
|
import tasks.packages
|
|
from bootstrapvz.common.tasks import apt
|
|
from bootstrapvz.common.tasks import boot
|
|
from bootstrapvz.common.tasks import loopback
|
|
from bootstrapvz.common.tasks import initd
|
|
from bootstrapvz.common.tasks import kernel
|
|
from bootstrapvz.common.tasks import ssh
|
|
from bootstrapvz.common.tasks import volume
|
|
|
|
|
|
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)
|
|
|
|
|
|
def resolve_tasks(taskset, manifest):
|
|
taskset.update(task_groups.get_standard_groups(manifest))
|
|
|
|
taskset.update([apt.AddBackports,
|
|
loopback.AddRequiredCommands,
|
|
loopback.Create,
|
|
tasks.apt.SetPackageRepositories,
|
|
tasks.apt.ImportGoogleKey,
|
|
tasks.packages.DefaultPackages,
|
|
tasks.packages.ReleasePackages,
|
|
tasks.packages.GooglePackages,
|
|
|
|
tasks.configuration.GatherReleaseInformation,
|
|
|
|
tasks.host.DisableIPv6,
|
|
tasks.host.InstallHostnameHook,
|
|
tasks.boot.ConfigureGrub,
|
|
initd.AddExpandRoot,
|
|
tasks.initd.AdjustExpandRootDev,
|
|
initd.InstallInitScripts,
|
|
boot.BlackListModules,
|
|
boot.UpdateInitramfs,
|
|
ssh.AddSSHKeyGeneration,
|
|
ssh.DisableSSHPasswordAuthentication,
|
|
tasks.apt.CleanGoogleRepositoriesAndKeys,
|
|
|
|
loopback.MoveImage,
|
|
tasks.image.CreateTarball,
|
|
volume.Delete,
|
|
])
|
|
|
|
if manifest.volume['partitions']['type'] != 'none':
|
|
taskset.add(initd.AdjustExpandRootScript)
|
|
|
|
if manifest.volume['partitions']['type'] != 'mbr':
|
|
taskset.update([tasks.initd.AddGrowRootDisable,
|
|
kernel.UpdateInitramfs])
|
|
|
|
if 'gcs_destination' in manifest.image:
|
|
taskset.add(tasks.image.UploadImage)
|
|
if 'gce_project' in manifest.image:
|
|
taskset.add(tasks.image.RegisterImage)
|
|
|
|
|
|
def resolve_rollback_tasks(taskset, manifest, completed, counter_task):
|
|
taskset.update(task_groups.get_standard_rollback_tasks(completed))
|