[snapshot]

python-consul · luhhujbb · Sep 23, 2019 · Sep 23, 2019 · Sep 23, 2019 · Sep 24, 2019
commit 081bdc0fe835ebe6648aaf881faa5c96c598a7cd
diff --git a/consul/aio.py b/consul/aio.py
@@ -25,9 +25,10 @@ def __init__(self, *args, loop=None, **kwargs):
     def _request(self, callback, method, uri, data=None):
         resp = yield from self._session.request(method, uri, data=data)
         body = yield from resp.text(encoding='utf-8')
+        content = yield from resp.read()
         if resp.status == 599:
             raise base.Timeout
-        r = base.Response(resp.status, resp.headers, body)
+        r = base.Response(resp.status, resp.headers, body, content)
         return callback(r)
 
     # python prior 3.4.1 does not play nice with __del__ method

diff --git a/consul/base.py b/consul/base.py
@@ -162,7 +162,7 @@ def _compat(
         return ret
 
 
-Response = collections.namedtuple('Response', ['code', 'headers', 'body'])
+Response = collections.namedtuple('Response', ['code', 'headers', 'body', 'content'])
 
 
 #
@@ -244,6 +244,16 @@ def cb(response):
             return data
         return cb
 
+    @classmethod
+    def binary(klass):
+        """
+        This method simply returns response body, usefull for snapshot
+        """
+        def cb(response):
+            CB._status(response)
+            return response.content
+        return cb
+
 
 class HTTPClient(six.with_metaclass(abc.ABCMeta, object)):
     def __init__(self, host='127.0.0.1', port=8500, scheme='http',
@@ -343,6 +353,7 @@ def __init__(
         self.query = Consul.Query(self)
         self.coordinate = Consul.Coordinate(self)
         self.operator = Consul.Operator(self)
+        self.snapshot = Consul.Snapshot(self)
 
     class Event(object):
         """
@@ -2434,3 +2445,22 @@ def raft_config(self):
             """
             return self.agent.http.get(
                 CB.json(), '/v1/operator/raft/configuration')
+
+    class Snapshot(object):
+        def __init__(self,agent):
+            self.agent = agent
+
+        def get(self):
+            """
+            Returns gzipped snapshot of current consul cluster
+            """
+            return self.agent.http.get(
+                CB.binary(),'/v1/snapshot')
+
+        def save(self, file_path):
+            """
+            Backup snapshot in a file
+            """
+            backup_file = open(file_path, 'w+b')
+            backup_file.write(self.get())
+            backup_file.close()
diff --git a/consul/std.py b/consul/std.py
@@ -14,7 +14,7 @@ def __init__(self, *args, **kwargs):
     def response(self, response):
         response.encoding = 'utf-8'
         return base.Response(
-            response.status_code, response.headers, response.text)
+            response.status_code, response.headers, response.text, response.content)
 
     def get(self, callback, path, params=None):
         uri = self.uri(path, params)

diff --git a/consul/tornado.py b/consul/tornado.py
@@ -16,7 +16,7 @@ def __init__(self, *args, **kwargs):
 
     def response(self, response):
         return base.Response(
-            response.code, response.headers, response.body.decode('utf-8'))
+            response.code, response.headers, response.body.decode('utf-8'), response.body)
 
     @gen.coroutine
     def _request(self, callback, request):

diff --git a/consul/twisted.py b/consul/twisted.py
@@ -47,8 +47,8 @@ def __init__(self, contextFactory, *args, **kwargs):
         self.client = TreqHTTPClient(Agent(**agent_kwargs))
 
     @staticmethod
-    def response(code, headers, text):
-        return base.Response(code, headers, text)
+    def response(code, headers, text, content):
+        return base.Response(code, headers, text, content)
 
     @staticmethod
     def compat_string(value):
@@ -70,7 +70,8 @@ def _get_resp(self, response):
             for k, v in dict(response.headers.getAllRawHeaders()).items()
         ])
         body = yield response.text(encoding='utf-8')
-        returnValue((response.code, headers, body))
+        content = yield response.content()
+        returnValue((response.code, headers, body, content))
 
     @inlineCallbacks
     def request(self, callback, method, url, **kwargs):