import cPickle as pickle
import os

def write(dfs_output_dict, output_folder, output_format):
    """
    Execute the writing process to the pickle file and normal output.

    @param dfs_output_dict : Output dictionary, with dictionary of pensum progress for retention a list of dictionaries for forecast.
    @param output_folder   : Folder name of output files
    @param output_format   : Output extension file
    """
    for data in dfs_output_dict:
        if data == "output_data":
            filename = "pensumProgressData.pkl"
            filepath = os.path.join(output_folder, filename)
            pickle.dump(dfs_output_dict[data], open(filepath, "wb"))
        elif output_format == "csv":
            write_csv(output_folder, dfs_output_dict[data], data)
        elif output_format == "pb":
            write_pb(output_folder, dfs_output_dict[data], data)
        else:
            write_csv(output_folder, dfs_output_dict[data], data)
            write_pb(output_folder, dfs_output_dict[data], data)
    return