common.py

import json

import dill


def intsec(list1: list, list2: list) -> list:
    """Simple intesection of two lists.

    Args:
        list1 (list): list1
        list2 (list): list2

    Returns:
        list (list): intersection of lists
    """
    return list(set.intersection(set(list1), set(list2)))


def diff(list1: list, list2: list) -> list:
    """Simple difference of items in list2 from list1.

    Args:
        list1 (list): list1
        list2 (list): list2

    Returns:
        list (list): difference of lists
    """
    return list(set(list1).difference(set(list2)))


def json_load(file_loc: str):
    """Helper function to open/close json file,
    otherwise the python outputs warning that the file remains opened

    Args:
        file_loc (str): location of the file

    Returns:
        content (dict): content of json file, usually dictionary
    """
    with open(file_loc, "rb") as f:
        content = json.load(f)
    return content


def json_dump(file_loc: str, content: object):
    """Helper function to open/close json file and dump content into it,
    otherwise the python outputs warning that the file remains opened

    Args:
        file_loc (str): location of the file
        content (object): data that will be saved to json, usually dictionary
    """
    with open(file_loc, "wb") as f:
        json.dump(content, f)


def dill_load(file_loc: str):
    """Helper function to open/close dill file,
    otherwise the python outputs warning that the file remains opened

    Args:
        file_loc (str): location of the file

    Returns:
        content (dict): content of dill file, usually dictionary
    """
    with open(file_loc, "rb") as f:
        content = dill.load(f)
    return content


def dill_dump(file_loc: str, content: object):
    """Helper function to open/close dill file and dump content into it,
    otherwise the python outputs warning that the file remains opened

    Args:
        file_loc (str): location of the file
        content (object): data that will be saved to dill, usually dictionary
    """
    with open(file_loc, "wb") as f:
        dill.dump(content, f)