in generateDefaultForecast/generateDefaultForecast.py [0:0]
def onEventHandler(event, context):
response = forecast_client.list_dataset_groups()
for datasetGroup in response["DatasetGroups"]:
datasetGroupName=datasetGroup["DatasetGroupName"]
defaultPredictorName=datasetGroupName+"_Predictor"
defaultForecastName=datasetGroupName+"_Forecast"
defaultPredictorArn=getPredictorArnByName(forecast_client, datasetGroup["DatasetGroupArn"], defaultPredictorName)
if (defaultPredictorArn is None ):
logger.info("For DatasetGroup="+datasetGroupName+" , default predictor="+defaultPredictorName + " is not trained yet or hasn't finished training, skip")
continue
if(isForcastExistInDataSetGroup(forecast_client,datasetGroup["DatasetGroupArn"],defaultForecastName)):
logger.info("For DatasetGroup="+datasetGroupName+" , default predictor="+defaultPredictorName + ", default forecast="+ defaultForecastName + " already exist, skip")
continue
createForecast(forecast_client,defaultForecastName,defaultPredictorArn)