'''

@Author: your name

@Date: 2020-01-06 17:08:52

@LastEditTime : 2020-01-07 13:25:31

@LastEditors  : Please set LastEditors

@Description: In User Settings Edit

@FilePath: /KGCN_Keras-master/utils/io.py

'''

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

import os

import json

import pickle

def pickle_load(filename: str):

    try:

        with open(filename, 'rb') as f:

            obj = pickle.load(f)

        print(f'Logging Info - Loaded: {filename}')

    except EOFError:

        print(f'Logging Error - Cannot load: {filename}')

        obj = None

    return obj

def pickle_dump(filename: str, obj):

    with open(filename, 'wb') as f:

        pickle.dump(obj, f)

    print(f'Logging Info - Saved: {filename}')

def write_log(filename: str, log, mode='w'):

    with open(filename, mode) as writers:

        writers.write('\n')

        json.dump(log, writers, indent=4, ensure_ascii=False)

def format_filename(_dir: str, filename_template: str, **kwargs):

    """Obtain the filename of data base on the provided template and parameters"""

    filename = os.path.join(_dir, filename_template.format(**kwargs))

    return filename