diff --git a/vcr/files.py b/vcr/files.py index 7e036c0..3864ccf 100644 --- a/vcr/files.py +++ b/vcr/files.py @@ -1,17 +1,18 @@ +from __future__ import with_statement import os import yaml + def load_cassette(cassette_path): try: return yaml.load(open(cassette_path)) except IOError: return None -def save_cassette(cassette_path,cassette): - dirname,filename = os.path.split(cassette_path) + +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,'wc') as cassette_file: + with open(cassette_path, 'wc') as cassette_file: cassette_file.write(cassette.serialize()) - -