mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-24 07:26:29 +00:00
f9372ce97e
Got stuck on the bundle image call, it uses argparse. Somehow we need to fake the arguments.
101 lines
3.6 KiB
Python
101 lines
3.6 KiB
Python
from manifest import Manifest
|
|
import logging
|
|
from tasks import packages
|
|
from tasks import connection
|
|
from tasks import host
|
|
from tasks import ami
|
|
from tasks import ebs
|
|
from tasks import loopback
|
|
from tasks import filesystem
|
|
from tasks import bootstrap
|
|
from tasks import locale
|
|
from tasks import apt
|
|
from tasks import boot
|
|
from tasks import security
|
|
from tasks import network
|
|
from tasks import initd
|
|
from tasks import cleanup
|
|
|
|
|
|
def initialize():
|
|
# Regardless of of loglevel, we don't want boto debug stuff, it's very noisy
|
|
logging.getLogger('boto').setLevel(logging.INFO)
|
|
|
|
|
|
def tasks(tasklist, manifest):
|
|
tasklist.add(packages.HostPackages(),
|
|
packages.ImagePackages(),
|
|
host.CheckPackages(),
|
|
connection.GetCredentials(),
|
|
host.GetInfo(),
|
|
ami.AMIName(),
|
|
connection.Connect(),
|
|
|
|
filesystem.FormatVolume(),
|
|
filesystem.CreateMountDir(),
|
|
filesystem.MountVolume(),
|
|
|
|
bootstrap.Bootstrap(),
|
|
filesystem.MountSpecials(),
|
|
locale.GenerateLocale(),
|
|
locale.SetTimezone(),
|
|
apt.DisableDaemonAutostart(),
|
|
apt.AptSources(),
|
|
apt.AptUpgrade(),
|
|
boot.ConfigureGrub(),
|
|
filesystem.ModifyFstab(),
|
|
boot.BlackListModules(),
|
|
boot.DisableGetTTYs(),
|
|
security.EnableShadowConfig(),
|
|
security.DisableSSHPasswordAuthentication(),
|
|
security.DisableSSHDNSLookup(),
|
|
network.RemoveDNSInfo(),
|
|
network.ConfigureNetworkIF(),
|
|
network.ConfigureDHCP(),
|
|
initd.ResolveInitScripts(),
|
|
initd.InstallInitScripts(),
|
|
cleanup.ClearMOTD(),
|
|
cleanup.ShredHostkeys(),
|
|
cleanup.CleanTMP(),
|
|
apt.PurgeUnusedPackages(),
|
|
apt.AptClean(),
|
|
apt.EnableDaemonAutostart(),
|
|
filesystem.UnmountSpecials(),
|
|
filesystem.UnmountVolume(),
|
|
filesystem.DeleteMountDir())
|
|
|
|
if manifest.bootstrapper['tarball']:
|
|
tasklist.add(bootstrap.MakeTarball())
|
|
|
|
backing_specific_tasks = {'ebs': [ebs.Create(),
|
|
ebs.Attach(),
|
|
ebs.Detach(),
|
|
ebs.Snapshot(),
|
|
ami.RegisterAMI(),
|
|
ebs.Delete()],
|
|
's3': [loopback.Create(),
|
|
loopback.Attach(),
|
|
loopback.Detach(),
|
|
loopback.Delete()]}
|
|
tasklist.add(*backing_specific_tasks.get(manifest.volume['backing'].lower()))
|
|
|
|
filesystem_specific_tasks = {'xfs': [filesystem.AddXFSProgs()],
|
|
'ext2': [filesystem.TuneVolumeFS()],
|
|
'ext3': [filesystem.TuneVolumeFS()],
|
|
'ext4': [filesystem.TuneVolumeFS()]}
|
|
tasklist.add(*filesystem_specific_tasks.get(manifest.volume['filesystem'].lower()))
|
|
|
|
|
|
def rollback_tasks(tasklist, tasks_completed, manifest):
|
|
completed = [type(task) for task in tasks_completed]
|
|
|
|
def counter_task(task, counter):
|
|
if task in completed and counter not in completed:
|
|
tasklist.add(counter())
|
|
|
|
if manifest.volume['backing'].lower() == 'ebs':
|
|
counter_task(ebs.Create, ebs.Delete)
|
|
counter_task(ebs.Attach, ebs.Detach)
|
|
counter_task(filesystem.CreateMountDir, filesystem.DeleteMountDir)
|
|
counter_task(filesystem.MountVolume, filesystem.UnmountVolume)
|
|
counter_task(filesystem.MountSpecials, filesystem.UnmountSpecials)
|