From: Ovidiu Mircescu <ovidiu.mircescu@edf.fr>
Date: Mon, 1 Oct 2018 12:29:08 +0000 (+0200)
Subject: Porting SWIG to Python3.
X-Git-Tag: V2_4_1
X-Git-Url: http://git.salome-platform.org/gitweb/?a=commitdiff_plain;h=8ab8764737a35352c010970946ec6634737d94d2;p=tools%2Flibbatch.git

Porting SWIG to Python3.
---

diff --git a/src/Python/libBatch_Swig_typemap.i b/src/Python/libBatch_Swig_typemap.i
index cbd1a23..5e23259 100644
--- a/src/Python/libBatch_Swig_typemap.i
+++ b/src/Python/libBatch_Swig_typemap.i
@@ -43,22 +43,35 @@
 # %typemap(in) SWIGTYPE ;
 
 %{
+static std::string pystr2stdstr(PyObject * val)
+{
+  std::string str_val;
+  if (PyUnicode_Check(val))
+  {
+    PyObject * temp_bytes = PyUnicode_AsEncodedString(val, "UTF-8", "strict");
+    if (temp_bytes != NULL)
+    {
+      str_val = PyBytes_AS_STRING(temp_bytes);
+      Py_DECREF(temp_bytes);
+    }
+  }
+  return str_val;
+}
+
 // Helper function to initialize a Batch::Versatile from a PyObj
 static bool initVersatile(Batch::Versatile & newVersatile, PyObject * input)
 {
   if (PyList_Check(input)) { // c'est une liste
     for (Py_ssize_t i=0; i<PyList_Size(input); i++) {
       PyObject * val = PyList_GetItem(input, i);
-      if (PyString_Check(val)) {
-        newVersatile += PyString_AsString(val);
-
+      if (PyUnicode_Check(val)) {
+        newVersatile += pystr2stdstr(val);
       } else if (PyTuple_Check(val) &&
           (PyTuple_Size(val) == 2) &&
-          PyString_Check( PyTuple_GetItem(val,0) ) &&
-          PyString_Check( PyTuple_GetItem(val,1) )   ) {
-        newVersatile += Batch::Couple( PyString_AsString( PyTuple_GetItem(val,0) ),
-                                       PyString_AsString( PyTuple_GetItem(val,1) )
-                                     );
+          PyUnicode_Check( PyTuple_GetItem(val,0) ) &&
+          PyUnicode_Check( PyTuple_GetItem(val,1) )   ) {
+        newVersatile += Batch::Couple( pystr2stdstr(PyTuple_GetItem(val,0)),
+                                       pystr2stdstr(PyTuple_GetItem(val,1)));
 
       } else {
         PyErr_SetString(PyExc_RuntimeWarning, "initVersatile : invalid PyObject");
@@ -66,8 +79,8 @@ static bool initVersatile(Batch::Versatile & newVersatile, PyObject * input)
       }
     }
 
-  } else if (PyString_Check(input)) { // c'est une string
-    newVersatile = PyString_AsString(input);
+  } else if (PyUnicode_Check(input)) { // c'est une string
+    newVersatile = pystr2stdstr(input);
   } else if (PyBool_Check(input)) { // c'est un bool
     newVersatile = (input == Py_True);
   } else if (PyInt_Check(input)) { // c'est un int
@@ -156,7 +169,7 @@ static bool initParameter(Batch::Parametre & newParam, PyObject * input)
     PyObject *key, *value;
     Py_ssize_t pos = 0;
     while (PyDict_Next(input, &pos, &key, &value)) {
-      std::string mk = PyString_AsString(key);
+      std::string mk = pystr2stdstr(key);
       bool res = initVersatile(newParam[mk], value);
       if (!res)
         return false;
@@ -186,8 +199,8 @@ static bool initEnvironment(Batch::Environnement & newEnv, PyObject * input)
   PyObject *key, *value;
   Py_ssize_t pos = 0;
   while (PyDict_Next(input, &pos, &key, &value)) {
-    std::string mk  = PyString_AsString(key);
-    std::string val = PyString_AsString(value);
+    std::string mk  = pystr2stdstr(key);
+    std::string val = pystr2stdstr(value);
     newEnv[mk] = val;
   }
   return true;