def load_datasets():
    """
    Loads ECMWF prediction datasets from data store for all watersheds
    """
    session = SettingsSessionMaker()
    main_settings = session.query(MainSettings).order_by(
        MainSettings.id).first()

    ecmwf_rapid_prediction_directory = main_settings.ecmwf_rapid_prediction_directory
    if ecmwf_rapid_prediction_directory and os.path.exists(
            ecmwf_rapid_prediction_directory):
        for watershed in session.query(Watershed).all():
            download_single_watershed_ecmwf_data(
                watershed, ecmwf_rapid_prediction_directory)
    else:
        print "ECMWF prediction location invalid. Please set to continue."

    wrf_hydro_rapid_prediction_directory = main_settings.wrf_hydro_rapid_prediction_directory
    if wrf_hydro_rapid_prediction_directory and \
        os.path.exists(wrf_hydro_rapid_prediction_directory):
        for watershed in session.query(Watershed).all():
            download_single_watershed_wrf_hydro_data(
                watershed, wrf_hydro_rapid_prediction_directory)
    else:
        print "WRF-Hydro prediction location invalid. Please set to continue."
def load_datasets():
    """
    Loads ECMWF prediction datasets from data store for all watersheds
    """
    session = SettingsSessionMaker()
    main_settings  = session.query(MainSettings).order_by(MainSettings.id).first()

    ecmwf_rapid_prediction_directory = main_settings.ecmwf_rapid_prediction_directory
    if ecmwf_rapid_prediction_directory and os.path.exists(ecmwf_rapid_prediction_directory):
        for watershed in session.query(Watershed).all():
            download_single_watershed_ecmwf_data(watershed, ecmwf_rapid_prediction_directory)
    else:
        print "ECMWF prediction location invalid. Please set to continue."
        
    wrf_hydro_rapid_prediction_directory = main_settings.wrf_hydro_rapid_prediction_directory
    if wrf_hydro_rapid_prediction_directory and \
        os.path.exists(wrf_hydro_rapid_prediction_directory):
        for watershed in session.query(Watershed).all():
            download_single_watershed_wrf_hydro_data(watershed, 
                                                     wrf_hydro_rapid_prediction_directory)
    else:
        print "WRF-Hydro prediction location invalid. Please set to continue."
def load_watershed(watershed):
    """
    Loads prediction datasets from data store for one watershed
    """
    session = SettingsSessionMaker()
    main_settings  = session.query(MainSettings).order_by(MainSettings.id).first()

    if main_settings.ecmwf_rapid_prediction_directory and \
        os.path.exists(main_settings.ecmwf_rapid_prediction_directory):
            
        download_single_watershed_ecmwf_data(watershed, 
                                             main_settings.ecmwf_rapid_prediction_directory)
    else:
        print "ECMWF prediction location invalid. Please set to continue."

    if main_settings.wrf_hydro_rapid_prediction_directory and \
        os.path.exists(main_settings.wrf_hydro_rapid_prediction_directory):
            
        download_single_watershed_wrf_hydro_data(watershed, 
                                                 main_settings.wrf_hydro_rapid_prediction_directory)
    else:
        print "WRF-Hydro prediction location invalid. Please set to continue."
def load_watershed(watershed):
    """
    Loads prediction datasets from data store for one watershed
    """
    session = SettingsSessionMaker()
    main_settings = session.query(MainSettings).order_by(
        MainSettings.id).first()

    if main_settings.ecmwf_rapid_prediction_directory and \
        os.path.exists(main_settings.ecmwf_rapid_prediction_directory):

        download_single_watershed_ecmwf_data(
            watershed, main_settings.ecmwf_rapid_prediction_directory)
    else:
        print "ECMWF prediction location invalid. Please set to continue."

    if main_settings.wrf_hydro_rapid_prediction_directory and \
        os.path.exists(main_settings.wrf_hydro_rapid_prediction_directory):

        download_single_watershed_wrf_hydro_data(
            watershed, main_settings.wrf_hydro_rapid_prediction_directory)
    else:
        print "WRF-Hydro prediction location invalid. Please set to continue."