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

Don't save cassette when it hasn't been modified

Browse Source
This commit is contained in:
shu zOMG chen
2013-08-21 14:27:29 -10:00
committed by Kevin McCarthy
parent 7264780960
commit b55834e929
3 changed files with 57 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
vcr/cassette.py
View File

@@ -25,6 +25,7 @@ class Cassette(object):
self._serializer = serializer
self.data = OrderedDict()
self.play_counts = Counter()
self.dirty = False
@property
def play_count(self):
@@ -47,22 +48,26 @@ class Cassette(object):
def append(self, request, response):
'''Add a request, response pair to this cassette'''
self.data[request] = response
self.dirty = True
def response_of(self, request):
'''Find the response corresponding to a request'''
return self.data[request]
def _save(self):
save_cassette(self._path, self._as_dict(), serializer=self._serializer)
def _as_dict(self):
return {"requests": self.requests, "responses": self.responses}
def _save(self, force=False):
if force or self.dirty:
save_cassette(self._path, self._as_dict(), serializer=self._serializer)
self.dirty = False
def _load(self):
try:
requests, responses = load_cassette(self._path, serializer=self._serializer)
for request, response in zip(requests, responses):
self.append(request, response)
self.dirty = False
except IOError:
pass