blob: 6b57b763ddbf4c0d8104744ce39a6f96f909c529 [file] [log] [blame]
Anthony Baxterc51ee692006-04-01 00:57:31 +00001/* util.c - various utility functions
2 *
Gerhard Häring3bbb6722010-03-05 09:12:37 +00003 * Copyright (C) 2005-2010 Gerhard Häring <gh@ghaering.de>
Anthony Baxterc51ee692006-04-01 00:57:31 +00004 *
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
Gerhard Häring6e1afcf2008-09-12 18:58:57 +000027int pysqlite_step(sqlite3_stmt* statement, pysqlite_Connection* connection)
Anthony Baxterc51ee692006-04-01 00:57:31 +000028{
29 int rc;
30
Gerhard Häring99b9df82007-12-11 21:07:40 +000031 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 }
Anthony Baxterc51ee692006-04-01 00:57:31 +000040
41 return rc;
42}
43
44/**
45 * Checks the SQLite error code and sets the appropriate DB-API exception.
Georg Brandlcddabbf2006-07-28 18:36:01 +000046 * Returns the error code (0 means no error occurred).
Anthony Baxterc51ee692006-04-01 00:57:31 +000047 */
Gerhard Häring1cc60ed2008-02-29 22:08:41 +000048int _pysqlite_seterror(sqlite3* db, sqlite3_stmt* st)
Anthony Baxterc51ee692006-04-01 00:57:31 +000049{
50 int errorcode;
51
Gerhard Häring1cc60ed2008-02-29 22:08:41 +000052 /* SQLite often doesn't report anything useful, unless you reset the statement first */
53 if (st != NULL) {
54 (void)sqlite3_reset(st);
55 }
56
Anthony Baxterc51ee692006-04-01 00:57:31 +000057 errorcode = sqlite3_errcode(db);
58
59 switch (errorcode)
60 {
61 case SQLITE_OK:
62 PyErr_Clear();
63 break;
Anthony Baxterc51ee692006-04-01 00:57:31 +000064 case SQLITE_INTERNAL:
Anthony Baxter72289a62006-04-04 06:29:05 +000065 case SQLITE_NOTFOUND:
Gerhard Häring0741a602007-01-14 01:43:50 +000066 PyErr_SetString(pysqlite_InternalError, sqlite3_errmsg(db));
Anthony Baxterc51ee692006-04-01 00:57:31 +000067 break;
Anthony Baxterc51ee692006-04-01 00:57:31 +000068 case SQLITE_NOMEM:
69 (void)PyErr_NoMemory();
70 break;
Anthony Baxter72289a62006-04-04 06:29:05 +000071 case SQLITE_ERROR:
72 case SQLITE_PERM:
73 case SQLITE_ABORT:
74 case SQLITE_BUSY:
75 case SQLITE_LOCKED:
Anthony Baxterc51ee692006-04-01 00:57:31 +000076 case SQLITE_READONLY:
Anthony Baxterc51ee692006-04-01 00:57:31 +000077 case SQLITE_INTERRUPT:
Anthony Baxterc51ee692006-04-01 00:57:31 +000078 case SQLITE_IOERR:
Anthony Baxter72289a62006-04-04 06:29:05 +000079 case SQLITE_FULL:
80 case SQLITE_CANTOPEN:
81 case SQLITE_PROTOCOL:
82 case SQLITE_EMPTY:
83 case SQLITE_SCHEMA:
Gerhard Häring0741a602007-01-14 01:43:50 +000084 PyErr_SetString(pysqlite_OperationalError, sqlite3_errmsg(db));
Anthony Baxterc51ee692006-04-01 00:57:31 +000085 break;
86 case SQLITE_CORRUPT:
Gerhard Häring0741a602007-01-14 01:43:50 +000087 PyErr_SetString(pysqlite_DatabaseError, sqlite3_errmsg(db));
Anthony Baxterc51ee692006-04-01 00:57:31 +000088 break;
Anthony Baxterc51ee692006-04-01 00:57:31 +000089 case SQLITE_TOOBIG:
Gerhard Häring0741a602007-01-14 01:43:50 +000090 PyErr_SetString(pysqlite_DataError, sqlite3_errmsg(db));
Anthony Baxterc51ee692006-04-01 00:57:31 +000091 break;
92 case SQLITE_CONSTRAINT:
Anthony Baxterc51ee692006-04-01 00:57:31 +000093 case SQLITE_MISMATCH:
Gerhard Häring0741a602007-01-14 01:43:50 +000094 PyErr_SetString(pysqlite_IntegrityError, sqlite3_errmsg(db));
Anthony Baxterc51ee692006-04-01 00:57:31 +000095 break;
96 case SQLITE_MISUSE:
Gerhard Häring0741a602007-01-14 01:43:50 +000097 PyErr_SetString(pysqlite_ProgrammingError, sqlite3_errmsg(db));
Anthony Baxterc51ee692006-04-01 00:57:31 +000098 break;
99 default:
Gerhard Häring0741a602007-01-14 01:43:50 +0000100 PyErr_SetString(pysqlite_DatabaseError, sqlite3_errmsg(db));
Anthony Baxter72289a62006-04-04 06:29:05 +0000101 break;
Anthony Baxterc51ee692006-04-01 00:57:31 +0000102 }
103
104 return errorcode;
105}
106