Source code for pyrfu.maven.db_init

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import json

# Built-in imports
import os

__author__ = "Louis Richard"
__email__ = "louisr@irfu.se"
__copyright__ = "Copyright 2020-2023"
__license__ = "MIT"
__version__ = "2.4.10"
__status__ = "Prototype"


def db_init(local_data_dir):
    r"""Setup the default path of MAVEN data.

    Parameters
    ----------
    local_data_dir : str
        Path to the data.

    """

    # Normalize the path and make sure that it exists
    local_data_dir = os.path.normpath(local_data_dir)
    assert os.path.exists(local_data_dir), f"{local_data_dir} doesn't exists!!"

    # Path to the configuration file.
    pkg_path = os.path.dirname(os.path.abspath(__file__))

    # Read the current version of the configuration
    with open(os.path.join(pkg_path, "config.json"), "r", encoding="utf-8") as fs:
        config = json.load(fs)

    # Overwrite the configuration file with the new path
    with open(os.path.join(pkg_path, "config.json"), "w", encoding="utf-8") as fs:
        config["local_data_dir"] = local_data_dir
        json.dump(config, fs)