bootstrap-vz/bootstrapvz/providers/gce/tasks/image.py

from bootstrapvz.base import Task
from bootstrapvz.common import phases
from bootstrapvz.common.tasks import loopback
from bootstrapvz.common.tools import log_check_call
import os.path


class CreateTarball(Task):
	description = 'Creating tarball with image'
	phase = phases.image_registration
	predecessors = [loopback.MoveImage]

	@classmethod
	def run(cls, info):
		import datetime
		image_name = info.manifest.image['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.image['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.image:
			image_description = info.manifest.image['description']
		log_check_call(['gcutil', '--project=' + info.manifest.image['gce_project'],
		                'addimage', info._gce['image_name'],
		                info.manifest.image['gcs_destination'] + info._gce['tarball_name'],
		                '--description=' + image_description])