From f8ad2b31a9a1a43ac415e8b203851c6f091dde6f Mon Sep 17 00:00:00 2001
From: David Shah <davey1576@gmail.com>
Date: Wed, 4 Jul 2018 12:23:35 +0200
Subject: python: Translate NPNR_ASSERTs to Python

Signed-off-by: David Shah <davey1576@gmail.com>
---
 common/pybindings.cc | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/common/pybindings.cc b/common/pybindings.cc
index e776e33a..ea57dbbb 100644
--- a/common/pybindings.cc
+++ b/common/pybindings.cc
@@ -61,8 +61,15 @@ Context *load_design_shim(std::string filename, ArchArgs args)
     return d;
 }
 
+void translate_assertfail(const assertion_failure &e) {
+    // Use the Python 'C' API to set up an exception object
+    PyErr_SetString(PyExc_AssertionError, e.what());
+}
+
 BOOST_PYTHON_MODULE(MODULE_NAME)
 {
+    register_exception_translator<assertion_failure>(&translate_assertfail);
+
     using namespace PythonConversion;
 
     class_<GraphicElement>("GraphicElement")
-- 
cgit v1.2.3