Fixed crash in Windows on Python exceptions.

wonder · wonder · commit 9f917b853e65 · 2007-05-03T17:46:09.000Z
git-svn-id: http://svn.osgeo.org/qgis/trunk/qgis@6928 c8812cc2-4d05-0410-92ff-de0c093fc19c
diff --git a/src/app/qgspythonutils.cpp b/src/app/qgspythonutils.cpp
@@ -21,9 +21,11 @@
 #include "qgspythonutils.h"
 
 #include "qgsapplication.h"
+#include "qgslogger.h"
 
 #include <QMessageBox>
 
+
 QString QgsPythonUtils::mPluginsPath;
 PyObject* QgsPythonUtils::mMainModule;
 PyObject* QgsPythonUtils::mMainDict;
@@ -167,6 +169,33 @@ bool QgsPythonUtils::runString(const QString& command)
 }
 
 
+QString QgsPythonUtils::getTypeAsString(PyObject* obj)
+{
+  if (obj == NULL)
+    return NULL;
+
+  if (PyClass_Check(obj))
+  {
+    QgsDebugMsg("got class");
+    return QString(PyString_AsString(((PyClassObject*)obj)->cl_name));
+  }
+  else if (PyType_Check(obj))
+  {
+    QgsDebugMsg("got type");
+	return QString(((PyTypeObject*)obj)->tp_name);
+  }
+  else
+  {
+    QgsDebugMsg("got object");
+    PyObject* s = PyObject_Str(obj);
+    QString str;
+    if (s && PyString_Check(s))
+      str = QString(PyString_AsString(s));
+    Py_XDECREF(s);
+    return str;
+  }
+}
+
 bool QgsPythonUtils::getError(QString& errorClassName, QString& errorText)
 {
   if (!PyErr_Occurred())
@@ -179,9 +208,9 @@ bool QgsPythonUtils::getError(QString& errorClassName, QString& errorText)
   
   // get the exception information
   PyErr_Fetch(&err_type, &err_value, &err_tb);
-    
+  
   // get exception's class name
-  errorClassName = PyString_AS_STRING(((PyClassObject*)err_type)->cl_name);
+  errorClassName = getTypeAsString(err_type);
     
   // get exception's text
   if (err_value != NULL && err_value != Py_None)
diff --git a/src/app/qgspythonutils.h b/src/app/qgspythonutils.h
@@ -57,6 +57,9 @@ class QgsPythonUtils
     
     static bool evalString(const QString& command, QString& result);
     
+    //! @return object's type name as a string
+    static QString getTypeAsString(PyObject* obj);
+
     //! get information about error to the supplied arguments
     //! @return false if there was no python error
     static bool getError(QString& errorClassName, QString& errorText);
