mirror of
https://github.com/kevin1024/vcrpy.git
synced 2025-12-09 09:13:23 +00:00
Adding json serializer
This commit is contained in:
committed by
Kevin McCarthy
parent
3e247a2efb
commit
51f0f1bacd
24
vcr/serializers/yamlserializer.py
Normal file
24
vcr/serializers/yamlserializer.py
Normal file
@@ -0,0 +1,24 @@
|
||||
import yaml
|
||||
|
||||
# Use the libYAML versions if possible
|
||||
try:
|
||||
from yaml import CLoader as Loader, CDumper as Dumper
|
||||
except ImportError:
|
||||
from yaml import Loader, Dumper
|
||||
|
||||
|
||||
class YamlSerializer(object):
|
||||
@classmethod
|
||||
def load(cls, cassette_path):
|
||||
data = yaml.load(open(cassette_path), Loader=Loader)
|
||||
requests = [r['request'] for r in data]
|
||||
responses = [r['response'] for r in data]
|
||||
return requests, responses
|
||||
|
||||
@classmethod
|
||||
def dumps(cls, requests, responses):
|
||||
data = ([{
|
||||
'request': request,
|
||||
'response': response,
|
||||
} for request, response in zip(requests, responses)])
|
||||
return yaml.dump(data, Dumper=Dumper)
|
||||
Reference in New Issue
Block a user