X-Git-Url: http://git.dolda2000.com/gitweb/?a=blobdiff_plain;f=lib%2Fpython%2Fdolmod.c;h=adf07a24d53d95664229be4ea731035462b94df4;hb=34905adcdf7b0ac4a459f8c921ae7291b9103a7b;hp=edae7ae702ef416d14e90a19fe2f7012aa36146c;hpb=fbe30a6da13729c16770608bb5a7f11d79b22fb2;p=doldaconnect.git

diff --git a/lib/python/dolmod.c b/lib/python/dolmod.c
index edae7ae..adf07a2 100644
--- a/lib/python/dolmod.c
+++ b/lib/python/dolmod.c
@@ -241,12 +241,15 @@ static PyObject *mod_qcmd(PyObject *self, PyObject *args, PyObject *kwargs)
     toks = NULL;
     tokssize = toksdata = 0;
     cmd = NULL;
+    ret = NULL;
     for(i = 0; i < PySequence_Size(args); i++) {
-	c = PySequence_GetItem(args, i);
+	if((c = PySequence_GetItem(args, i)) == NULL)
+	    goto out;
 	if(!PyUnicode_Check(c)) {
 	    n = PyUnicode_FromObject(c);
 	    Py_DECREF(c);
-	    c = n;
+	    if((c = n) == NULL)
+		goto out;
 	}
 	tok = smalloc((toksize = (PyUnicode_GetSize(c) + 1)) * sizeof(*tok));
 	tok[PyUnicode_AsWideChar((PyUnicodeObject *)c, tok, toksize)] = L'\0';
@@ -258,7 +261,7 @@ static PyObject *mod_qcmd(PyObject *self, PyObject *args, PyObject *kwargs)
     }
     if(cmd == NULL) {
 	PyErr_SetString(PyExc_TypeError, "qcmd needs at least 1 argument");
-	return(NULL);
+	goto out;
     }
     addtobuf(toks, NULL);
     ret = NULL;
@@ -273,8 +276,11 @@ static PyObject *mod_qcmd(PyObject *self, PyObject *args, PyObject *kwargs)
     } else {
 	ret = PyInt_FromLong(dc_queuecmd(NULL, NULL, cmd, L"%%a", toks, NULL));
     }
+
+out:
     dc_freewcsarr(toks);
-    free(cmd);
+    if(cmd != NULL)
+	free(cmd);
     return(ret);
 }
 
@@ -376,6 +382,14 @@ static PyObject *mod_lexsexpr(PyObject *self, PyObject *args)
     return(ret);
 }
 
+static PyObject *mod_wantwrite(PyObject *self)
+{
+    if(dc_wantwrite())
+	Py_RETURN_TRUE;
+    else
+	Py_RETURN_FALSE;
+}
+
 static PyMethodDef methods[] = {
     {"connect", mod_connect, METH_VARARGS,
      "Connect to a Dolda Connect server"},
@@ -393,6 +407,8 @@ static PyMethodDef methods[] = {
      "Perform an asynchronous login procedure"},
     {"lexsexpr", mod_lexsexpr, METH_VARARGS,
      "Use a standard algorithm to lex a search expression"},
+    {"wantwrite", (PyCFunction)mod_wantwrite, METH_NOARGS,
+     "Return a boolean indicating whether there is output to be fed to the server"},
     {NULL, NULL, 0, NULL}
 };