/Modules/_sqlite/util.c

  1/* util.c - various utility functions
  2 *
  3 * Copyright (C) 2005-2007 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 "module.h"
 25#include "connection.h"
 26
 27int pysqlite_step(sqlite3_stmt* statement, pysqlite_Connection* connection)
 28{
 29    int rc;
 30
 31    if (statement == NULL) {
 32        /* this is a workaround for SQLite 3.5 and later. it now apparently
 33         * returns NULL for "no-operation" statements */
 34        rc = SQLITE_OK;
 35    } else {
 36        Py_BEGIN_ALLOW_THREADS
 37        rc = sqlite3_step(statement);
 38        Py_END_ALLOW_THREADS
 39    }
 40
 41    return rc;
 42}
 43
 44/**
 45 * Checks the SQLite error code and sets the appropriate DB-API exception.
 46 * Returns the error code (0 means no error occurred).
 47 */
 48int _pysqlite_seterror(sqlite3* db, sqlite3_stmt* st)
 49{
 50    int errorcode;
 51
 52    /* SQLite often doesn't report anything useful, unless you reset the statement first */
 53    if (st != NULL) {
 54        (void)sqlite3_reset(st);
 55    }
 56
 57    errorcode = sqlite3_errcode(db);
 58
 59    switch (errorcode)
 60    {
 61        case SQLITE_OK:
 62            PyErr_Clear();
 63            break;
 64        case SQLITE_INTERNAL:
 65        case SQLITE_NOTFOUND:
 66            PyErr_SetString(pysqlite_InternalError, sqlite3_errmsg(db));
 67            break;
 68        case SQLITE_NOMEM:
 69            (void)PyErr_NoMemory();
 70            break;
 71        case SQLITE_ERROR:
 72        case SQLITE_PERM:
 73        case SQLITE_ABORT:
 74        case SQLITE_BUSY:
 75        case SQLITE_LOCKED:
 76        case SQLITE_READONLY:
 77        case SQLITE_INTERRUPT:
 78        case SQLITE_IOERR:
 79        case SQLITE_FULL:
 80        case SQLITE_CANTOPEN:
 81        case SQLITE_PROTOCOL:
 82        case SQLITE_EMPTY:
 83        case SQLITE_SCHEMA:
 84            PyErr_SetString(pysqlite_OperationalError, sqlite3_errmsg(db));
 85            break;
 86        case SQLITE_CORRUPT:
 87            PyErr_SetString(pysqlite_DatabaseError, sqlite3_errmsg(db));
 88            break;
 89        case SQLITE_TOOBIG:
 90            PyErr_SetString(pysqlite_DataError, sqlite3_errmsg(db));
 91            break;
 92        case SQLITE_CONSTRAINT:
 93        case SQLITE_MISMATCH:
 94            PyErr_SetString(pysqlite_IntegrityError, sqlite3_errmsg(db));
 95            break;
 96        case SQLITE_MISUSE:
 97            PyErr_SetString(pysqlite_ProgrammingError, sqlite3_errmsg(db));
 98            break;
 99        default:
100            PyErr_SetString(pysqlite_DatabaseError, sqlite3_errmsg(db));
101            break;
102    }
103
104    return errorcode;
105}
106