Add a mule_file method to the Python API

Alexandre Bonnetain · Alexandre Bonnetain · commit c80d51ed6efe · 2016-11-02T11:26:29.000+01:00
diff --git a/plugins/python/uwsgi_pymodule.c b/plugins/python/uwsgi_pymodule.c
@@ -1488,6 +1488,27 @@ PyObject *py_uwsgi_mule_get_msg(PyObject * self, PyObject * args, PyObject *kwar
 	return msg;
 }
 
+PyObject *py_uwsgi_mule_file(PyObject * self, PyObject * args) {
+	
+	int mule_id = uwsgi.muleid;
+
+	if (PyArg_ParseTuple(args, "|i", &mule_id) && mule_id == 0) {
+		return NULL;
+	}
+
+	if (uwsgi.mules_cnt < mule_id && mule_id > 0) {
+		return PyErr_Format(PyExc_ValueError, "The id you provided doesn't correspond to any mule");
+	}
+
+	struct uwsgi_mule *mule = get_mule_by_id(mule_id);
+	if (!mule || !mule->patch) {
+		Py_INCREF(Py_None);
+		return Py_None;
+	}
+
+	return PyString_FromString(mule->patch);
+}
+
 PyObject *py_uwsgi_farm_get_msg(PyObject * self, PyObject * args) {
 
         ssize_t len = 0;
@@ -2815,6 +2836,8 @@ static PyMethodDef uwsgi_advanced_methods[] = {
 
 	{"micros", py_uwsgi_micros, METH_VARARGS, ""},
 
+	{"mule_file", py_uwsgi_mule_file, METH_VARARGS, ""},
+
 	{NULL, NULL},
 };
 
