openego / eGon-data

from egon.data.datasets import Dataset

def download():

    """

    Download small scale input data from Zenodo

    Parameters

    ----------

    """

    data_bundle_path = Path(".") / "data_bundle_egon_data"

    # Delete folder if it already exists

    if data_bundle_path.exists() and data_bundle_path.is_dir():

        shutil.rmtree(data_bundle_path)

    # Get parameters from config and set download URL

    sources = config.datasets()["data-bundle"]["sources"]["zenodo"]

    url = (

        f"https://zenodo.org/record/{sources['deposit_id']}/files/"

        "data_bundle_egon_data.zip"

    )

    target_file = config.datasets()["data-bundle"]["targets"]["file"]

    # check if file exists

    if not Path(target_file).exists():

        # Retrieve files

        urlretrieve(url, target_file)

    with zipfile.ZipFile(target_file, "r") as zip_ref:

        zip_ref.extractall(".")

class DataBundle(Dataset):

    return values

def download_pypsa_technology_data():

    """Downlad PyPSA technology data results."""

    data_path = Path(".") / "pypsa_technology_data"

    # Delete folder if it already exists

    if data_path.exists() and data_path.is_dir():

        shutil.rmtree(data_path)

    # Get parameters from config and set download URL

    sources = egon.data.config.datasets()["pypsa-technology-data"]["sources"][

        "zenodo"

    ]

    url = f"""https://zenodo.org/record/{sources['deposit_id']}/files/{sources['file']}"""

    target_file = egon.data.config.datasets()["pypsa-technology-data"][

        "targets"

    ]["file"]

    # Retrieve files

    urlretrieve(url, target_file)

    with zipfile.ZipFile(target_file, "r") as zip_ref:

        zip_ref.extractall(".")

class ScenarioParameters(Dataset):