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

Add before_record_response to Cassette and VCR.

Browse Source
This commit is contained in:
Ivan Malison
2014-09-17 04:10:05 -07:00
parent 366e2b75bb
commit 9a4f5f23a4
4 changed files with 21 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
vcr/cassette.py
View File

@@ -76,6 +76,7 @@ class Cassette(object):
filter_headers=(),
filter_query_parameters=(),
before_record=None,
before_record_response=None,
ignore_hosts=(),
ignore_localhost=()
):
@@ -85,6 +86,7 @@ class Cassette(object):
self._filter_headers = filter_headers
self._filter_query_parameters = filter_query_parameters
self._before_record = before_record
self._before_record_response = before_record_response
self._ignore_hosts = ignore_hosts
if ignore_localhost:
self._ignore_hosts = list(set(
@@ -136,6 +138,8 @@ class Cassette(object):
request = self._filter_request(request)
if not request:
return
if self._before_record_response:
response = self._before_record_response(response)
self.data.append((request, response))
self.dirty = True