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

Use the libYAML version of yaml if it's available

Browse Source
This commit is contained in:
Dan Lecocq
2013-07-23 15:25:33 -07:00
parent 5c1ccd9dd5
commit e9cb0bff47
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
vcr/files.py
View File

@@ -2,10 +2,16 @@ import os
import yaml
from .cassette import Cassette
# Use the libYAML versions if possible
try:
from yaml import CLoader as Loader, CDumper as Dumper
except ImportError:
from yaml import Loader, Dumper
def load_cassette(cassette_path):
try:
pc = yaml.load(open(cassette_path))
pc = yaml.load(open(cassette_path), Loader=Loader)
cassette = Cassette(pc)
return cassette
except IOError:
@@ -17,4 +23,4 @@ def save_cassette(cassette_path, cassette):
if not os.path.exists(dirname):
os.makedirs(dirname)
with open(cassette_path, 'a') as cassette_file:
cassette_file.write(yaml.dump(cassette.serialize()))
cassette_file.write(yaml.dump(cassette.serialize(), Dumper=Dumper))