81 lines
2.2 KiB
Python
81 lines
2.2 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
# SPDX-FileCopyrightText: : 2017-2023 The PyPSA-Eur Authors
|
||
|
|
#
|
||
|
|
# SPDX-License-Identifier: MIT
|
||
|
|
"""
|
||
|
|
Created on Mon Jul 3 11:19:54 2023.
|
||
|
|
|
||
|
|
@author: fabian
|
||
|
|
"""
|
||
|
|
|
||
|
|
|
||
|
|
import logging
|
||
|
|
|
||
|
|
import pandas as pd
|
||
|
|
from entsoe import EntsoePandasClient
|
||
|
|
from entsoe.exceptions import NoMatchingDataError
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
carrier_grouper = {
|
||
|
|
"Waste": "Biomass",
|
||
|
|
"Hydro Pumped Storage": "Hydro",
|
||
|
|
"Hydro Water Reservoir": "Hydro",
|
||
|
|
"Hydro Run-of-river and poundage": "Run of River",
|
||
|
|
"Fossil Coal-derived gas": "Gas",
|
||
|
|
"Fossil Gas": "Gas",
|
||
|
|
"Fossil Oil": "Oil",
|
||
|
|
"Fossil Oil shale": "Oil",
|
||
|
|
"Fossil Brown coal/Lignite": "Lignite",
|
||
|
|
"Fossil Peat": "Lignite",
|
||
|
|
"Fossil Hard coal": "Coal",
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
if "snakemake" not in globals():
|
||
|
|
from _helpers import mock_snakemake
|
||
|
|
|
||
|
|
snakemake = mock_snakemake("retrieve_historical_electricity_generation")
|
||
|
|
|
||
|
|
|
||
|
|
api_key = "aeff3346-a240-40df-bd12-692772b845d0"
|
||
|
|
client = EntsoePandasClient(api_key=api_key)
|
||
|
|
|
||
|
|
start = pd.Timestamp(snakemake.params.snapshots["start"], tz="Europe/Brussels")
|
||
|
|
end = pd.Timestamp(snakemake.params.snapshots["end"], tz="Europe/Brussels")
|
||
|
|
|
||
|
|
countries = snakemake.params.countries
|
||
|
|
|
||
|
|
|
||
|
|
generation = []
|
||
|
|
unavailable_countries = []
|
||
|
|
|
||
|
|
for country in countries:
|
||
|
|
country_code = country
|
||
|
|
|
||
|
|
try:
|
||
|
|
gen = client.query_generation(country, start=start, end=end, nett=True)
|
||
|
|
gen = gen.tz_localize(None).resample("1h").mean()
|
||
|
|
gen = gen.rename(columns=carrier_grouper).groupby(level=0, axis=1).sum()
|
||
|
|
generation.append(gen)
|
||
|
|
except NoMatchingDataError:
|
||
|
|
unavailable_countries.append(country)
|
||
|
|
|
||
|
|
|
||
|
|
if unavailable_countries:
|
||
|
|
logger.warning(
|
||
|
|
f"Historical electricity production for countries {', '.join(unavailable_countries)} not available."
|
||
|
|
)
|
||
|
|
|
||
|
|
keys = [c for c in countries if c not in unavailable_countries]
|
||
|
|
generation = pd.concat(generation, keys=keys, axis=1)
|
||
|
|
generation = generation.loc[start.tz_localize(None) : end.tz_localize(None)]
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
# SPDX-FileCopyrightText: : 2017-2023 The PyPSA-Eur Authors
|
||
|
|
#
|
||
|
|
# SPDX-License-Identifier: MIT
|
||
|
|
generation.to_csv(snakemake.output[0])
|