Added exceptions for unregistered python function and module use

csullivan · Apr 17, 2016 · 9d2c162 · 9d2c162
1 parent ce21625
commit 9d2c162
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/include/pylingual.hpp b/include/pylingual.hpp
@@ -11,9 +11,11 @@
 #include <Python.h>
 #include <converters.hpp>
 #include <error_handle.hpp>
+#include <pylingual_exception.hpp>
 
 using namespace std;
 
+//class pylingual_exception;
 class pylingual {
 
 public:
@@ -29,6 +31,8 @@ class pylingual {
   void function(const char* pyname, Args&&... args) {
     if (functions.count(pyname)>0) {
       functions[pyname](std::forward<Args>(args)...);
+    } else {
+      throw pylingual_unregistered();
     }
   }
 
@@ -95,5 +99,8 @@ pylingual::pylingual() {
   modules["__builtin__"]=built_in;
 }
 
+
+
+
 #endif
 
diff --git a/include/pylingual_exception.hpp b/include/pylingual_exception.hpp
@@ -0,0 +1,12 @@
+#include <exception>
+class pylingual_exception : public std::exception
+{
+public:
+  virtual const char* what() const noexcept { return "standard pylingual exception"; };
+};
+
+class pylingual_unregistered : public pylingual_exception
+{
+public:
+  virtual const char* what() const noexcept { return "Module/Function has not been imported"; };
+};