From 1168d1ff9087e30d88bf18ec405ad0a8cf25f5c7 Mon Sep 17 00:00:00 2001
From: Miodrag Milanovic <mmicko@gmail.com>
Date: Fri, 7 Aug 2020 09:46:02 +0200
Subject: Made proper exception translation

---
 common/pybindings.cc | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/common/pybindings.cc b/common/pybindings.cc
index b6d5e48a..8f824f1e 100644
--- a/common/pybindings.cc
+++ b/common/pybindings.cc
@@ -86,7 +86,13 @@ template <> struct string_converter<Property>
 
 PYBIND11_MODULE(MODULE_NAME, m)
 {
-    py::register_exception<assertion_failure>(m, "PyExc_AssertionError");
+    py::register_exception_translator([](std::exception_ptr p) {
+        try {
+            if (p) std::rethrow_exception(p);
+        } catch (const assertion_failure &e) {
+            PyErr_SetString(PyExc_AssertionError, e.what());
+        }
+    });
 
     using namespace PythonConversion;
 
-- 
cgit v1.2.3