import cdsapi
import sys
from datetime import datetime, timedelta
def retrieve_data(date, c):
year = date.strftime('%Y')
month = date.strftime('%m')
day = date.strftime('%d')
# Retrieve pressure levels data
c.retrieve(
'reanalysis-era5-pressure-levels',
{
'product_type': 'reanalysis',
'format': 'grib',
'variable': [
'geopotential', 'relative_humidity', 'specific_humidity', 'temperature',
'u_component_of_wind', 'v_component_of_wind', 'vertical_velocity'
],
'pressure_level': [
"50", "100", "150",
"200", "250", "300",
"350", "400", "450",
"500", "550", "600",
"650", "700", "750",
"800", "850", "900",
"925", "950", "975",
"1000"
],
'year': year,
'month': month,
'day': day,
'time': [
"00:00", "01:00", "02:00",
"03:00", "04:00", "05:00",
"06:00", "07:00", "08:00",
"09:00", "10:00", "11:00",
"12:00", "13:00", "14:00",
"15:00", "16:00", "17:00",
"18:00", "19:00", "20:00",
"21:00", "22:00", "23:00"
],
# 'area': [44, 26, 38, 32]
},
f'era5_pl_{year}{month}{day}.grib'
)
# Retrieve surface levels data
c.retrieve(
'reanalysis-era5-single-levels',
{
'product_type': 'reanalysis',
'variable': [
'10m_u_component_of_wind', '10m_v_component_of_wind', '2m_dewpoint_temperature',
'2m_temperature', 'geopotential', 'land_sea_mask', 'mean_sea_level_pressure', 'sea_ice_cover',
'sea_surface_temperature', 'snow_depth', 'soil_temperature_level_1', 'skin_temperature',
'soil_temperature_level_2', 'soil_temperature_level_3', 'soil_temperature_level_4', 'surface_pressure',
'volumetric_soil_water_layer_1', 'volumetric_soil_water_layer_2', 'volumetric_soil_water_layer_3', 'volumetric_soil_water_layer_4',
],
'year': year,
'month': month,
'day': day,
'time': [
"00:00", "01:00", "02:00",
"03:00", "04:00", "05:00",
"06:00", "07:00", "08:00",
"09:00", "10:00", "11:00",
"12:00", "13:00", "14:00",
"15:00", "16:00", "17:00",
"18:00", "19:00", "20:00",
"21:00", "22:00", "23:00"
],
'format': 'grib',
# 'area': [44, 26, 38, 32]
},
f'era5_sfc_{year}{month}{day}.grib'
)
def main():
if len(sys.argv) != 2:
print("Usage: script.py YYYYMMDD-YYYYMMDD")
sys.exit(1)
date_range = sys.argv[1]
start_date_str, end_date_str = date_range.split('-')
start_date = datetime.strptime(start_date_str, '%Y%m%d')
end_date = datetime.strptime(end_date_str, '%Y%m%d')
c = cdsapi.Client()
current_date = start_date
while current_date <= end_date:
retrieve_data(current_date, c)
current_date += timedelta(days=1)
if __name__ == "__main__":
main()
