utils.py

import os, pickle, shutil

def summaries_to_rouge_format(summaries_file, outputs_dir, extention):
    if not os.path.exists(outputs_dir):
        os.makedirs(outputs_dir)

    with open(summaries_file, "r") as source: 
        lines = source.readlines()
        for i, line in enumerate(lines):
            with open(outputs_dir + str(i) + "." + extention, "w") as destination:
                destination.write(line)

def read_pickle(file):
    with open(file, 'rb') as f:
        data = pickle.loads(f.read())
    return data

def write_pickle(file, data):
    with open(file, 'wb') as f:
        pickle.dump(data, f, protocol=2)

def delete_dir(directory):
    shutil.rmtree(directory)