2014-05-03 17:04:37 +02:00
|
|
|
from bootstrapvz.common import task_groups
|
2014-05-02 17:36:08 +02:00
|
|
|
import tasks.apt
|
|
|
|
import tasks.boot
|
|
|
|
import tasks.configuration
|
|
|
|
import tasks.image
|
|
|
|
import tasks.host
|
|
|
|
import tasks.packages
|
2014-07-12 14:42:52 -03:00
|
|
|
from bootstrapvz.common.tasks import apt
|
2014-05-02 17:36:08 +02:00
|
|
|
from bootstrapvz.common.tasks import loopback
|
2014-05-09 00:31:36 -07:00
|
|
|
from bootstrapvz.common.tasks import initd
|
2014-05-03 12:56:40 +02:00
|
|
|
from bootstrapvz.common.tasks import ssh
|
2014-06-07 12:45:58 -03:00
|
|
|
from bootstrapvz.common.tasks import volume
|
2014-05-02 17:36:08 +02:00
|
|
|
|
|
|
|
|
|
|
|
def initialize():
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
|
|
def validate_manifest(data, validator, error):
|
|
|
|
import os.path
|
2014-07-05 20:01:20 +02:00
|
|
|
schema_path = os.path.normpath(os.path.join(os.path.dirname(__file__), 'manifest-schema.yml'))
|
2014-05-02 17:36:08 +02:00
|
|
|
validator(data, schema_path)
|
|
|
|
|
|
|
|
|
2014-05-03 17:04:37 +02:00
|
|
|
def resolve_tasks(taskset, manifest):
|
|
|
|
taskset.update(task_groups.get_standard_groups(manifest))
|
2014-05-02 17:36:08 +02:00
|
|
|
|
2014-07-12 14:42:52 -03:00
|
|
|
taskset.update([apt.AddBackports,
|
2014-05-22 17:28:17 +02:00
|
|
|
loopback.AddRequiredCommands,
|
2014-05-03 17:04:37 +02:00
|
|
|
loopback.Create,
|
|
|
|
tasks.apt.SetPackageRepositories,
|
|
|
|
tasks.apt.ImportGoogleKey,
|
|
|
|
tasks.packages.DefaultPackages,
|
|
|
|
tasks.packages.GooglePackages,
|
|
|
|
tasks.packages.InstallGSUtil,
|
2014-05-02 17:36:08 +02:00
|
|
|
|
2014-05-03 17:04:37 +02:00
|
|
|
tasks.configuration.GatherReleaseInformation,
|
2014-05-02 17:36:08 +02:00
|
|
|
|
2014-05-03 17:04:37 +02:00
|
|
|
tasks.host.DisableIPv6,
|
|
|
|
tasks.boot.ConfigureGrub,
|
2014-05-09 00:31:36 -07:00
|
|
|
initd.InstallInitScripts,
|
2014-05-03 17:04:37 +02:00
|
|
|
ssh.AddSSHKeyGeneration,
|
|
|
|
tasks.apt.CleanGoogleRepositoriesAndKeys,
|
2014-05-02 17:36:08 +02:00
|
|
|
|
2014-05-03 17:04:37 +02:00
|
|
|
loopback.MoveImage,
|
|
|
|
tasks.image.CreateTarball,
|
2014-06-07 12:45:58 -03:00
|
|
|
volume.Delete,
|
2014-05-03 17:04:37 +02:00
|
|
|
])
|
2014-05-02 17:36:08 +02:00
|
|
|
|
|
|
|
if 'gcs_destination' in manifest.image:
|
2014-05-03 17:04:37 +02:00
|
|
|
taskset.add(tasks.image.UploadImage)
|
2014-05-02 17:36:08 +02:00
|
|
|
if 'gce_project' in manifest.image:
|
2014-05-03 17:04:37 +02:00
|
|
|
taskset.add(tasks.image.RegisterImage)
|
2014-05-02 17:36:08 +02:00
|
|
|
|
|
|
|
|
2014-05-03 17:04:37 +02:00
|
|
|
def resolve_rollback_tasks(taskset, manifest, completed, counter_task):
|
2014-05-03 13:28:24 +02:00
|
|
|
taskset.update(task_groups.get_standard_rollback_tasks(completed))
|