Add isatty() method to PythonQtStdIn class

lassoan · fepegar · usiems · commit 7012000d8451 · 2021-06-24T13:57:16.000+02:00
pip and some other Python modules use this attribute to check if they are running in an interactive console. fixes MeVisLab#48 Co-authored-by: Fernando Perez-Garcia <fernando.perezgarcia.17@ucl.ac.uk>
diff --git a/src/PythonQtStdIn.cpp b/src/PythonQtStdIn.cpp
@@ -48,6 +48,7 @@ static PyObject *PythonQtStdInRedirect_new(PyTypeObject *type, PyObject * /*args
   self = (PythonQtStdInRedirect *)type->tp_alloc(type, 0);
   self->_cb = NULL;
   self->_callData = NULL;
+  self->_isatty = false;
 
   return (PyObject *)self;
 }
@@ -62,9 +63,20 @@ static PyObject *PythonQtStdInRedirect_readline(PyObject * self, PyObject * args
   return Py_BuildValue("s", QStringToPythonConstCharPointer(string));
 }
 
+static PyObject *PythonQtStdInRedirect_isatty(PyObject * self, PyObject * /*args*/)
+{
+  PythonQtStdInRedirect* s = (PythonQtStdInRedirect*)self;
+  PyObject* r = s->_isatty ? Py_True : Py_False;
+  Py_INCREF(r);
+  return r;
+}
+
 static PyMethodDef PythonQtStdInRedirect_methods[] = {
   {"readline", (PyCFunction)PythonQtStdInRedirect_readline, METH_VARARGS,
    "read input line"},
+  {"isatty", (PyCFunction)PythonQtStdInRedirect_isatty,   METH_NOARGS,
+   "returns True if this is a tty-like device. False by default."
+  },
   {NULL,    NULL, 0 , NULL} /* sentinel */
 };
 
diff --git a/src/PythonQtStdIn.h b/src/PythonQtStdIn.h
@@ -58,6 +58,7 @@ typedef struct {
   PyObject_HEAD
   PythonQtInputChangedCB* _cb;
   void * _callData;
+  bool _isatty;
 } PythonQtStdInRedirect;
 
 #endif
