source: trunk/zoo-kernel/service_internal_python.c @ 6

/**
 * Author : Gérald FENOY
 *
 * Copyright (c) 2009-2010 GeoLabs SARL
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */

#include "service_internal_python.h"

int python_support(maps* m,service* s,int argc,char** argv,map *inputs,map *outputs){
  Py_Initialize();
  PyObject *pName, *pModule, *pFunc;
  pName = PyString_FromString(argv[2]);
  pModule = PyImport_Import(pName);
  int i;
  if (pModule != NULL) {
    pFunc=PyObject_GetAttrString(pModule,argv[4]);
    if (pFunc && PyCallable_Check(pFunc)){
      PyDictObject* arg1=PyDict_FromMap(inputs);
      PyDictObject* arg2=PyDict_FromMap(outputs);
      PyObject *pArgs=PyTuple_New(2);
      PyObject *pValue;
      PyTuple_SetItem(pArgs, 0, (PyObject *)arg1);
      PyTuple_SetItem(pArgs, 1, (PyObject *)arg2);
      pValue = PyString_FromString(argv[argc-1]);
#ifdef DEBUG
      fprintf(stderr,"Function successfully loaded\n");
#endif  
      /**
       * Need to check if we need to fork to load a status enabled 
       */
      if(strcmp(PyString_AsString(pValue),"bg")!=0){
#ifdef DEBUG
        fprintf(stderr,"RUN IN NORMAL MODE \n");
#endif
        pValue = PyObject_CallObject(pFunc, pArgs);
        if (pValue != NULL) {
          inputs=mapFromPyDict(arg1);
          outputs=mapFromPyDict(arg2);
#ifdef DEBUG
          fprintf(stderr,"Result of call: %i\n", PyInt_AsLong(pValue));
          dumpMap(inputs);
          dumpMap(outputs);
#ifndef WIN32
          fprintf(stderr,"printProcessResponse(%i,\"%s\",%i,inputs,outputs);",getpid(),argv[1],PyInt_AsLong(pValue));
#endif
#endif
          printProcessResponse(m,getpid(),s,argv[2],PyInt_AsLong(pValue),inputs,outputs);
        }
      }
      else{
        pid_t pid;
#ifdef WIN32
        pid = 0;
#else
        pid = fork ();
#endif
        if (pid > 0) {
          /* dady */
          printProcessResponse(m,pid,s,argv[2],SERVICE_STARTED,inputs,outputs);
        }else if (pid == 0) {
          /* son */
#ifdef DEBUG
          fprintf(stderr,"RUN IN BACKGROUND MODE \n");
#endif
          char tmp1[256];
          sprintf(tmp1,"service/temp/%s_%d.xml",argv[2],getpid());
#ifndef WIN32
          stdout =
#endif
                  freopen(tmp1 , "w+" , stdout);
          printProcessResponse(m,getpid(),s,argv[2],SERVICE_STARTED,inputs,outputs);
          fflush(stdout);
          pValue = PyObject_CallObject(pFunc, pArgs);
          if (pValue != NULL) {
            rewind(stdout);
            inputs=mapFromPyDict(arg1);
            outputs=mapFromPyDict(arg2);
#ifdef DEBUG
            fprintf(stderr,"Result of call: %ld\n", PyInt_AsLong(pValue));
            dumpMap(inputs);
            dumpMap(outputs);
            fprintf(stderr,"printProcessResponse(%i,%s,%i,inputs,outputs);",s,getpid(),argv[2],PyInt_AsLong(pValue));
#endif
//          sleep(6);
            printProcessResponse(m,getpid(),s,argv[2],PyInt_AsLong(pValue),inputs,outputs);
          }
          exit(1);
        } else {
          /* error */
        }
      }
      Py_DECREF(pArgs);
      Py_DECREF(pValue);
#ifdef DEBUG
      fprintf(stderr,"Function ran successfully\n");
#endif  
    }
    else{
      char tmpS[1024];
      sprintf(tmpS, "Cannot find function %s \n", argv[4]);
      map* tmps=createMap("text",tmpS);
      printExceptionReportResponse(m,tmps);
      //fprintf(stderr, "Cannot find function %s \n", argv[2]);
      exit(-1);
    }
  }else{
    char tmpS[1024];
    sprintf(tmpS, "Cannot find function %s \n", argv[4]);
    map* tmps=createMap("text",tmpS);
    printExceptionReportResponse(m,tmps);
    //fprintf(stderr, "Cannot find function %s \n", argv[4]);
    if (PyErr_Occurred())
      PyErr_Print();

  }
  Py_Finalize();
  return 2;
}

