1
0
mirror of https://github.com/kevin1024/vcrpy.git synced 2025-12-09 09:13:23 +00:00

Added checkfor old cassette on load cassete to persist module

This commit is contained in:
Max Shytikov
2014-04-24 02:03:56 +02:00
parent 9c9612f93e
commit 0408bdaadb

View File

@@ -1,9 +1,31 @@
import tempfile
from .persisters.filesystem import FilesystemPersister
from . import migration
def _check_for_old_cassette(cassette_content):
# crate tmp with cassete content and try to migrate it
with tempfile.NamedTemporaryFile(mode='w+') as f:
f.write(cassette_content)
f.seek(0)
if migration.try_migrate(f.name):
raise ValueError(
"Your cassette files were generated in an older version "
"of VCR. Delete your cassettes or run the migration script."
"See http://git.io/mHhLBg for more details."
)
def load_cassette(cassette_path, serializer):
with open(cassette_path) as f:
return serializer.deserialize(f.read())
cassette_content = f.read()
try:
cassette = serializer.deserialize(cassette_content)
except TypeError:
_check_for_old_cassette(cassette_content)
raise
return cassette
def save_cassette(cassette_path, cassette_dict, serializer):