lib/base/console.cpp: fix eConsoleAppContainer write (when called with one string only)

1 files changed, 13 insertions, 10 deletions

diff --git a/lib/base/console.cpp b/lib/base/console.cpp
index add87066..43f9f61e 100644
--- a/lib/base/console.cpp
+++ b/lib/base/console.cpp
@@ -516,18 +516,15 @@ eConsolePy_write(eConsolePy* self, PyObject *args)
 {
 	int len;
 	char *data;
-	if (PyArg_ParseTuple(args, "si", &data, &len))
-		;
-	else
+	int ret = -1;
+	Py_ssize_t argc = PyTuple_Size(args);
+	if (argc > 1)
+		ret = PyArg_ParseTuple(args, "si", &data, &len);
+	else if (argc == 1)
 	{
 		PyObject *ob;
-		if (!PyArg_ParseTuple(args, "O", &ob) || !PyString_Check(ob))
-		{
-			PyErr_SetString(PyExc_TypeError,
-				"1st arg must be a string, optionaly 2nd arg can be the string length");
-			return NULL;
-		}
-		else
+		ret = !PyArg_ParseTuple(args, "O", &ob) || !PyString_Check(ob);
+		if (!ret)
 		{
 			Py_ssize_t length;
 			if (!PyString_AsStringAndSize(ob, &data, &length))
@@ -536,6 +533,12 @@ eConsolePy_write(eConsolePy* self, PyObject *args)
 				len = 0;
 		}
 	}
+	if (ret)
+	{
+		PyErr_SetString(PyExc_TypeError,
+			"1st arg must be a string, optionaly 2nd arg can be the string length");
+		return NULL;
+	}
 	self->cont->write(data, len);
 	Py_RETURN_NONE;
 }