int zoo_python_support(maps** main_conf,map* request,service* s,maps **real_inputs,maps **real_outputs){
  maps* m=*main_conf;
  maps* inputs=*real_inputs;
  maps* outputs=*real_outputs;
  char ntmp[1024];
  getcwd(ntmp,1024);
  map* tmp=NULL;
  tmp=getMapFromMaps(*main_conf,"env","PYTHONPATH");
  char *python_path=NULL;
  if(tmp!=NULL){
          python_path=(char*)malloc((1+strlen(tmp->value))*sizeof(char));
          sprintf(python_path,"%s",tmp->value);
  }
  else{
        python_path=(char*)malloc(2*sizeof(char));
        python_path=".";
  }
  tmp=NULL;
  tmp=getMap(request,"metapath");
  char *pythonpath=(char*)malloc((1+strlen(python_path)+2048)*sizeof(char));
  if(tmp!=NULL && strcmp(tmp->value,"")!=0)
#ifdef WIN32
          sprintf(pythonpath,"%s/%s/;%s",ntmp,tmp->value,python_path);
#else
          sprintf(pythonpath,"%s/%s/:%s",ntmp,tmp->value,python_path);
#endif
  else
#ifdef WIN32
    sprintf(pythonpath,"%s;%s",ntmp,python_path);
#else
    sprintf(pythonpath,"%s:%s",ntmp,python_path);
#endif
#ifdef DEBUG
  fprintf(stderr,"PYTHONPATH=%s\n",pythonpath);
#endif
#ifndef WIN32
  setenv("PYTHONPATH",pythonpath,1);
#else
  SetEnvironmentVariable("PYTHONPATH",pythonpath);
#endif
  free(python_path);
  free(pythonpath);
  //
  Py_Initialize();
  PyObject *pName, *pModule, *pFunc;
  tmp=getMap(s->content,"serviceProvider");
  //if(tmp!=NULL)
  pName = PyString_FromString(tmp->value);
  pModule = PyImport_Import(pName);
  int i;
  int res=SERVICE_FAILED;
  int cpid=getpid();
  if (pModule != NULL) {
    pFunc=PyObject_GetAttrString(pModule,s->name);
    if (pFunc && PyCallable_Check(pFunc)){
      PyDictObject* arg1=PyDict_FromMaps(m);
      PyDictObject* arg2=PyDict_FromMaps(inputs);
      PyDictObject* arg3=PyDict_FromMaps(outputs);
      PyObject *pArgs=PyTuple_New(3);
      PyObject *pValue;
      PyTuple_SetItem(pArgs, 0, (PyObject *)arg1);
      PyTuple_SetItem(pArgs, 1, (PyObject *)arg2);
      PyTuple_SetItem(pArgs, 2, (PyObject *)arg3);
      tmp=getMap(request,"storeExecuteResponse");
#ifdef DEBUG
      fprintf(stderr,"RUN IN NORMAL MODE \n");
      //fflush(stderr);
#endif
      pValue = PyObject_CallObject(pFunc, pArgs);
      if (pValue != NULL) {
        res=PyInt_AsLong(pValue);
        /*inputs=mapsFromPyDict(arg2);
        dumpMaps(inputs);*/
        outputs=mapsFromPyDict(arg3);
        *real_outputs=outputs;

#ifdef DEBUG
        fprintf(stderr,"Result of call: %i\n", PyInt_AsLong(pValue));
        dumpMaps(inputs);
        dumpMaps(outputs);
        fprintf(stderr,"printProcessResponse(%i,\"%s\",%i,inputs,outputs);",
                getpid(),tmp->value,PyInt_AsLong(pValue));
        dumpMaps(outputs);
#endif
        Py_DECREF(pValue);
        Py_XDECREF(pFunc);
        Py_DECREF(pArgs);
        Py_DECREF(pModule);
      }else{
          
        PyObject *ptype,*pvalue, *ptraceback;
        PyErr_Fetch(&ptype, &pvalue, &ptraceback);
        PyObject *trace=PyObject_Str(pvalue);
        char tb[1024];
        char pbt[10240];
        if(PyString_Check(trace))
          sprintf(pbt,"TRACE : %s",PyString_AsString(trace));
        else
          fprintf(stderr,"EMPTY TRACE ?");
        trace=NULL;
        trace=PyObject_Str(ptype);
        if(PyString_Check(trace))
          sprintf(pbt,"%s\nTRACE : %s",strdup(pbt),PyString_AsString(trace));
        else
          fprintf(stderr,"EMPTY TRACE ?");
        PyObject *t;
        pName = PyString_FromString("traceback");
        pModule = PyImport_Import(pName);
        pArgs = PyTuple_New(1);
        PyTuple_SetItem(pArgs, 0, ptraceback);
        pFunc = PyObject_GetAttrString(pModule,"format_tb");
        pValue = PyObject_CallObject(pFunc, pArgs);
        trace=NULL;
        trace=PyObject_Str(pValue);
        if(PyString_Check(trace))
          sprintf(pbt,"%s\nUnable to run your python process properly. Please check the following messages : %s",strdup(pbt),PyString_AsString(trace));
        else
          sprintf(pbt,"%s \n Unable to run your python process properly. Unable to provide any futher informations.",strdup(pbt));
        map* err=createMap("text",pbt);
        addToMap(err,"code","NoApplicableCode");
        printExceptionReportResponse(m,err);
        Py_XDECREF(pFunc);
        Py_DECREF(pArgs);
        Py_DECREF(pModule);
        exit(-1);
      }
    }
    else{
      char tmpS[1024];
      sprintf(tmpS, "Cannot find the %s function int the %s file.\n", s->name, tmp->value);
      map* tmps=createMap("text",tmpS);
      printExceptionReportResponse(m,tmps);
      //fprintf(stderr, "Cannot find function %s \n", argv[2]);
      Py_XDECREF(pFunc);
      Py_DECREF(pModule);
      exit(-1);
    }
  } else{
    char tmpS[1024];
    sprintf(tmpS, "Cannot find function %s \n", tmp->value);
    map* tmps=createMap("text",tmpS);
    printExceptionReportResponse(m,tmps);
    //fprintf(stderr, "Cannot find function %s \n", argv[4]);
    if (PyErr_Occurred())
      PyErr_Print();
    exit(-1);
  } 
  Py_Finalize();
  return res;
}

