graphPaname/app/preprocessing.py

from json import load
from pandas import json_normalize, DataFrame
from .constants import FILES, COLUMNS


def open_json(dataset) -> dict:
    """
    Loads a dictionary with data from a JSON file
    """
    with open(FILES[dataset]) as f:
        json = load(f)
    return json


def create_dataframe(dataset) -> DataFrame:
    """
    Creates a DataFrame from a JSON file
    """
    json = open_json(dataset)
    df = json_normalize(data=json, record_path=["records"], errors="ignore",)
    filtered_df = df.filter(items=COLUMNS[dataset])
    return filtered_df