def generate_script(self, files): self.__prepare_output_folder() mock_files = {} output_file_name = os.path.join(self.work_dir, self.__MOCK_SCRIPT_NAME) # generate file names and content for k, v in files.items(): mock_files.update(self.__create_mock_file(k, v, write_files=False)) # add files to script template = read_template('mock_script_template.py') output = template.render(mocks=pickle.dumps(mock_files)) write_all_text(output_file_name, output)
def __init__(self, output_folder): self.output_folder = output_folder self.template = read_template('stubs_template.py')