Source code for cookiecutter.replay

# -*- coding: utf-8 -*-

"""
cookiecutter.replay.

-------------------
"""

from __future__ import unicode_literals

import json
import os

import six

from cookiecutter.utils import make_sure_path_exists


[docs]def get_file_name(replay_dir, template_name): """Get the name of file.""" file_name = '{}.json'.format(template_name) return os.path.join(replay_dir, file_name)
[docs]def dump(replay_dir, template_name, context): """Write json data to file.""" if not make_sure_path_exists(replay_dir): raise IOError('Unable to create replay dir at {}'.format(replay_dir)) if not isinstance(template_name, six.string_types): raise TypeError('Template name is required to be of type str') if not isinstance(context, dict): raise TypeError('Context is required to be of type dict') if 'cookiecutter' not in context: raise ValueError('Context is required to contain a cookiecutter key') replay_file = get_file_name(replay_dir, template_name) with open(replay_file, 'w') as outfile: json.dump(context, outfile, indent=2)
[docs]def load(replay_dir, template_name): """Read json data from file.""" if not isinstance(template_name, six.string_types): raise TypeError('Template name is required to be of type str') replay_file = get_file_name(replay_dir, template_name) with open(replay_file, 'r') as infile: context = json.load(infile) if 'cookiecutter' not in context: raise ValueError('Context is required to contain a cookiecutter key') return context