mirror of
https://github.com/kevin1024/vcrpy.git
synced 2025-12-10 01:25:34 +00:00
27 lines
704 B
Python
27 lines
704 B
Python
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), Loader=Loader)
|
|
cassette = Cassette(pc)
|
|
return cassette
|
|
except IOError:
|
|
return None
|
|
|
|
|
|
def save_cassette(cassette_path, cassette):
|
|
dirname, filename = os.path.split(cassette_path)
|
|
if not os.path.exists(dirname):
|
|
os.makedirs(dirname)
|
|
with open(cassette_path, 'a') as cassette_file:
|
|
cassette_file.write(yaml.dump(cassette.serialize(), Dumper=Dumper))
|