python下载指定时段的gdas1数据

import sys
import os
import datetime
from urllib.request import urlretrieve

def generate_gdas1_urls(start_date, end_date):
    """生成指定日期范围内的GDAS1数据文件URL列表"""
    base_url = "https://data.rda.ucar.edu/ds083.3/{year}/{year}{month:02d}/gdas1.fnl0p25.{year}{month:02d}{day:02d}{hour:02d}.f00.grib2"
    urls = []
    current_date = start_date
    while current_date <= end_date:
        # 假设每6小时一个文件，分别对应00, 06, 12, 18小时的文件
        for hour in [0, 6, 12, 18]:  
            url = base_url.format(year=current_date.year, month=current_date.month, day=current_date.day, hour=hour)
            urls.append(url)
            print(url)
        current_date += datetime.timedelta(days=1)  # 每天一个日期
    return urls

def download_files(urls):
    """下载文件列表中的所有文件"""
    for url in urls:
        filename = os.path.basename(url)
        print(f"Downloading {filename} ... ", end='', flush=True)
        try:
            urlretrieve(url, filename)
            print("done")
        except Exception as e:
            print(f"Failed: {e}")

if __name__ == "__main__":
    start_date = datetime.datetime(2021, 9, 3)  # 下载开始日期
    end_date = datetime.datetime(2021, 9, 4)    # 下载结束日期
    
    file_urls = generate_gdas1_urls(start_date, end_date)
    download_files(file_urls)

import xarray as xr 
ds = xr.open_dataset('/home/mw/project/gdas1.fnl0p25.2021090300.f00.grib2',engine='pynio')
ds