bjoern/pypsa-eur
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Merge branch 'energy_totals' of https://github.com/PyPSA/pypsa-eur into energy_totals

Browse Source
This commit is contained in:
toniseibold 2024-02-27 14:32:21 +01:00
commit ae187b8ba3
3 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
rules/retrieve.smk
View File

@ -160,7 +160,9 @@ if config["enable"]["retrieve"] and config["enable"].get(
rule retrieve_eurostat_data:
output:
protected(
directory("data/bundle-sector/eurostat-energy_balances-april_2023_edition")
directory(
"data/bundle-sector/eurostat-energy_balances-april_2023_edition"
)
),
log:
"logs/retrieve_eurostat_data.log",
@ -168,6 +170,7 @@ if config["enable"]["retrieve"] and config["enable"].get(
script:
"../scripts/retrieve_eurostat_data.py"
if config["enable"]["retrieve"]:
datafiles = [
"IGGIELGN_LNGs.geojson",

12
scripts/retrieve_eurostat_data.py
View File

@ -11,11 +11,7 @@ import logging
import zipfile
from pathlib import Path
from _helpers import (
configure_logging,
progress_retrieve,
set_scenario_config,
)
from _helpers import configure_logging, progress_retrieve, set_scenario_config
logger = logging.getLogger(__name__)
@ -33,13 +29,15 @@ if __name__ == "__main__":
disable_progress = snakemake.config["run"].get("disable_progressbar", False)
url_eurostat = "https://ec.europa.eu/eurostat/documents/38154/4956218/Balances-December2022.zip/f7cf0d19-5c0f-60ad-4e48-098a5ddd6e48?t=1671184070589"
tarball_fn = Path(f"{rootpath}/data/bundle-sector/eurostat_2023.zip")
to_fn = Path(f"{rootpath}/data/bundle-sector/eurostat-energy_balances-april_2023_edition/")
to_fn = Path(
f"{rootpath}/data/bundle-sector/eurostat-energy_balances-april_2023_edition/"
)
logger.info(f"Downloading Eurostat data from '{url_eurostat}'.")
progress_retrieve(url_eurostat, tarball_fn, disable=disable_progress)
logger.info("Extracting Eurostat data.")
with zipfile.ZipFile(tarball_fn, 'r') as zip_ref:
with zipfile.ZipFile(tarball_fn, "r") as zip_ref:
zip_ref.extractall(to_fn)
logger.info(f"Eurostat data available in '{to_fn}'.")