Mirroring/vcrpy
1
0
Fork 0
mirror of https://github.com/kevin1024/vcrpy.git synced 2025-12-08 16:53:23 +00:00
Code Releases Activity

support python 3.4

Browse Source
This commit is contained in:
Olutobi Owoputi
2015-12-02 14:43:06 -08:00
parent 2d08358b5c
commit 5d866dd77c
3 changed files with 29 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
vcr/filters.py
View File

@@ -1,8 +1,11 @@
from six import BytesIO, text_type
from six.moves.urllib.parse import urlparse, urlencode, urlunparse
import copy
import json
import zlib
from .util import CaseInsensitiveDict
def replace_headers(request, replacements):
"""
@@ -130,8 +133,8 @@ def decode_response(response):
2. delete the content-encoding header
3. update content-length header to decompressed length
"""
def is_compressed(response):
encoding = response['headers'].get('content-encoding', [])
def is_compressed(headers):
encoding = headers.get('content-encoding', [])
return encoding and encoding[0] in ('gzip', 'deflate')
def decompress_body(body, encoding):
@@ -143,15 +146,16 @@ def decode_response(response):
else: # encoding == 'deflate'
return zlib.decompress(body)
if is_compressed(response):
response = response.copy()
encoding = response['headers']['content-encoding'][0]
response['headers']['content-encoding'].remove(encoding)
if not response['headers']['content-encoding']:
del response['headers']['content-encoding']
headers = CaseInsensitiveDict(response['headers'])
if is_compressed(headers):
response = copy.deepcopy(response)
encoding = headers['content-encoding'][0]
headers['content-encoding'].remove(encoding)
if not headers['content-encoding']:
del headers['content-encoding']
new_body = decompress_body(response['body']['string'], encoding)
response['body']['string'] = new_body
response['content-length'] = len(new_body)
headers['content-length'] = [str(len(new_body))]
response['headers'] = dict(headers)
return response