from vcr.request import Request
try:
    import simplejson as json
except ImportError:
    import json

def _json_default(obj):
    if isinstance(obj, frozenset):
        return dict(obj)
    return obj

def _fix_response_unicode(d):
		d['body']['string'] = d['body']['string'].encode('utf-8')
		return d

class JSONSerializer(object):
    @classmethod
    def load(cls, cassette_path):
        with open(cassette_path) as fh:
            data = json.load(fh)
        requests = [Request._from_dict(r['request']) for r in data]
        responses = [_fix_response_unicode(r['response']) for r in data]
        return requests, responses

    @classmethod
    def dumps(cls, requests, responses):
        data = ([{
            'request': request._to_dict(),
            'response': response,
        } for request, response in zip(requests, responses)])
        return json.dumps(data, indent=4, default=_json_default)