Python-2.7.3/Modules/_sqlite/row.c

No issues found

  1 /* row.c - an enhanced tuple for database rows
  2  *
  3  * Copyright (C) 2005-2010 Gerhard Häring <gh@ghaering.de>
  4  *
  5  * This file is part of pysqlite.
  6  *
  7  * This software is provided 'as-is', without any express or implied
  8  * warranty.  In no event will the authors be held liable for any damages
  9  * arising from the use of this software.
 10  *
 11  * Permission is granted to anyone to use this software for any purpose,
 12  * including commercial applications, and to alter it and redistribute it
 13  * freely, subject to the following restrictions:
 14  *
 15  * 1. The origin of this software must not be misrepresented; you must not
 16  *    claim that you wrote the original software. If you use this software
 17  *    in a product, an acknowledgment in the product documentation would be
 18  *    appreciated but is not required.
 19  * 2. Altered source versions must be plainly marked as such, and must not be
 20  *    misrepresented as being the original software.
 21  * 3. This notice may not be removed or altered from any source distribution.
 22  */
 23 
 24 #include "row.h"
 25 #include "cursor.h"
 26 #include "sqlitecompat.h"
 27 
 28 void pysqlite_row_dealloc(pysqlite_Row* self)
 29 {
 30     Py_XDECREF(self->data);
 31     Py_XDECREF(self->description);
 32 
 33     Py_TYPE(self)->tp_free((PyObject*)self);
 34 }
 35 
 36 int pysqlite_row_init(pysqlite_Row* self, PyObject* args, PyObject* kwargs)
 37 {
 38     PyObject* data;
 39     pysqlite_Cursor* cursor;
 40 
 41     self->data = 0;
 42     self->description = 0;
 43 
 44     if (!PyArg_ParseTuple(args, "OO", &cursor, &data)) {
 45         return -1;
 46     }
 47 
 48     if (!PyObject_IsInstance((PyObject*)cursor, (PyObject*)&pysqlite_CursorType)) {
 49         PyErr_SetString(PyExc_TypeError, "instance of cursor required for first argument");
 50         return -1;
 51     }
 52 
 53     if (!PyTuple_Check(data)) {
 54         PyErr_SetString(PyExc_TypeError, "tuple required for second argument");
 55         return -1;
 56     }
 57 
 58     Py_INCREF(data);
 59     self->data = data;
 60 
 61     Py_INCREF(cursor->description);
 62     self->description = cursor->description;
 63 
 64     return 0;
 65 }
 66 
 67 PyObject* pysqlite_row_subscript(pysqlite_Row* self, PyObject* idx)
 68 {
 69     long _idx;
 70     char* key;
 71     int nitems, i;
 72     char* compare_key;
 73 
 74     char* p1;
 75     char* p2;
 76 
 77     PyObject* item;
 78 
 79     if (PyInt_Check(idx)) {
 80         _idx = PyInt_AsLong(idx);
 81         item = PyTuple_GetItem(self->data, _idx);
 82         Py_XINCREF(item);
 83         return item;
 84     } else if (PyLong_Check(idx)) {
 85         _idx = PyLong_AsLong(idx);
 86         item = PyTuple_GetItem(self->data, _idx);
 87         Py_XINCREF(item);
 88         return item;
 89     } else if (PyString_Check(idx)) {
 90         key = PyString_AsString(idx);
 91 
 92         nitems = PyTuple_Size(self->description);
 93 
 94         for (i = 0; i < nitems; i++) {
 95             compare_key = PyString_AsString(PyTuple_GET_ITEM(PyTuple_GET_ITEM(self->description, i), 0));
 96             if (!compare_key) {
 97                 return NULL;
 98             }
 99 
100             p1 = key;
101             p2 = compare_key;
102 
103             while (1) {
104                 if ((*p1 == (char)0) || (*p2 == (char)0)) {
105                     break;
106                 }
107 
108                 if ((*p1 | 0x20) != (*p2 | 0x20)) {
109                     break;
110                 }
111 
112                 p1++;
113                 p2++;
114             }
115 
116             if ((*p1 == (char)0) && (*p2 == (char)0)) {
117                 /* found item */
118                 item = PyTuple_GetItem(self->data, i);
119                 Py_INCREF(item);
120                 return item;
121             }
122 
123         }
124 
125         PyErr_SetString(PyExc_IndexError, "No item with that key");
126         return NULL;
127     } else if (PySlice_Check(idx)) {
128         PyErr_SetString(PyExc_ValueError, "slices not implemented, yet");
129         return NULL;
130     } else {
131         PyErr_SetString(PyExc_IndexError, "Index must be int or string");
132         return NULL;
133     }
134 }
135 
136 Py_ssize_t pysqlite_row_length(pysqlite_Row* self, PyObject* args, PyObject* kwargs)
137 {
138     return PyTuple_GET_SIZE(self->data);
139 }
140 
141 PyObject* pysqlite_row_keys(pysqlite_Row* self, PyObject* args, PyObject* kwargs)
142 {
143     PyObject* list;
144     int nitems, i;
145 
146     list = PyList_New(0);
147     if (!list) {
148         return NULL;
149     }
150     nitems = PyTuple_Size(self->description);
151 
152     for (i = 0; i < nitems; i++) {
153         if (PyList_Append(list, PyTuple_GET_ITEM(PyTuple_GET_ITEM(self->description, i), 0)) != 0) {
154             Py_DECREF(list);
155             return NULL;
156         }
157     }
158 
159     return list;
160 }
161 
162 static int pysqlite_row_print(pysqlite_Row* self, FILE *fp, int flags)
163 {
164     return (&PyTuple_Type)->tp_print(self->data, fp, flags);
165 }
166 
167 static PyObject* pysqlite_iter(pysqlite_Row* self)
168 {
169     return PyObject_GetIter(self->data);
170 }
171 
172 static long pysqlite_row_hash(pysqlite_Row *self)
173 {
174     return PyObject_Hash(self->description) ^ PyObject_Hash(self->data);
175 }
176 
177 static PyObject* pysqlite_row_richcompare(pysqlite_Row *self, PyObject *_other, int opid)
178 {
179     if (opid != Py_EQ && opid != Py_NE) {
180         Py_INCREF(Py_NotImplemented);
181         return Py_NotImplemented;
182     }
183     if (PyType_IsSubtype(Py_TYPE(_other), &pysqlite_RowType)) {
184         pysqlite_Row *other = (pysqlite_Row *)_other;
185         PyObject *res = PyObject_RichCompare(self->description, other->description, opid);
186         if ((opid == Py_EQ && res == Py_True)
187             || (opid == Py_NE && res == Py_False)) {
188             Py_DECREF(res);
189             return PyObject_RichCompare(self->data, other->data, opid);
190         }
191     }
192     Py_INCREF(Py_NotImplemented);
193     return Py_NotImplemented;
194 }
195 
196 PyMappingMethods pysqlite_row_as_mapping = {
197     /* mp_length        */ (lenfunc)pysqlite_row_length,
198     /* mp_subscript     */ (binaryfunc)pysqlite_row_subscript,
199     /* mp_ass_subscript */ (objobjargproc)0,
200 };
201 
202 static PyMethodDef pysqlite_row_methods[] = {
203     {"keys", (PyCFunction)pysqlite_row_keys, METH_NOARGS,
204         PyDoc_STR("Returns the keys of the row.")},
205     {NULL, NULL}
206 };
207 
208 
209 PyTypeObject pysqlite_RowType = {
210         PyVarObject_HEAD_INIT(NULL, 0)
211         MODULE_NAME ".Row",                             /* tp_name */
212         sizeof(pysqlite_Row),                           /* tp_basicsize */
213         0,                                              /* tp_itemsize */
214         (destructor)pysqlite_row_dealloc,               /* tp_dealloc */
215         (printfunc)pysqlite_row_print,                  /* tp_print */
216         0,                                              /* tp_getattr */
217         0,                                              /* tp_setattr */
218         0,                                              /* tp_compare */
219         0,                                              /* tp_repr */
220         0,                                              /* tp_as_number */
221         0,                                              /* tp_as_sequence */
222         0,                                              /* tp_as_mapping */
223         (hashfunc)pysqlite_row_hash,                    /* tp_hash */
224         0,                                              /* tp_call */
225         0,                                              /* tp_str */
226         0,                                              /* tp_getattro */
227         0,                                              /* tp_setattro */
228         0,                                              /* tp_as_buffer */
229         Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE,         /* tp_flags */
230         0,                                              /* tp_doc */
231         (traverseproc)0,                                /* tp_traverse */
232         0,                                              /* tp_clear */
233         (richcmpfunc)pysqlite_row_richcompare,          /* tp_richcompare */
234         0,                                              /* tp_weaklistoffset */
235         (getiterfunc)pysqlite_iter,                     /* tp_iter */
236         0,                                              /* tp_iternext */
237         pysqlite_row_methods,                           /* tp_methods */
238         0,                                              /* tp_members */
239         0,                                              /* tp_getset */
240         0,                                              /* tp_base */
241         0,                                              /* tp_dict */
242         0,                                              /* tp_descr_get */
243         0,                                              /* tp_descr_set */
244         0,                                              /* tp_dictoffset */
245         (initproc)pysqlite_row_init,                    /* tp_init */
246         0,                                              /* tp_alloc */
247         0,                                              /* tp_new */
248         0                                               /* tp_free */
249 };
250 
251 extern int pysqlite_row_setup_types(void)
252 {
253     pysqlite_RowType.tp_new = PyType_GenericNew;
254     pysqlite_RowType.tp_as_mapping = &pysqlite_row_as_mapping;
255     return PyType_Ready(&pysqlite_RowType);
256 }