support restore operation on s3 keys

moto/s3/models.py

@@ -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):

moto/s3/responses.py

@@ -6,6 +6,7 @@ from jinja2 import Template
 from .exceptions import BucketAlreadyExists
 from .models import s3_backend
 from .utils import bucket_name_from_url
+from xml.dom import minidom
 
 
 def parse_key_name(pth):
@@ -261,8 +262,17 @@ class ResponseObject(object):
                     )
                 template = Template(S3_MULTIPART_COMPLETE_TOO_SMALL_ERROR)
                 return 400, headers, template.render()
+            elif parsed_url.query == 'restore':
+                es = minidom.parseString(body).getElementsByTagName('Days')
+                days = es[0].childNodes[0].wholeText
+                key = self.backend.get_key(bucket_name, key_name)
+                r = 202
+                if key.expiry_date is not None:
+                    r = 200
+                key.restore(int(days))
+                return r, headers, ""
             else:
-                raise NotImplementedError("Method POST had only been implemented for multipart uploads so far")
+                raise NotImplementedError("Method POST had only been implemented for multipart uploads and restore operations, so far")
         else:
             raise NotImplementedError("Method {0} has not been impelemented in the S3 backend yet".format(method))