/Modules/_sqlite/row.c
http://unladen-swallow.googlecode.com/ · C · 256 lines · 195 code · 38 blank · 23 comment · 41 complexity · 186f646c7e98d91dc6b55d3c6ecdfba3 MD5 · raw file
- /* row.c - an enhanced tuple for database rows
- *
- * Copyright (C) 2005-2006 Gerhard Häring <gh@ghaering.de>
- *
- * This file is part of pysqlite.
- *
- * This software is provided 'as-is', without any express or implied
- * warranty. In no event will the authors be held liable for any damages
- * arising from the use of this software.
- *
- * Permission is granted to anyone to use this software for any purpose,
- * including commercial applications, and to alter it and redistribute it
- * freely, subject to the following restrictions:
- *
- * 1. The origin of this software must not be misrepresented; you must not
- * claim that you wrote the original software. If you use this software
- * in a product, an acknowledgment in the product documentation would be
- * appreciated but is not required.
- * 2. Altered source versions must be plainly marked as such, and must not be
- * misrepresented as being the original software.
- * 3. This notice may not be removed or altered from any source distribution.
- */
- #include "row.h"
- #include "cursor.h"
- #include "sqlitecompat.h"
- void pysqlite_row_dealloc(pysqlite_Row* self)
- {
- Py_XDECREF(self->data);
- Py_XDECREF(self->description);
- Py_TYPE(self)->tp_free((PyObject*)self);
- }
- int pysqlite_row_init(pysqlite_Row* self, PyObject* args, PyObject* kwargs)
- {
- PyObject* data;
- pysqlite_Cursor* cursor;
- self->data = 0;
- self->description = 0;
- if (!PyArg_ParseTuple(args, "OO", &cursor, &data)) {
- return -1;
- }
- if (!PyObject_IsInstance((PyObject*)cursor, (PyObject*)&pysqlite_CursorType)) {
- PyErr_SetString(PyExc_TypeError, "instance of cursor required for first argument");
- return -1;
- }
- if (!PyTuple_Check(data)) {
- PyErr_SetString(PyExc_TypeError, "tuple required for second argument");
- return -1;
- }
- Py_INCREF(data);
- self->data = data;
- Py_INCREF(cursor->description);
- self->description = cursor->description;
- return 0;
- }
- PyObject* pysqlite_row_subscript(pysqlite_Row* self, PyObject* idx)
- {
- long _idx;
- char* key;
- int nitems, i;
- char* compare_key;
- char* p1;
- char* p2;
- PyObject* item;
- if (PyInt_Check(idx)) {
- _idx = PyInt_AsLong(idx);
- item = PyTuple_GetItem(self->data, _idx);
- Py_XINCREF(item);
- return item;
- } else if (PyLong_Check(idx)) {
- _idx = PyLong_AsLong(idx);
- item = PyTuple_GetItem(self->data, _idx);
- Py_XINCREF(item);
- return item;
- } else if (PyString_Check(idx)) {
- key = PyString_AsString(idx);
- nitems = PyTuple_Size(self->description);
- for (i = 0; i < nitems; i++) {
- compare_key = PyString_AsString(PyTuple_GET_ITEM(PyTuple_GET_ITEM(self->description, i), 0));
- if (!compare_key) {
- return NULL;
- }
- p1 = key;
- p2 = compare_key;
- while (1) {
- if ((*p1 == (char)0) || (*p2 == (char)0)) {
- break;
- }
- if ((*p1 | 0x20) != (*p2 | 0x20)) {
- break;
- }
- p1++;
- p2++;
- }
- if ((*p1 == (char)0) && (*p2 == (char)0)) {
- /* found item */
- item = PyTuple_GetItem(self->data, i);
- Py_INCREF(item);
- return item;
- }
- }
- PyErr_SetString(PyExc_IndexError, "No item with that key");
- return NULL;
- } else if (PySlice_Check(idx)) {
- PyErr_SetString(PyExc_ValueError, "slices not implemented, yet");
- return NULL;
- } else {
- PyErr_SetString(PyExc_IndexError, "Index must be int or string");
- return NULL;
- }
- }
- Py_ssize_t pysqlite_row_length(pysqlite_Row* self, PyObject* args, PyObject* kwargs)
- {
- return PyTuple_GET_SIZE(self->data);
- }
- PyObject* pysqlite_row_keys(pysqlite_Row* self, PyObject* args, PyObject* kwargs)
- {
- PyObject* list;
- int nitems, i;
- list = PyList_New(0);
- if (!list) {
- return NULL;
- }
- nitems = PyTuple_Size(self->description);
- for (i = 0; i < nitems; i++) {
- if (PyList_Append(list, PyTuple_GET_ITEM(PyTuple_GET_ITEM(self->description, i), 0)) != 0) {
- Py_DECREF(list);
- return NULL;
- }
- }
- return list;
- }
- static int pysqlite_row_print(pysqlite_Row* self, FILE *fp, int flags)
- {
- return (&PyTuple_Type)->tp_print(self->data, fp, flags);
- }
- static PyObject* pysqlite_iter(pysqlite_Row* self)
- {
- return PyObject_GetIter(self->data);
- }
- static long pysqlite_row_hash(pysqlite_Row *self)
- {
- return PyObject_Hash(self->description) ^ PyObject_Hash(self->data);
- }
- static PyObject* pysqlite_row_richcompare(pysqlite_Row *self, PyObject *_other, int opid)
- {
- if (opid != Py_EQ && opid != Py_NE) {
- Py_INCREF(Py_NotImplemented);
- return Py_NotImplemented;
- }
- if (PyType_IsSubtype(Py_TYPE(_other), &pysqlite_RowType)) {
- pysqlite_Row *other = (pysqlite_Row *)_other;
- PyObject *res = PyObject_RichCompare(self->description, other->description, opid);
- if ((opid == Py_EQ && res == Py_True)
- || (opid == Py_NE && res == Py_False)) {
- Py_DECREF(res);
- return PyObject_RichCompare(self->data, other->data, opid);
- }
- }
- Py_INCREF(Py_NotImplemented);
- return Py_NotImplemented;
- }
- PyMappingMethods pysqlite_row_as_mapping = {
- /* mp_length */ (lenfunc)pysqlite_row_length,
- /* mp_subscript */ (binaryfunc)pysqlite_row_subscript,
- /* mp_ass_subscript */ (objobjargproc)0,
- };
- static PyMethodDef pysqlite_row_methods[] = {
- {"keys", (PyCFunction)pysqlite_row_keys, METH_NOARGS,
- PyDoc_STR("Returns the keys of the row.")},
- {NULL, NULL}
- };
- PyTypeObject pysqlite_RowType = {
- PyVarObject_HEAD_INIT(NULL, 0)
- MODULE_NAME ".Row", /* tp_name */
- sizeof(pysqlite_Row), /* tp_basicsize */
- 0, /* tp_itemsize */
- (destructor)pysqlite_row_dealloc, /* tp_dealloc */
- (printfunc)pysqlite_row_print, /* tp_print */
- 0, /* tp_getattr */
- 0, /* tp_setattr */
- 0, /* tp_compare */
- 0, /* tp_repr */
- 0, /* tp_as_number */
- 0, /* tp_as_sequence */
- 0, /* tp_as_mapping */
- (hashfunc)pysqlite_row_hash, /* tp_hash */
- 0, /* tp_call */
- 0, /* tp_str */
- 0, /* tp_getattro */
- 0, /* tp_setattro */
- 0, /* tp_as_buffer */
- Py_TPFLAGS_DEFAULT|Py_TPFLAGS_BASETYPE, /* tp_flags */
- 0, /* tp_doc */
- (traverseproc)0, /* tp_traverse */
- 0, /* tp_clear */
- (richcmpfunc)pysqlite_row_richcompare, /* tp_richcompare */
- 0, /* tp_weaklistoffset */
- (getiterfunc)pysqlite_iter, /* tp_iter */
- 0, /* tp_iternext */
- pysqlite_row_methods, /* tp_methods */
- 0, /* tp_members */
- 0, /* tp_getset */
- 0, /* tp_base */
- 0, /* tp_dict */
- 0, /* tp_descr_get */
- 0, /* tp_descr_set */
- 0, /* tp_dictoffset */
- (initproc)pysqlite_row_init, /* tp_init */
- 0, /* tp_alloc */
- 0, /* tp_new */
- 0 /* tp_free */
- };
- extern int pysqlite_row_setup_types(void)
- {
- pysqlite_RowType.tp_new = PyType_GenericNew;
- pysqlite_RowType.tp_as_mapping = &pysqlite_row_as_mapping;
- return PyType_Ready(&pysqlite_RowType);
- }