Add std::ostream& operator<<(std::ostream& out, PyObject* obj) to create better and easier debug messages.

1 files changed, 13 insertions, 0 deletions

diff --git a/libshiboken/sbkdbg.h b/libshiboken/sbkdbg.h
index 62dacb8a6..b3db0227a 100644
--- a/libshiboken/sbkdbg.h
+++ b/libshiboken/sbkdbg.h
@@ -35,6 +35,7 @@
 #ifndef SBKDBG_H
 #define SBKDBG_H
 
+#include <Python.h>
 #include <iostream>
 
 #ifndef NOCOLOR
@@ -77,6 +78,18 @@ private:
     const char* m_context;
 };
 
+inline std::ostream& operator<<(std::ostream& out, PyObject* obj)
+{
+    PyObject* repr = PyObject_Repr(obj);
+    if (repr) {
+        out << PyString_AS_STRING(repr);
+        Py_DECREF(repr);
+    } else {
+        out << reinterpret_cast<void*>(obj);
+    }
+    return out;
+}
+
 class _SbkDbg : public BaseLogger
 {
 public: