mirror of
https://github.com/kevin1024/vcrpy.git
synced 2025-12-08 16:53:23 +00:00
25 lines
700 B
Python
25 lines
700 B
Python
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)
|