entsoe_api.py

#%% 
# # Extracting data from ENTSO-E Transparency Platform's Restful API
# https://github.com/EnergieID/entsoe-py

#%%
# import libraries
from entsoe import EntsoePandasClient
from entsoe.mappings import DOMAIN_MAPPINGS, BIDDING_ZONES
import pandas as pd
import os, errno

# import security token saved in a separate file (login.py)
from login import token

#%%
# combine domain and bidding zone keys and values into the DOMAIN_MAPPINGS dictionary
DOMAIN_MAPPINGS.update(BIDDING_ZONES)

#%%
# use security token to access the api through the entsoe pandas client
client = EntsoePandasClient(api_key=token)

#%%
# define attributes for the data to be extracted
# timestamps for the first half of 2019
start = pd.Timestamp('20190101', tz='Europe/Brussels') 
end = pd.Timestamp('20190601', tz='Europe/Brussels')

#%%
# list of bidding zones in the North Sea region
bz_ns = ['BE', 'DE-LU', 'DK-1', 'DK-2', # Belgium, Germany + Luxembourg, Denmark 
          'FR', 'GB', 'IE-SEM', 'NL', # France, United Kingdom (Great Britain + (Northern) Ireland), Netherlands
          'NO-1', 'NO-2', 'NO-3', 'NO-4', 'NO-5', # Norway
          'SE-1', 'SE-2', 'SE-3', 'SE-4'] # Sweden

#%%
# create a directory to store files if it does not exist
path = 'data/entsoe_api'
try:
    os.makedirs(path)
except OSError as exception:
    if exception.errno != errno.EEXIST:
        raise
    else:
        print ('\nBE CAREFUL! Directory %s already exists.' % path)

#%%
# extracting data for each bidding zone in a loop
for country_code in bz_ns:
  # create a pandas dataframe for the generation data with the above attributes
  # generation per production type
  ts_gen = client.query_generation(country_code, start=start, end=end, psr_type=None)
  # save as csv 
  ts_gen.to_csv('data/entsoe_api/generation_'+str(country_code)+'.csv')

  # load
  ts_load = client.query_load(country_code, start=start, end=end)
  # save as csv
  ts_load.to_csv('data/entsoe_api/load_'+str(country_code)+'.csv', header=['Load'])

  # installed generation capacity per unit
  ts_cap = client.query_installed_generation_capacity_per_unit(country_code, start=start, end=end, psr_type=None)
  ts_cap.to_csv('data/entsoe_api/installed_capacity_'+str(country_code)+'.csv')