Synnefo » Pithos

    # TODO: Automatic/manual clean-up after a time interval.

        sql = '''create table if not exists versions (

                    version_id integer primary key,

                    name text,

                    tstamp datetime default current_timestamp,

                    size integer default 0,

                    hide integer default 0)'''

                    version_id integer, key text, value text, primary key (version_id, key))'''

                    version_id integer, pos integer, block_id text, primary key (version_id, pos))'''

    def delete_account(self, account):

        """Delete the account with the given name."""

        logger.debug("delete_account: %s", account)

        count, bytes, tstamp = self._get_pathstats(account)

        if count > 0:

            raise IndexError('Account is not empty')

        self._del_path(account) # Point of no return.

    def get_account_meta(self, account, until=None):

        logger.debug("get_account_meta: %s %s", account, until)

        try:

            version_id, mtime = self._get_accountinfo(account, until)

        except NameError:

            version_id = None

        count, bytes, tstamp = self._get_pathstats(account, until)

        if until is None:

            modified = tstamp

        else:

            modified = self._get_pathstats(account)[2] # Overall last modification

        sql = 'select count(name) from (%s) where name glob ? and not name glob ?'

        sql = sql % self._sql_until(until)

        meta = self._get_metadata(account, version_id)

        if modified:

            meta.update({'modified': modified})

        if until is not None:

            meta.update({'until_timestamp': tstamp})

        self._put_metadata(account, meta, replace)

    def list_containers(self, account, marker=None, limit=10000, until=None):

        """Return a list of containers existing under an account."""

        logger.debug("list_containers: %s %s %s %s", account, marker, limit, until)

        return self._list_objects(account, '', '/', marker, limit, False, [], until)

    def put_container(self, account, container):

        logger.debug("put_container: %s %s", account, container)

            path, version_id, mtime = self._get_containerinfo(account, container)

            path = os.path.join(account, container)

            version_id = self._put_version(path)

    def delete_container(self, account, container):

        logger.debug("delete_container: %s %s", account, container)

        path, version_id, mtime = self._get_containerinfo(account, container)

        count, bytes, tstamp = self._get_pathstats(path)

        self._del_path(path) # Point of no return.

        self._copy_version(account, account, True, True) # New account version.

    def get_container_meta(self, account, container, until=None):

        logger.debug("get_container_meta: %s %s %s", account, container, until)

        path, version_id, mtime = self._get_containerinfo(account, container, until)

        count, bytes, tstamp = self._get_pathstats(path, until)

        if until is None:

            modified = tstamp

        else:

            modified = self._get_pathstats(account)[2] # Overall last modification

        meta = self._get_metadata(path, version_id)

        meta.update({'name': container, 'count': count, 'bytes': bytes, 'modified': modified})

        if until is not None:

            meta.update({'until_timestamp': tstamp})

    def update_container_meta(self, account, container, meta, replace=False):

        logger.debug("update_container_meta: %s %s %s %s", account, container, meta, replace)

        path, version_id, mtime = self._get_containerinfo(account, container)

        self._put_metadata(path, meta, replace)

    def list_objects(self, account, container, prefix='', delimiter=None, marker=None, limit=10000, virtual=True, keys=[], until=None):

        logger.debug("list_objects: %s %s %s %s %s %s %s", account, container, prefix, delimiter, marker, limit, until)

        path, version_id, mtime = self._get_containerinfo(account, container, until)

        return self._list_objects(path, prefix, delimiter, marker, limit, virtual, keys, until)

    def list_object_meta(self, account, container, until=None):

        logger.debug("list_object_meta: %s %s %s", account, container, until)

        path, version_id, mtime = self._get_containerinfo(account, container, until)

        sql = '''select distinct m.key from (%s) o, metadata m

                    where m.version_id = o.version_id and o.name like ?'''

        sql = sql % self._sql_until(until)

    def get_object_meta(self, account, container, name, version=None):

        logger.debug("get_object_meta: %s %s %s %s", account, container, name, version)

        path, version_id, mtime, size = self._get_objectinfo(account, container, name, version)

        if version is None:

            modified = mtime

        else:

            modified = self._get_version(path)[1] # Overall last modification

        meta = self._get_metadata(path, version_id)

        meta.update({'name': name, 'bytes': size, 'version': version_id, 'version_timestamp': mtime, 'modified': modified})

        path, version_id, mtime, size = self._get_objectinfo(account, container, name)

        self._put_metadata(path, meta, replace)

        path, version_id, mtime, size = self._get_objectinfo(account, container, name, version)

        sql = 'select block_id from hashmaps where version_id = ? order by pos asc'

        c = self.con.execute(sql, (version_id,))

        path = self._get_containerinfo(account, container)[0]

        path = os.path.join(path, name)

        src_version_id, dest_version_id = self._copy_version(path, path, True, False)

        sql = 'update versions set size = ? where version_id = ?'

        self.con.execute(sql, (size, dest_version_id))

        # TODO: Check for block_id existence.

            self.con.execute(sql, (dest_version_id, i, hashmap[i]))

    def copy_object(self, account, src_container, src_name, dest_container, dest_name, dest_meta={}, replace_meta=False, src_version=None):

        logger.debug("copy_object: %s %s %s %s %s %s %s %s", account, src_container, src_name, dest_container, dest_name, dest_meta, replace_meta, src_version)

        if src_version is None:

            src_path = self._get_objectinfo(account, src_container, src_name)[0]

            src_path = os.path.join(account, src_container, src_name)

        dest_path = self._get_containerinfo(account, dest_container)[0]

        dest_path = os.path.join(dest_path, dest_name)

        src_version_id, dest_version_id = self._copy_version(src_path, dest_path, not replace_meta, True, src_version)

        for k, v in dest_meta.iteritems():

            sql = 'insert or replace into metadata (version_id, key, value) values (?, ?, ?)'

            self.con.execute(sql, (dest_version_id, k, v))

        self.con.commit()

    def move_object(self, account, src_container, src_name, dest_container, dest_name, dest_meta={}, replace_meta=False, src_version=None):

        logger.debug("move_object: %s %s %s %s, %s %s %s %s", account, src_container, src_name, dest_container, dest_name, dest_meta, replace_meta, src_version)

        self.copy_object(account, src_container, src_name, dest_container, dest_name, dest_meta, replace_meta, src_version)

        path, version_id, mtime, size = self._get_objectinfo(account, container, name)

        self._put_version(path, 0, 1)

    def list_versions(self, account, container, name):

        """Return a list of version (version_id, version_modified) tuples for an object."""

        # This will even show deleted versions.

        sql = '''select distinct version_id, strftime('%s', tstamp) from versions where name = ?'''

        c = self.con.execute(sql, (path,))

        return [(str(x[0]), int(x[1])) for x in c.fetchall()]

    def _sql_until(self, until=None):

        """Return the sql to get the latest versions until the timestamp given."""

        if until is None:

            until = int(time.time())

        sql = '''select version_id, name, strftime('%s', tstamp) as tstamp, size from versions v

                    where version_id = (select max(version_id) from versions

                                        where v.name = name and tstamp <= datetime(%s, 'unixepoch'))

                    and hide = 0'''

        return sql % ('%s', until)

    def _get_pathstats(self, path, until=None):

        """Return count and sum of size of everything under path and latest timestamp."""

        sql = 'select count(version_id), total(size), max(tstamp) from (%s) where name like ?'

        sql = sql % self._sql_until(until)

        c = self.con.execute(sql, (path + '/%',))

        tstamp = row[2] if row[2] is not None else 0

        return int(row[0]), int(row[1]), int(tstamp)

    def _get_version(self, path, version=None):

        if version is None:            

            sql = '''select version_id, strftime('%s', tstamp), size, hide from versions where name = ?

                        order by version_id desc limit 1'''

            c = self.con.execute(sql, (path,))

            row = c.fetchone()

            if not row or int(row[3]):

                raise NameError('Object does not exist')

            sql = '''select version_id, strftime('%s', tstamp), size from versions where name = ?

                        and version_id = ?'''

            c = self.con.execute(sql, (path, version))

            row = c.fetchone()

            if not row:

                raise IndexError('Version does not exist')

        return str(row[0]), int(row[1]), int(row[2])

    def _put_version(self, path, size=0, hide=0):

        sql = 'insert into versions (name, size, hide) values (?, ?, ?)'

        id = self.con.execute(sql, (path, size, hide)).lastrowid

    def _copy_version(self, src_path, dest_path, copy_meta=True, copy_data=True, src_version=None):

        if src_version is not None:

            src_version_id, mtime, size = self._get_version(src_path, src_version)

        else:

            # Latest or create from scratch.

            try:

                src_version_id, mtime, size = self._get_version(src_path)

            except NameError:

                src_version_id = None

                size = 0

        if not copy_data:

            size = 0

        dest_version_id = self._put_version(dest_path, size)

        if copy_meta and src_version_id is not None:

            sql = 'insert into metadata select %s, key, value from metadata where version_id = ?'

            sql = sql % dest_version_id

            self.con.execute(sql, (src_version_id,))

        if copy_data and src_version_id is not None:

            sql = 'insert into hashmaps select %s, pos, block_id from hashmaps where version_id = ?'

            sql = sql % dest_version_id

            self.con.execute(sql, (src_version_id,))

        self.con.commit()

        return src_version_id, dest_version_id

    def _get_versioninfo(self, account, container, name, until=None):

        """Return path, latest version, associated timestamp and size until the timestamp given."""

        p = (account, container, name)

            p = p[:p.index(None)]

        except ValueError:

            pass

        path = os.path.join(*p)

        sql = '''select version_id, tstamp, size from (%s) where name = ?'''

        sql = sql % self._sql_until(until)

        c = self.con.execute(sql, (path,))

        row = c.fetchone()

        if row is None:

            raise NameError('Path does not exist')

        return path, str(row[0]), int(row[1]), int(row[2])

    def _get_accountinfo(self, account, until=None):

        try:

            path, version_id, mtime, size = self._get_versioninfo(account, None, None, until)

            return version_id, mtime

        except:

            raise NameError('Account does not exist')

    def _get_containerinfo(self, account, container, until=None):

        try:

            path, version_id, mtime, size = self._get_versioninfo(account, container, None, until)

            return path, version_id, mtime

        except:

        version_id, mtime, size = self._get_version(path, version)

        return path, version_id, mtime, size

    def _get_metadata(self, path, version):

        sql = 'select key, value from metadata where version_id = ?'

        c = self.con.execute(sql, (version,))

        return dict(c.fetchall())

    def _put_metadata(self, path, meta, replace=False):

        """Create a new version and store metadata."""

        src_version_id, dest_version_id = self._copy_version(path, path, not replace, True)

        for k, v in meta.iteritems():

            sql = 'insert or replace into metadata (version_id, key, value) values (?, ?, ?)'

            self.con.execute(sql, (dest_version_id, k, v))

        self.con.commit()

    def _list_objects(self, path, prefix='', delimiter=None, marker=None, limit=10000, virtual=True, keys=[], until=None):

            sql = '''select distinct o.name, o.version_id from (%s) o, metadata m where o.name like ? and

                        m.version_id = o.version_id and m.key in (%s) order by o.name'''

            sql = sql % (self._sql_until(until), ', '.join('?' * len(keys)))

            sql = 'select name, version_id from (%s) where name like ? order by name'

            sql = sql % self._sql_until(until)

        objects = [(x[0][len(cont_prefix):], x[1]) for x in c.fetchall()]

                pseudo_name = x[0]

                        pseudo_objects.append(x)

                    if pseudo_name not in [y[0] for y in pseudo_objects]:

                        pseudo_objects.append((pseudo_name, x[1]))

                start = [x[0] for x in objects].index(marker) + 1

    def _del_path(self, path):

                    (select version_id from versions where name = ?)'''

        self.con.execute(sql, (path,))

        sql = '''delete from metadata where version_id in

                    (select version_id from versions where name = ?)'''

        sql = '''delete from versions where name = ?'''