support restore operation on s3 keys

This commit is contained in:
Konstantinos Koukopoulos 2014-03-26 19:15:08 +02:00
commit 8ada1d7829
3 changed files with 59 additions and 2 deletions

View file

@ -20,6 +20,7 @@ class FakeKey(object):
self.last_modified = datetime.datetime.now()
self._storage_class = storage
self._metadata = {}
self._expiry = None
def copy(self, new_name = None):
r = copy.deepcopy(self)
@ -40,6 +41,9 @@ class FakeKey(object):
self.value += value
self.last_modified = datetime.datetime.now()
def restore(self, days):
self._expiry = datetime.datetime.now() + datetime.timedelta(days)
@property
def etag(self):
value_md5 = hashlib.md5()
@ -69,7 +73,9 @@ class FakeKey(object):
if self._storage_class is not None:
if self._storage_class != 'STANDARD':
r['x-amz-storage-class'] = self._storage_class
if self._expiry is not None:
r['x-amz-restore'] = 'ongoing-request="false", expiry-date={}'.format(
self.expiry_date)
return r
@property
@ -83,6 +89,11 @@ class FakeKey(object):
else:
return 'STANDARD'
@property
def expiry_date(self):
if self._expiry is not None:
return self._expiry.strftime("%a, %d %b %Y %H:%M:%S GMT")
class FakeMultipart(object):
def __init__(self, key_name):