X-Git-Url: https://code.grnet.gr/git/snf-image-creator/blobdiff_plain/f99fe99d6f3beaf5b282267f30c1ca288be92e40..b3a4845c0699ab1d22ca88f6209f8c704a069e7a:/image_creator/kamaki_wrapper.py diff --git a/image_creator/kamaki_wrapper.py b/image_creator/kamaki_wrapper.py index 6b05142..af588c4 100644 --- a/image_creator/kamaki_wrapper.py +++ b/image_creator/kamaki_wrapper.py @@ -1,3 +1,5 @@ +# -*- coding: utf-8 -*- +# # Copyright 2012 GRNET S.A. All rights reserved. # # Redistribution and use in source and binary forms, with or @@ -31,35 +33,115 @@ # interpreted as representing official policies, either expressed # or implied, of GRNET S.A. +"""This modules provides the interface for working with the ./kamaki library. +The library is used to upload images to and register them with a Synnefo +deployment. +""" + from os.path import basename -from kamaki.config import Config +from kamaki.cli.config import Config from kamaki.clients import ClientError from kamaki.clients.image import ImageClient from kamaki.clients.pithos import PithosClient +from kamaki.clients.astakos import AstakosClient -from image_creator.util import FatalError -CONTAINER = "images" +config = Config() class Kamaki(object): - def __init__(self, account, token, output): + """Wrapper class for the ./kamaki library""" + CONTAINER = "images" + + @staticmethod + def get_default_cloud_name(): + """Returns the name of the default cloud""" + clouds = config.keys('cloud') + default = config.get('global', 'default_cloud') + if not default: + return clouds[0] if len(clouds) else "" + return default if default in clouds else "" + + @staticmethod + def set_default_cloud(name): + """Sets a cloud account as default""" + config.set('global', 'default_cloud', name) + config.write() + + @staticmethod + def get_clouds(): + """Returns the list of available clouds""" + names = config.keys('cloud') + + clouds = {} + for name in names: + clouds[name] = config.get('cloud', name) + + return clouds + + @staticmethod + def get_cloud_by_name(name): + """Returns a dict with cloud info""" + return config.get('cloud', name) + + @staticmethod + def save_cloud(name, url, token, description=""): + """Save a new cloud account""" + cloud = {'url': url, 'token': token} + if len(description): + cloud['description'] = description + config.set('cloud', name, cloud) + + # Make the saved cloud the default one + config.set('global', 'default_cloud', name) + config.write() + + @staticmethod + def remove_cloud(name): + """Deletes an existing cloud from the Kamaki configuration file""" + config.remove_option('cloud', name) + config.write() + + @staticmethod + def create_account(url, token): + """Given a valid (URL, tokens) pair this method returns an Astakos + client instance + """ + client = AstakosClient(url, token) + try: + client.authenticate() + except ClientError: + return None + + return client + + @staticmethod + def get_account(cloud_name): + """Given a saved cloud name this method returns an Astakos client + instance + """ + cloud = config.get('cloud', cloud_name) + assert cloud, "cloud: `%s' does not exist" % cloud_name + assert 'url' in cloud, "url attr is missing in %s" % cloud_name + assert 'token' in cloud, "token attr is missing in %s" % cloud_name + + return Kamaki.create_account(cloud['url'], cloud['token']) + + def __init__(self, account, output): + """Create a Kamaki instance""" self.account = account - self.token = token self.out = output - config = Config() - - pithos_url = config.get('storage', 'url') - self.container = CONTAINER - self.pithos_client = PithosClient(pithos_url, token, self.account, - self.container) - - image_url = config.get('image', 'url') - self.image_client = ImageClient(image_url, token) + self.pithos = PithosClient( + self.account.get_service_endpoints('object-store')['publicURL'], + self.account.token, + self.account.user_info()['id'], + self.CONTAINER) - self.uploaded_object = None + self.image = ImageClient( + self.account.get_service_endpoints('image')['publicURL'], + self.account.token) def upload(self, file_obj, size=None, remote_path=None, hp=None, up=None): """Upload a file to pithos""" @@ -67,7 +149,7 @@ class Kamaki(object): path = basename(file_obj.name) if remote_path is None else remote_path try: - self.pithos_client.create_container(self.container) + self.pithos.create_container(self.CONTAINER) except ClientError as e: if e.status != 202: # Ignore container already exists errors raise e @@ -75,20 +157,37 @@ class Kamaki(object): hash_cb = self.out.progress_generator(hp) if hp is not None else None upload_cb = self.out.progress_generator(up) if up is not None else None - self.pithos_client.create_object(path, file_obj, size, hash_cb, - upload_cb) + self.pithos.upload_object(path, file_obj, size, hash_cb, upload_cb) - return "pithos://%s/%s/%s" % (self.account, self.container, path) + return "pithos://%s/%s/%s" % (self.account.user_info()['id'], + self.CONTAINER, path) - def register(self, name, location, metadata): + def register(self, name, location, metadata, public=False): """Register an image to ~okeanos""" # Convert all metadata to strings str_metadata = {} for (key, value) in metadata.iteritems(): str_metadata[str(key)] = str(value) + is_public = 'true' if public else 'false' + params = {'is_public': is_public, 'disk_format': 'diskdump'} + return self.image.register(name, location, params, str_metadata) - params = {'is_public': 'true', 'disk_format': 'diskdump'} - self.image_client.register(name, location, params, str_metadata) + def share(self, location): + """Share this file with all the users""" + + self.pithos.set_object_sharing(location, "*") + + def object_exists(self, location): + """Check if an object exists in pythos""" + + try: + self.pithos.get_object_info(location) + except ClientError as e: + if e.status == 404: # Object not found error + return False + else: + raise + return True # vim: set sta sts=4 shiftwidth=4 sw=4 et ai :