Test for undefined constants

lf-lang · Jul 2, 2024 · ae9e5f4 · ae9e5f4
1 parent 193b629
commit ae9e5f4
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/python/lib/pythontarget.c b/python/lib/pythontarget.c
@@ -162,10 +162,24 @@ PyObject* py_request_stop(PyObject* self, PyObject* args) {
   return Py_None;
 }
 
-PyObject* py_source_directory(PyObject* self, PyObject* args) { return PyUnicode_DecodeFSDefault(LF_SOURCE_DIRECTORY); }
+PyObject* py_source_directory(PyObject* self, PyObject* args) {
+#ifndef LF_SOURCE_DIRECTORY
+  // This should not occur.
+  PyErr_SetString(PyExc_RuntimeError, "LF_SOURCE_DIRECTORY constant is not defined.");
+  return NULL;
+#else
+  return PyUnicode_DecodeFSDefault(LF_SOURCE_DIRECTORY);
+#endif
+}
 
 PyObject* py_package_directory(PyObject* self, PyObject* args) {
+#ifndef LF_PACKAGE_DIRECTORY
+  // This should not occur.
+  PyErr_SetString(PyExc_RuntimeError, "LF_PACKAGE_DIRECTORY constant is not defined.");
+  return NULL;
+#else
   return PyUnicode_DecodeFSDefault(LF_PACKAGE_DIRECTORY);
+#endif
 }
 
 /**