Added capability to not install signal handlers

themarpe · themarpe · commit cbee9c3f9650 · 2021-09-02T17:24:42.000+02:00
diff --git a/depthai-core b/depthai-core
@@ -1 +1 @@
-Subproject commit 8c13909dd1f9798c2ea0ad467bd9bb0cbd7ec34b
+Subproject commit f375e951e0ce21998bc326845553f191a0013ac5
diff --git a/src/py_bindings.cpp b/src/py_bindings.cpp
@@ -59,8 +59,28 @@ PYBIND11_MODULE(depthai,m)
     // Initial call
     CommonBindings::bind(m, &callstackAdapter);
 
+    // Install signal handler option
+    bool installSignalHandler = true;
+    constexpr static const char* signalHandlerKey = "DEPTHAI_INSTALL_SIGNAL_HANDLER";
+    try {
+        auto sysModule = py::module_::import("sys");
+        if(py::hasattr(sysModule, signalHandlerKey)){
+            installSignalHandler = installSignalHandler && sysModule.attr(signalHandlerKey).cast<bool>();
+        }
+    } catch (...) {
+        // ignore
+    }
+    try {
+        auto builtinsModule = py::module_::import("builtins");
+        if(py::hasattr(builtinsModule, signalHandlerKey)){
+            installSignalHandler = installSignalHandler && builtinsModule.attr(signalHandlerKey).cast<bool>();
+        }
+    } catch (...){
+        // ignore
+    }
+
     // Call dai::initialize on 'import depthai' to initialize asap with additional information to print
-    dai::initialize(std::string("Python bindings - version: ") + DEPTHAI_PYTHON_VERSION + " from " + DEPTHAI_PYTHON_COMMIT_DATETIME + " build: " + DEPTHAI_PYTHON_BUILD_DATETIME);
+    dai::initialize(std::string("Python bindings - version: ") + DEPTHAI_PYTHON_VERSION + " from " + DEPTHAI_PYTHON_COMMIT_DATETIME + " build: " + DEPTHAI_PYTHON_BUILD_DATETIME, installSignalHandler);
 
 }
 
