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"
+class Kamaki(object):
+ CONTAINER = "images"
-class Kamaki:
- __init__(self, account, token):
- self.username = username
- self.token = token
+ @staticmethod
+ def get_token():
+ """Get the saved token"""
+ config = Config()
+ return config.get('global', 'token')
+
+ @staticmethod
+ def save_token(token):
+ """Save this token to the configuration file"""
+ config = Config()
+ config.set('global', 'token', token)
+ config.write()
+ @staticmethod
+ def get_account(token):
+ """Return the account corresponding to this token"""
config = Config()
+ astakos = AstakosClient(config.get('user', 'url'), token)
+ try:
+ account = astakos.info()
+ except ClientError as e:
+ if e.status == 401: # Unauthorized: invalid token
+ return None
+ else:
+ raise
+ return account
+
+ def __init__(self, account, output):
+ """Create a Kamaki instance"""
+ self.account = account
+ self.out = output
- pithos_url = config.get('storage', 'url')
- self.account = config.get('storage', 'account')
- self.container = CONTAINER
- self.pithos_client = PithosClient(pithos_url, token, self.account,
- self.container)
+ config = Config()
+
+ pithos_url = config.get('file', 'url')
+ self.pithos_client = PithosClient(
+ pithos_url, self.account['auth_token'], self.account['uuid'],
+ self.CONTAINER)
image_url = config.get('image', 'url')
- self.image_client = ImageClient(image_url, token)
+ self.image_client = ImageClient(image_url, self.account['auth_token'])
+
+ def upload(self, file_obj, size=None, remote_path=None, hp=None, up=None):
+ """Upload a file to pithos"""
+
+ path = basename(file_obj.name) if remote_path is None else remote_path
+
+ try:
+ self.pithos_client.create_container(self.CONTAINER)
+ except ClientError as e:
+ if e.status != 202: # Ignore container already exists errors
+ raise e
+
+ 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.upload_object(path, file_obj, size, hash_cb,
+ upload_cb)
+
+ return "pithos://%s/%s/%s" % (self.account['uuid'], self.CONTAINER,
+ path)
- self.uploaded_object = None
+ def register(self, name, location, metadata, public=False):
+ """Register an image to ~okeanos"""
- set_container(self, container):
- self.pithos_client.container = container
+ # 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'}
+ self.image_client.register(name, location, params, str_metadata)
- upload(self, filename, size=None, remote_path=None):
+ def share(self, location):
+ """Share this file with all the users"""
- if remote_path is None:
- remote_path = basename(filename)
+ self.pithos_client.set_object_sharing(location, "*")
- with open(filename) as f:
- # TODO: create container if necessary
- self.pithos_client.create_object(remote_path, f, size)
- self.uploaded_object = "pithos://%s/%s/%s" % \
- (self.account, self.container, remote_path)
+ def object_exists(self, location):
+ """Check if an object exists in pythos"""
- register(self, metadata):
- pass
+ try:
+ self.pithos_client.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
+# vim: set sta sts=4 shiftwidth=4 sw=4 et ai :