##########################################################################
# NSAp - Copyright (C) CEA, 2019
# Distributed under the terms of the CeCILL-B license, as published by
# the CEA-CNRS-INRIA. Refer to the LICENSE file or to
# http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
# for details.
##########################################################################

"""
This module defines the JSON dataset loader.
"""

# Third party import
import json

# Package import
from .loader_base import LoaderBase



[docs]
class JSON(LoaderBase):
    """ Define the JSON loader.
    """
    allowed_extensions = [".json", ]


[docs]
    def load(self, path):
        """ A method that loads a JSON file.

        Parameters
        ----------
        path: str
            the path to the JSON file.

        Returns
        -------
        data: object
            the loaded data.
        """
        with open(path, "rt") as open_file:
            data = json.load(open_file)
        return data



[docs]
    def save(self, data, path):
        """ A method that saves data into a JSON file.

        Parameters
        ----------
        data: object
            the data to be saved.
        path: str
            the path to the JSON file to save data into.
        """
        with open(path, "wt") as open_file:
            json.dump(data, open_file, indent=4)