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

make config options actually do stuff

Browse Source
This commit is contained in:
Kevin McCarthy
2013-08-19 19:12:32 -10:00
parent 9073cf137e
commit 1478ce82fd
4 changed files with 35 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
vcr/persist.py
View File

@@ -1,21 +1,8 @@
from .persisters.filesystem import FilesystemPersister
from .serializers.yamlserializer import YamlSerializer
from .serializers.jsonserializer import JSONSerializer
def _get_serializer_cls(serializer):
serializer_cls = {
'yaml': YamlSerializer,
'json': JSONSerializer,
}.get(serializer)
if not serializer_cls:
raise ImportError('Invalid serializer %s' % serializer)
return serializer_cls
def load_cassette(cassette_path, serializer):
serializer_cls = _get_serializer_cls(serializer)
return serializer_cls.load(cassette_path)
return serializer.load(cassette_path)
def save_cassette(cassette_path, requests, responses, serializer):
serializer_cls = _get_serializer_cls(serializer)
data = serializer_cls.dumps(requests, responses)
data = serializer.dumps(requests, responses)
FilesystemPersister.write(cassette_path, data)