tris/recoll
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Python module: check types in setbinurl

Browse Source
This commit is contained in:
Jean-Francois Dockes 2012-10-21 18:21:48 +02:00
parent 5f0e348216
commit b3b9b47866
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/python/recoll/pyrecoll.cpp
View File

@ -18,6 +18,7 @@
#include <Python.h> #include <Python.h>
#include <structmember.h> #include <structmember.h>
#include <bytearrayobject.h>
#include <strings.h> #include <strings.h>
@ -321,18 +322,22 @@ PyDoc_STRVAR(doc_setbinurl,
"Set the URL from binary path like file://may/contain/unencodable/bytes\n" "Set the URL from binary path like file://may/contain/unencodable/bytes\n"
); );
static int static PyObject *
Doc_setbinurl(recoll_DocObject *self, PyObject *value) Doc_setbinurl(recoll_DocObject *self, PyObject *value)
{ {
if (self->doc == 0 || if (self->doc == 0 ||
the_docs.find(self->doc) == the_docs.end()) { the_docs.find(self->doc) == the_docs.end()) {
PyErr_SetString(PyExc_AttributeError, "doc??"); PyErr_SetString(PyExc_AttributeError, "doc??");
return -1; return 0;
}
if (!PyByteArray_Check(value)) {
PyErr_SetString(PyExc_TypeError, "setbinurl needs byte array argument");
return 0;
} }
self->doc->url = string(PyByteArray_AsString(value), self->doc->url = string(PyByteArray_AsString(value),
PyByteArray_Size(value)); PyByteArray_Size(value));
return 0; Py_RETURN_NONE;
} }