Synnefo » ./kamaki

# Copyright 2012-2013 GRNET S.A. All rights reserved.

#

# Redistribution and use in source and binary forms, with or

# without modification, are permitted provided that the following

# conditions are met:

#

#   1. Redistributions of source code must retain the above

#      copyright notice, this list of conditions and the following

#      disclaimer.

#

#   2. Redistributions in binary form must reproduce the above

#      copyright notice, this list of conditions and the following

#      disclaimer in the documentation and/or other materials

#      provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS

# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR

# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF

# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED

# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN

# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

# POSSIBILITY OF SUCH DAMAGE.

#

# The views and conclusions contained in the software and

# documentation are those of the authors and should not be

# interpreted as representing official policies, either expressed

# or implied, of GRNET S.A.

from kamaki.clients.storage import StorageClient

from kamaki.clients.utils import path4url

class PithosRestClient(StorageClient):

    def account_head(

            self,

            until=None,

            if_modified_since=None,

            if_unmodified_since=None,

            *args, **kwargs):

        """ Full Pithos+ HEAD at account level

        --- request parameters ---

        :param until: (string) optional timestamp

        --- request headers ---

        :param if_modified_since: (string) Retrieve if account has changed

            since provided timestamp

        :param if_unmodified_since: (string) Retrieve if account has not

            change since provided timestamp

        :returns: ConnectionResponse

        """

        self._assert_account()

        path = path4url(self.account)

        self.set_param('until', until, iff=until)

        self.set_header('If-Modified-Since', if_modified_since)

        self.set_header('If-Unmodified-Since', if_unmodified_since)

        success = kwargs.pop('success', 204)

        return self.head(path, *args, success=success, **kwargs)

    def account_get(

            self,

            limit=None,

            marker=None,

            format='json',

            show_only_shared=False,

            until=None,

            if_modified_since=None,

            if_unmodified_since=None,

            *args, **kwargs):

        """  Full Pithos+ GET at account level

        --- request parameters ---

        :param limit: (integer) The amount of results requested

            (server will use default value if None)

        :param marker: string Return containers with name

            lexicographically after marker

        :param format: (string) reply format can be json or xml

            (default: json)

        :param shared: (bool) If true, only shared containers will be

            included in results

        :param until: (string) optional timestamp

        --- request headers ---

        :param if_modified_since: (string) Retrieve if account has changed

            since provided timestamp

        :param if_unmodified_since: (string) Retrieve if account has not

            changed since provided timestamp

        :returns: ConnectionResponse

        """

        self._assert_account()

        self.set_param('limit', limit, iff=limit)

        self.set_param('marker', marker, iff=marker)

        self.set_param('format', format, iff=format)

        self.set_param('shared', iff=show_only_shared)

        self.set_param('until', until, iff=until)

        self.set_header('If-Modified-Since', if_modified_since)

        self.set_header('If-Unmodified-Since', if_unmodified_since)

        path = path4url(self.account)

        success = kwargs.pop('success', (200, 204))

        return self.get(path, *args, success=success, **kwargs)

    def account_post(

            self,

            update=True,

            groups={},

            metadata=None,

            quota=None,

            versioning=None,

            *args, **kwargs):

        """ Full Pithos+ POST at account level

        --- request parameters ---

        :param update: (bool) if True, Do not replace metadata/groups

        --- request headers ---

        :param groups: (dict) Optional user defined groups in the form

            { 'group1':['user1', 'user2', ...],

            'group2':['userA', 'userB', ...], }

        :param metadata: (dict) Optional user defined metadata in the form

            { 'name1': 'value1', 'name2': 'value2', ... }

        :param quota: (integer) If supported, sets the Account quota

        :param versioning: (string) If supported, sets the Account versioning

            to 'auto' or some other supported versioning string

        :returns: ConnectionResponse

        """

        self._assert_account()

        self.set_param('update', iff=update)

        if groups:

            for group, usernames in groups.items():

                userstr = ''

                dlm = ''

                for user in usernames:

                    userstr = userstr + dlm + user

                    dlm = ','

                self.set_header('X-Account-Group-' + group, userstr)

        if metadata:

            for metaname, metaval in metadata.items():

                self.set_header('X-Account-Meta-' + metaname, metaval)

        self.set_header('X-Account-Policy-Quota', quota)

        self.set_header('X-Account-Policy-Versioning', versioning)

        path = path4url(self.account)

        success = kwargs.pop('success', 202)

        return self.post(path, *args, success=success, **kwargs)

    def container_head(

            self,

            until=None,

            if_modified_since=None,

            if_unmodified_since=None,

            *args, **kwargs):

        """ Full Pithos+ HEAD at container level

        --- request params ---

        :param until: (string) optional timestamp

        --- request headers ---

        :param if_modified_since: (string) Retrieve if account has changed

            since provided timestamp

        :param if_unmodified_since: (string) Retrieve if account has not

            changed since provided timestamp

        :returns: ConnectionResponse

        """

        self._assert_container()

        self.set_param('until', until, iff=until)

        self.set_header('If-Modified-Since', if_modified_since)

        self.set_header('If-Unmodified-Since', if_unmodified_since)

        path = path4url(self.account, self.container)

        success = kwargs.pop('success', 204)

        return self.head(path, *args, success=success, **kwargs)

    def container_get(

            self,

            limit=None,

            marker=None,

            prefix=None,

            delimiter=None,

            path=None,

            format='json',

            meta=[],

            show_only_shared=False,

            until=None,

            if_modified_since=None,

            if_unmodified_since=None,

            *args, **kwargs):

        """ Full Pithos+ GET at container level

        --- request parameters ---

        :param limit: (integer) The amount of results requested

            (server will use default value if None)

        :param marker: (string) Return containers with name lexicographically

            after marker

        :param prefix: (string) Return objects starting with prefix

        :param delimiter: (string) Return objects up to the delimiter

        :param path: (string) assume prefix = path and delimiter = /

            (overwrites prefix and delimiter)

        :param format: (string) reply format can be json or xml (default:json)

        :param meta: (list) Return objects that satisfy the key queries in

            the specified comma separated list (use <key>, !<key> for

            existence queries, <key><op><value> for value queries, where <op>

            can be one of =, !=, <=, >=, <, >)

        :param show_only_shared: (bool) If true, only shared containers will

            be included in results

        :param until: (string) optional timestamp

        --- request headers ---

        :param if_modified_since: (string) Retrieve if account has changed

            since provided timestamp

        :param if_unmodified_since: (string) Retrieve if account has not

            changed since provided timestamp

        :returns: ConnectionResponse

        """

        self._assert_container()

        self.set_param('limit', limit, iff=limit)

        self.set_param('marker', marker, iff=marker)

        if not path:

            self.set_param('prefix', prefix, iff=prefix)

            self.set_param('delimiter', delimiter, iff=delimiter)

        else:

            self.set_param('path', path)

        self.set_param('format', format, iff=format)

        self.set_param('shared', iff=show_only_shared)

        if meta:

            self.set_param('meta',  ','.join(meta))

        self.set_param('until', until, iff=until)

        self.set_header('If-Modified-Since', if_modified_since)

        self.set_header('If-Unmodified-Since', if_unmodified_since)

        path = path4url(self.account, self.container)

        success = kwargs.pop('success', 200)

        return self.get(path, *args, success=success, **kwargs)

    def container_put(

            self,

            quota=None, versioning=None, metadata=None,

            *args, **kwargs):

        """ Full Pithos+ PUT at container level

        --- request headers ---

        :param quota: (integer) Size limit in KB

        :param versioning: (string) 'auto' or other string supported by server

        :param metadata: (dict) Optional user defined metadata in the form

            { 'name1': 'value1', 'name2': 'value2', ... }

        :returns: ConnectionResponse

        """

        self._assert_container()

        if metadata:

            for metaname, metaval in metadata.items():

                self.set_header('X-Container-Meta-' + metaname, metaval)

        self.set_header('X-Container-Policy-Quota', quota)

        self.set_header('X-Container-Policy-Versioning', versioning)

        path = path4url(self.account, self.container)

        success = kwargs.pop('success', (201, 202))

        return self.put(path, *args, success=success, **kwargs)

    def container_post(

            self,

            update=True,

            format='json',

            quota=None,

            versioning=None,

            metadata=None,

            content_type=None,

            content_length=None,

            transfer_encoding=None,

            *args, **kwargs):

        """ Full Pithos+ POST at container level

        --- request params ---

        :param update: (bool)  if True, Do not replace metadata/groups

        :param format: (string) json (default) or xml

        --- request headers ---

        :param quota: (integer) Size limit in KB

        :param versioning: (string) 'auto' or other string supported by server

        :param metadata: (dict) Optional user defined metadata in the form

            { 'name1': 'value1', 'name2': 'value2', ... }

        :param content_type: (string) set a custom content type

        :param content_length: (string) set a custrom content length

        :param transfer_encoding: (string) set a custrom transfer encoding

        :returns: ConnectionResponse

        """

        self._assert_container()

        self.set_param('format', format, iff=format)

        self.set_param('update', iff=update)

        if metadata:

            for metaname, metaval in metadata.items():

                self.set_header('X-Container-Meta-' + metaname, metaval)

        self.set_header('X-Container-Policy-Quota', quota)

        self.set_header('X-Container-Policy-Versioning', versioning)

        self.set_header('Content-Type', content_type)

        self.set_header('Content-Length', content_length)

        self.set_header('Transfer-Encoding', transfer_encoding)

        path = path4url(self.account, self.container)

        success = kwargs.pop('success', 202)

        return self.post(path, *args, success=success, **kwargs)

    def container_delete(self, until=None, delimiter=None, *args, **kwargs):

        """ Full Pithos+ DELETE at container level

        --- request parameters ---

        :param until: (timestamp string) if defined, container is purged up to

            that time

        :returns: ConnectionResponse

        """

        self._assert_container()

        self.set_param('until', until, iff=until)

        self.set_param('delimiter', delimiter, iff=delimiter)

        path = path4url(self.account, self.container)

        success = kwargs.pop('success', 204)

        return self.delete(path, success=success)

    def object_head(

            self, object,

            version=None,

            if_etag_match=None,

            if_etag_not_match=None,

            if_modified_since=None,

            if_unmodified_since=None,

            *args, **kwargs):

        """ Full Pithos+ HEAD at object level

        --- request parameters ---

        :param version: (string) optional version identified

        --- request headers ---

        :param if_etag_match: (string) if provided, return only results

            with etag matching with this

        :param if_etag_not_match: (string) if provided, return only results

            with etag not matching with this

        :param if_modified_since: (string) Retrieve if account has changed

            since provided timestamp

        :param if_unmodified_since: (string) Retrieve if account has not

            changed since provided timestamp

        :returns: ConnectionResponse

        """

        self._assert_container()

        self.set_param('version', version, iff=version)

        self.set_header('If-Match', if_etag_match)

        self.set_header('If-None-Match', if_etag_not_match)

        self.set_header('If-Modified-Since', if_modified_since)

        self.set_header('If-Unmodified-Since', if_unmodified_since)

        path = path4url(self.account, self.container, object)

        success = kwargs.pop('success', 200)

        return self.head(path, *args, success=success, **kwargs)

    def object_get(

            self, object,

            format='json',

            hashmap=False,

            version=None,

            data_range=None,

            if_range=False,

            if_etag_match=None,

            if_etag_not_match=None,

            if_modified_since=None,

            if_unmodified_since=None,

            *args, **kwargs):

        """ Full Pithos+ GET at object level

        --- request parameters ---

        :param format: (string) json (default) or xml

        :param hashmap: (bool) Optional request for hashmap

        :param version: (string) optional version identified

        --- request headers ---

        :param data_range: (string) Optional range of data to retrieve

        :param if_range: (bool)

        :param if_etag_match: (string) if provided, return only results

            with etag matching with this

        :param if_etag_not_match: (string) if provided, return only results

            with etag not matching with this

        :param if_modified_since: (string) Retrieve if account has changed

            since provided timestamp

        :param if_unmodified_since: (string) Retrieve if account has not

            changed since provided timestamp

        :returns: ConnectionResponse

        """

        self._assert_container()

        self.set_param('format', format, iff=format)

        self.set_param('version', version, iff=version)

        self.set_param('hashmap', hashmap, iff=hashmap)

        self.set_header('Range', data_range)

        self.set_header('If-Range', '', if_range and data_range)

        self.set_header('If-Match', if_etag_match, )

        self.set_header('If-None-Match', if_etag_not_match)

        self.set_header('If-Modified-Since', if_modified_since)

        self.set_header('If-Unmodified-Since', if_unmodified_since)

        path = path4url(self.account, self.container, object)

        success = kwargs.pop('success', 200)

        return self.get(path, *args, success=success, **kwargs)

    def object_put(

            self, object,

            format='json',

            hashmap=False,

            delimiter=None,

            if_etag_match=None,

            if_etag_not_match=None,

            etag=None,

            content_length=None,

            content_type=None,

            transfer_encoding=None,

            copy_from=None,

            move_from=None,

            source_account=None,

            source_version=None,

            content_encoding=None,

            content_disposition=None,

            manifest=None,

            permissions=None,

            public=None,

            metadata=None,

            *args, **kwargs):

        """ Full Pithos+ PUT at object level

        --- request parameters ---

        :param format: (string) json (default) or xml

        :param hashmap: (bool) Optional hashmap provided instead of data

        --- request headers ---

        :param if_etag_match: (string) if provided, return only results

            with etag matching with this

        :param if_etag_not_match: (string) if provided, return only results

            with etag not matching with this

        :param etag: (string) The MD5 hash of the object (optional to check

            written data)

        :param content_length: (integer) The size of the data written

        :param content_type: (string) The MIME content type of the object

        :param transfer_encoding: (string) Set to chunked to specify

            incremental uploading (if used, Content-Length is ignored)

        :param copy_from: (string) The source path in the form

            /<container>/<object>

        :param move_from: (string) The source path in the form

            /<container>/<object>

        :param source_account: (string) The source account to copy/move from

        :param source_version: (string) The source version to copy from

        :param conent_encoding: (string) The encoding of the object

        :param content_disposition: (string) Presentation style of the object

        :param manifest: (string) Object parts prefix in

            /<container>/<object> form

        :param permissions: (dict) Object permissions in the form (all fields

            are optional)

            { 'read':[user1, group1, user2, ...],

            'write':['user3, group2, group3, ...] }

        :param public: (bool) If true, Object is publicly accessible,

            if false, not

        :param metadata: (dict) Optional user defined metadata in the form

            {'meta-key-1':'meta-value-1', 'meta-key-2':'meta-value-2', ...}

        :returns: ConnectionResponse

        """

        self._assert_container()

        self.set_param('format', format, iff=format)

        self.set_param('hashmap', hashmap, iff=hashmap)

        self.set_param('delimiter', delimiter, iff=delimiter)

        self.set_header('If-Match', if_etag_match)

        self.set_header('If-None-Match', if_etag_not_match)

        self.set_header('ETag', etag)

        self.set_header('Content-Length', content_length)

        self.set_header('Content-Type', content_type)

        self.set_header('Transfer-Encoding', transfer_encoding)

        self.set_header('X-Copy-From', copy_from)

        self.set_header('X-Move-From', move_from)

        self.set_header('X-Source-Account', source_account)

        self.set_header('X-Source-Version', source_version)

        self.set_header('Content-Encoding', content_encoding)

        self.set_header('Content-Disposition', content_disposition)

        self.set_header('X-Object-Manifest', manifest)

        if permissions:

            perms = None

            if permissions:

                for perm_type, perm_list in permissions.items():

                    if not perms:

                        perms = ''  # Remove permissions

                    if perm_list:

                        perms += ';' if perms else ''

                        perms += '%s=%s' % (perm_type, ','.join(perm_list))

            self.set_header('X-Object-Sharing', perms)

        self.set_header('X-Object-Public', public)

        if metadata:

            for key, val in metadata.items():

                self.set_header('X-Object-Meta-' + key, val)

        path = path4url(self.account, self.container, object)

        success = kwargs.pop('success', 201)

        return self.put(path, *args, success=success, **kwargs)

    def object_copy(

            self, object, destination,

            format='json',

            ignore_content_type=False,

            if_etag_match=None,

            if_etag_not_match=None,

            destination_account=None,

            content_type=None,

            content_encoding=None,

            content_disposition=None,

            source_version=None,

            permissions=None,

            public=False,

            metadata=None,

            *args, **kwargs):

        """ Full Pithos+ COPY at object level

        --- request parameters ---

        :param format: (string) json (default) or xml

        :param ignore_content_type: (bool) Ignore the supplied Content-Type

        --- request headers ---

        :param if_etag_match: (string) if provided, copy only results

            with etag matching with this

        :param if_etag_not_match: (string) if provided, copy only results

            with etag not matching with this

        :param destination: (string) The destination path in the form

            /<container>/<object>

        :param destination_account: (string) The destination account to copy to

        :param content_type: (string) The MIME content type of the object

        :param content_encoding: (string) The encoding of the object

        :param content_disposition: (string) Object resentation style

        :param source_version: (string) The source version to copy from

        :param permissions: (dict) Object permissions in the form

            (all fields are optional)

            { 'read':[user1, group1, user2, ...],

            'write':['user3, group2, group3, ...] }

        :param permissions: update permissions

        :param public: (bool) If true, Object is publicly accessible

        :param metadata: (dict) Optional user defined metadata in the form

            {'meta-key-1':'meta-value-1', 'meta-key-2':'meta-value-2', ...}

            Metadata are appended to the source metadata. In case of same

            keys, they replace the old metadata

        :returns: ConnectionResponse

        """

        self._assert_container()

        self.set_param('format', format, iff=format)

        self.set_param('ignore_content_type', iff=ignore_content_type)

        self.set_header('If-Match', if_etag_match)

        self.set_header('If-None-Match', if_etag_not_match)

        self.set_header('Destination', destination)

        self.set_header('Destination-Account', destination_account)

        self.set_header('Content-Type', content_type)

        self.set_header('Content-Encoding', content_encoding)

        self.set_header('Content-Disposition', content_disposition)

        self.set_header('X-Source-Version', source_version)

        if permissions:

            perms = ''

            for perm_type, perm_list in permissions.items():

                if not perms:

                    perms = ''  # Remove permissions

                if perm_list:

                    perms += ';' if perms else ''

                    perms += '%s=%s' % (perm_type, ','.join(perm_list))

            self.set_header('X-Object-Sharing', perms)

        self.set_header('X-Object-Public', public)

        if metadata:

            for key, val in metadata.items():

                self.set_header('X-Object-Meta-' + key, val)

        path = path4url(self.account, self.container, object)

        success = kwargs.pop('success', 201)

        return self.copy(path, *args, success=success, **kwargs)

    def object_move(

            self, object,

            format='json',

            ignore_content_type=False,

            if_etag_match=None,

            if_etag_not_match=None,

            destination=None,

            destination_account=None,

            content_type=None,

            content_encoding=None,

            content_disposition=None,

            permissions={},

            public=False,