Add orinal response field to asyncio response status

pubnub · Mar 20, 2017 · 037a682 · 037a682
1 parent 8c079cf
commit 037a682
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/pubnub/pubnub_asyncio.py b/pubnub/pubnub_asyncio.py
@@ -180,7 +180,8 @@ def _request_helper(self, options_func, cancellation_event):
                 origin=request_url.hostname,
                 uuid=uuid,
                 auth_key=auth_key,
-                client_request=None
+                client_request=None,
+                client_response=response
             )
 
         if body is not None and len(body) > 0:

diff --git a/pubnub/structures.py b/pubnub/structures.py
@@ -66,13 +66,14 @@ def __init__(self, headers, pn_config):
 
 
 class ResponseInfo(object):
-    def __init__(self, status_code, tls_enabled, origin, uuid, auth_key, client_request):
+    def __init__(self, status_code, tls_enabled, origin, uuid, auth_key, client_request, client_response=None):
         self.status_code = status_code
         self.tls_enabled = tls_enabled
         self.origin = origin
         self.uuid = uuid
         self.auth_key = auth_key
         self.client_request = client_request
+        self.client_response = client_response
 
 
 class Envelope(object):