Mapping actions to functions

python-function.py

ACTIONS_MAPPING = {
    "GET_ACCIDENT_DATA": get_accident_data,
    "GET_ACCIDENT_DETAILS": get_accident_details,
    "GET_CLUSTERS": get_clusters,
    "GET_CLUSTERS_DETAILS": get_cluster_details,
    "SAVE_CLUSTERS_RESULTS": save_clusters_results,
    "GET_PREDICTIONS": get_predictions,
    "GET_MODELS": get_models,
    "GET_PREDICT_INPUT_FILES": get_predict_input_files,
}

def score(payload):
    req = payload["input_data"][0]["values"][0]
    action = req["ACTION"]
    params = req["PARAMS"]
    if action in ACTIONS_MAPPING:
        df = ACTIONS_MAPPING[action](params)
        return {"predictions": [{"fields": df.columns.tolist(), "values": df.values.tolist()}]}
    return {"predictions": [{"values": ["NO ACTION"]}]}