Common Python File I/O Tool

directory.py

import os
import shutil

def create_dir(fp: str):
    if not os.path.exists(fp):
        os.mkdir(fp)


def remove_dir(fp: str):
    if os.path.exists(fp):
        shutil.rmtree(fp)

json.py

import json

def read_json(fp: str):
    return json.load(open(fp, 'r', encoding='utf-8'))


def to_json(data, fp: str):
    json.dump(data, open(fp, 'w', encoding='utf-8'), indent=4, ensure_ascii=False)

pickle.py

import pickle


def read_pickle(fp: str):
    with open(fp, 'rb') as f:
        return pickle.load(f)


def to_pickle(data, fp: str):
    with open(fp, 'wb') as f:
        pickle.dump(data, f)