Function DAGAddScriptRun

plugins/python/redisgears_python.c:2488–2576

}

static PyObject* DAGAddScriptRun(PyObject *self, PyObject *args, PyObject *kargs) {
    verifyRedisAILoaded();
    PyDAGRunner *pyDag = (PyDAGRunner *)self;
    if(!_IsDagAPISupported(RedisAI_DAGCreateScriptRunOp) || !_IsValidDag(pyDag)) {
        return NULL;
    }

    PyObject* scriptName = GearsPyDict_GetItemString(kargs, "name");
    if (!scriptName) {
        PyErr_SetString(GearsError, "Script key name was not given");
        return NULL;
    }
    if(!PyUnicode_Check(scriptName)){
        PyErr_SetString(GearsError, "Script name argument must be a string");
        return NULL;
    }
    PyObject* funcName = GearsPyDict_GetItemString(kargs, "func");
    if (!funcName) {
        PyErr_SetString(GearsError, "Function name was not given");
        return NULL;
    }
    if(!PyUnicode_Check(funcName)){
        PyErr_SetString(GearsError, "Function name argument must be a string");
        return NULL;
    }
    const char* scriptNameStr = PyUnicode_AsUTF8AndSize(scriptName, NULL);
    const char* functionNameStr = PyUnicode_AsUTF8AndSize(funcName, NULL);

    // Create SCRIPTRUN op after bringing script from keyspace (raise an exception if it does not exist)
    RAI_Error *err;
    RedisAI_InitError(&err);
    RAI_DAGRunOp *scriptRunOp = _createScriptRunOp(scriptNameStr, functionNameStr, err);
    if (scriptRunOp == NULL) {
        RedisAI_FreeError(err);
        return NULL;
    }
    PyObject *ret = NULL;

    // Add to the scriptRun op with its inputs and output keys and insert it to the DAG
    PyObject* scriptInputs = GearsPyDict_GetItemString(kargs, "inputs");
    PyObject* inputsIter = NULL;
    PyObject* scriptOutputs = NULL;
    PyObject* outputsIter = NULL;
    if (scriptInputs) {
        inputsIter = PyObject_GetIter(scriptInputs);
        if(!inputsIter) {
            PyErr_SetString(GearsError, "Script inputs must be iterable");
            goto cleanup;
        }
        PyObject* input;
        while((input = (PyObject*)PyIter_Next(inputsIter)) != NULL) {
            if(!PyUnicode_Check(input)){
                PyErr_SetString(GearsError, "Input name must be a string");
                goto cleanup;
            }
            const char* inputStr = PyUnicode_AsUTF8AndSize(input, NULL);
            RedisAI_DAGRunOpAddInput(scriptRunOp, inputStr);
        }