PyDictObject* PyDict_FromMaps(maps* t){
  PyObject* res=PyDict_New( );
  maps* tmp=t;
  while(tmp!=NULL){
    if(PyDict_SetItem(res,PyString_FromString(tmp->name),(PyObject*)PyDict_FromMap(tmp->content))<0){
      fprintf(stderr,"Unable to parse params...");
      exit(1);
    }
    tmp=tmp->next;
  }  
  return (PyDictObject*) res;
}

PyDictObject* PyDict_FromMap(map* t){
  PyObject* res=PyDict_New( );
  map* tmp=t;
  while(tmp!=NULL){
    if(PyDict_SetItem(res,PyString_FromString(tmp->name),PyString_FromString(tmp->value))<0){
      fprintf(stderr,"Unable to parse params...");
      exit(1);
    }
    tmp=tmp->next;
  }
  return (PyDictObject*) res;
}

maps* mapsFromPyDict(PyDictObject* t){
  maps* res=NULL;
  maps* cursor=res;
  PyObject* list=PyDict_Keys((PyObject*)t);
  int nb=PyList_Size(list);
  int i;
  for(i=0;i<nb;i++){
#ifdef DEBUG
    fprintf(stderr,">> parsing maps %d\n",i);
#endif
    PyObject* key=PyList_GetItem(list,i);
    PyObject* value=PyDict_GetItem((PyObject*)t,key);
#ifdef DEBUG
    fprintf(stderr,">> DEBUG VALUES : %s => %s\n",
            PyString_AsString(key),PyString_AsString(value));
#endif
    if(cursor!=NULL){
      cursor=cursor->next;
    }
    cursor=(maps*)malloc(sizeof(maps*));
    cursor->name=PyString_AsString(key);
#ifdef DEBUG
    dumpMap(mapFromPyDict((PyDictObject*)value));
#endif
    cursor->content=mapFromPyDict((PyDictObject*)value);
    cursor->next=NULL;
    if(res==NULL)
      res=cursor;
    else
      addMapsToMaps(&res,cursor);
#ifdef DEBUG
    dumpMaps(res);
    fprintf(stderr,">> parsed maps %d\n",i);
#endif
  }
  return res;
}

map* mapFromPyDict(PyDictObject* t){
  map* res=NULL;
  PyObject* list=PyDict_Keys((PyObject*)t);
  int nb=PyList_Size(list);
  int i;
  for(i=0;i<nb;i++){
    PyObject* key=PyList_GetItem(list,i);
    PyObject* value=PyDict_GetItem((PyObject*)t,key);
#ifdef DEBUG
    fprintf(stderr,">> DEBUG VALUES : %s => %s\n",
            PyString_AsString(key),PyString_AsString(value));
#endif
    if(res!=NULL)
      addToMap(res,PyString_AsString(key),PyString_AsString(value));
    else
      res=createMap(PyString_AsString(key),PyString_AsString(value));
  }
  return res;
}