mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-24 07:26:29 +00:00
c880a6849a
All provider specific settings have been moved to the provider section. The image name itself is now located at the top level and called "name". It is required for all providers.
61 lines
2.4 KiB
Python
61 lines
2.4 KiB
Python
from bootstrapvz.base import Task
|
|
from bootstrapvz.common import phases
|
|
from bootstrapvz.common.tasks import image
|
|
from bootstrapvz.common.tools import log_check_call
|
|
import os.path
|
|
|
|
|
|
class CreateTarball(Task):
|
|
description = 'Creating tarball with image'
|
|
phase = phases.image_registration
|
|
predecessors = [image.MoveImage]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
import datetime
|
|
image_name = info.manifest.name.format(**info.manifest_vars)
|
|
filename = image_name + '.' + info.volume.extension
|
|
today = datetime.datetime.today()
|
|
name_suffix = today.strftime('%Y%m%d')
|
|
image_name_format = '{lsb_distribution}-{lsb_release}-{release}-v{name_suffix}'
|
|
image_name = image_name_format.format(lsb_distribution=info._gce['lsb_distribution'],
|
|
lsb_release=info._gce['lsb_release'],
|
|
release=info.manifest.system['release'],
|
|
name_suffix=name_suffix)
|
|
# ensure that we do not use disallowed characters in image name
|
|
image_name = image_name.lower()
|
|
image_name = image_name.replace(".", "-")
|
|
info._gce['image_name'] = image_name
|
|
tarball_name = image_name + '.tar.gz'
|
|
tarball_path = os.path.join(info.manifest.bootstrapper['workspace'], tarball_name)
|
|
info._gce['tarball_name'] = tarball_name
|
|
info._gce['tarball_path'] = tarball_path
|
|
log_check_call(['tar', '--sparse', '-C', info.manifest.bootstrapper['workspace'],
|
|
'-caf', tarball_path, filename])
|
|
|
|
|
|
class UploadImage(Task):
|
|
description = 'Uploading image to GCS'
|
|
phase = phases.image_registration
|
|
predecessors = [CreateTarball]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
log_check_call(['gsutil', 'cp', info._gce['tarball_path'],
|
|
info.manifest.provider['gcs_destination'] + info._gce['tarball_name']])
|
|
|
|
|
|
class RegisterImage(Task):
|
|
description = 'Registering image with GCE'
|
|
phase = phases.image_registration
|
|
predecessors = [UploadImage]
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
image_description = info._gce['lsb_description']
|
|
if 'description' in info.manifest.provider:
|
|
image_description = info.manifest.provider['description']
|
|
log_check_call(['gcutil', '--project=' + info.manifest.provider['gce_project'],
|
|
'addimage', info._gce['image_name'],
|
|
info.manifest.provider['gcs_destination'] + info._gce['tarball_name'],
|
|
'--description=' + image_description])
|