mirror of https://github.com/python/cpython
443 lines
14 KiB
C
443 lines
14 KiB
C
/* module.c - the module itself
|
|
*
|
|
* Copyright (C) 2004-2010 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 "connection.h"
|
|
#include "statement.h"
|
|
#include "cursor.h"
|
|
#include "prepare_protocol.h"
|
|
#include "microprotocols.h"
|
|
#include "row.h"
|
|
|
|
#if SQLITE_VERSION_NUMBER < 3007015
|
|
#error "SQLite 3.7.15 or higher required"
|
|
#endif
|
|
|
|
#define clinic_state() (pysqlite_get_state(NULL))
|
|
#include "clinic/module.c.h"
|
|
#undef clinic_state
|
|
|
|
/*[clinic input]
|
|
module _sqlite3
|
|
[clinic start generated code]*/
|
|
/*[clinic end generated code: output=da39a3ee5e6b4b0d input=81e330492d57488e]*/
|
|
|
|
pysqlite_state pysqlite_global_state;
|
|
|
|
// NOTE: This must equal sqlite3.Connection.__init__ argument spec!
|
|
/*[clinic input]
|
|
_sqlite3.connect as pysqlite_connect
|
|
|
|
database: object(converter='PyUnicode_FSConverter')
|
|
timeout: double = 5.0
|
|
detect_types: int = 0
|
|
isolation_level: object = NULL
|
|
check_same_thread: bool(accept={int}) = True
|
|
factory: object(c_default='(PyObject*)clinic_state()->ConnectionType') = ConnectionType
|
|
cached_statements: int = 128
|
|
uri: bool = False
|
|
|
|
Opens a connection to the SQLite database file database.
|
|
|
|
You can use ":memory:" to open a database connection to a database that resides
|
|
in RAM instead of on disk.
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
pysqlite_connect_impl(PyObject *module, PyObject *database, double timeout,
|
|
int detect_types, PyObject *isolation_level,
|
|
int check_same_thread, PyObject *factory,
|
|
int cached_statements, int uri)
|
|
/*[clinic end generated code: output=450ac9078b4868bb input=ea6355ba55a78e12]*/
|
|
{
|
|
if (isolation_level == NULL) {
|
|
isolation_level = PyUnicode_FromString("");
|
|
if (isolation_level == NULL) {
|
|
return NULL;
|
|
}
|
|
}
|
|
else {
|
|
Py_INCREF(isolation_level);
|
|
}
|
|
PyObject *res = PyObject_CallFunction(factory, "OdiOiOii", database,
|
|
timeout, detect_types,
|
|
isolation_level, check_same_thread,
|
|
factory, cached_statements, uri);
|
|
Py_DECREF(database); // needed bco. the AC FSConverter
|
|
Py_DECREF(isolation_level);
|
|
return res;
|
|
}
|
|
|
|
/*[clinic input]
|
|
_sqlite3.complete_statement as pysqlite_complete_statement
|
|
|
|
statement: str
|
|
|
|
Checks if a string contains a complete SQL statement. Non-standard.
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
pysqlite_complete_statement_impl(PyObject *module, const char *statement)
|
|
/*[clinic end generated code: output=e55f1ff1952df558 input=f6b24996b31c5c33]*/
|
|
{
|
|
if (sqlite3_complete(statement)) {
|
|
return Py_NewRef(Py_True);
|
|
} else {
|
|
return Py_NewRef(Py_False);
|
|
}
|
|
}
|
|
|
|
/*[clinic input]
|
|
_sqlite3.enable_shared_cache as pysqlite_enable_shared_cache
|
|
|
|
do_enable: int
|
|
|
|
Enable or disable shared cache mode for the calling thread.
|
|
|
|
Experimental/Non-standard.
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
pysqlite_enable_shared_cache_impl(PyObject *module, int do_enable)
|
|
/*[clinic end generated code: output=259c74eedee1516b input=8400e41bc58b6b24]*/
|
|
{
|
|
int rc;
|
|
|
|
rc = sqlite3_enable_shared_cache(do_enable);
|
|
|
|
if (rc != SQLITE_OK) {
|
|
pysqlite_state *state = pysqlite_get_state(module);
|
|
PyErr_SetString(state->OperationalError, "Changing the shared_cache flag failed");
|
|
return NULL;
|
|
} else {
|
|
Py_RETURN_NONE;
|
|
}
|
|
}
|
|
|
|
/*[clinic input]
|
|
_sqlite3.register_adapter as pysqlite_register_adapter
|
|
|
|
type: object(type='PyTypeObject *')
|
|
caster: object
|
|
/
|
|
|
|
Registers an adapter with pysqlite's adapter registry. Non-standard.
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
pysqlite_register_adapter_impl(PyObject *module, PyTypeObject *type,
|
|
PyObject *caster)
|
|
/*[clinic end generated code: output=a287e8db18e8af23 input=839dad90e2492725]*/
|
|
{
|
|
int rc;
|
|
|
|
/* a basic type is adapted; there's a performance optimization if that's not the case
|
|
* (99 % of all usages) */
|
|
if (type == &PyLong_Type || type == &PyFloat_Type
|
|
|| type == &PyUnicode_Type || type == &PyByteArray_Type) {
|
|
pysqlite_state *state = pysqlite_get_state(module);
|
|
state->BaseTypeAdapted = 1;
|
|
}
|
|
|
|
pysqlite_state *state = pysqlite_get_state(NULL);
|
|
PyObject *protocol = (PyObject *)state->PrepareProtocolType;
|
|
rc = pysqlite_microprotocols_add(type, protocol, caster);
|
|
if (rc == -1) {
|
|
return NULL;
|
|
}
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
/*[clinic input]
|
|
_sqlite3.register_converter as pysqlite_register_converter
|
|
|
|
name as orig_name: unicode
|
|
converter as callable: object
|
|
/
|
|
|
|
Registers a converter with pysqlite. Non-standard.
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
pysqlite_register_converter_impl(PyObject *module, PyObject *orig_name,
|
|
PyObject *callable)
|
|
/*[clinic end generated code: output=a2f2bfeed7230062 input=e074cf7f4890544f]*/
|
|
{
|
|
PyObject* name = NULL;
|
|
PyObject* retval = NULL;
|
|
_Py_IDENTIFIER(upper);
|
|
|
|
/* convert the name to upper case */
|
|
name = _PyObject_CallMethodIdNoArgs(orig_name, &PyId_upper);
|
|
if (!name) {
|
|
goto error;
|
|
}
|
|
|
|
pysqlite_state *state = pysqlite_get_state(module);
|
|
if (PyDict_SetItem(state->converters, name, callable) != 0) {
|
|
goto error;
|
|
}
|
|
|
|
retval = Py_NewRef(Py_None);
|
|
error:
|
|
Py_XDECREF(name);
|
|
return retval;
|
|
}
|
|
|
|
/*[clinic input]
|
|
_sqlite3.enable_callback_tracebacks as pysqlite_enable_callback_trace
|
|
|
|
enable: int
|
|
/
|
|
|
|
Enable or disable callback functions throwing errors to stderr.
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
pysqlite_enable_callback_trace_impl(PyObject *module, int enable)
|
|
/*[clinic end generated code: output=4ff1d051c698f194 input=cb79d3581eb77c40]*/
|
|
{
|
|
pysqlite_state *state = pysqlite_get_state(module);
|
|
state->enable_callback_tracebacks = enable;
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
/*[clinic input]
|
|
_sqlite3.adapt as pysqlite_adapt
|
|
|
|
obj: object
|
|
proto: object(c_default='(PyObject *)clinic_state()->PrepareProtocolType') = PrepareProtocolType
|
|
alt: object = NULL
|
|
/
|
|
|
|
Adapt given object to given protocol. Non-standard.
|
|
[clinic start generated code]*/
|
|
|
|
static PyObject *
|
|
pysqlite_adapt_impl(PyObject *module, PyObject *obj, PyObject *proto,
|
|
PyObject *alt)
|
|
/*[clinic end generated code: output=0c3927c5fcd23dd9 input=c8995aeb25d0e542]*/
|
|
{
|
|
return pysqlite_microprotocols_adapt(obj, proto, alt);
|
|
}
|
|
|
|
static int converters_init(PyObject* module)
|
|
{
|
|
pysqlite_state *state = pysqlite_get_state(module);
|
|
state->converters = PyDict_New();
|
|
if (state->converters == NULL) {
|
|
return -1;
|
|
}
|
|
|
|
return PyModule_AddObjectRef(module, "converters", state->converters);
|
|
}
|
|
|
|
static int
|
|
load_functools_lru_cache(PyObject *module)
|
|
{
|
|
PyObject *functools = PyImport_ImportModule("functools");
|
|
if (functools == NULL) {
|
|
return -1;
|
|
}
|
|
|
|
pysqlite_state *state = pysqlite_get_state(module);
|
|
state->lru_cache = PyObject_GetAttrString(functools, "lru_cache");
|
|
Py_DECREF(functools);
|
|
if (state->lru_cache == NULL) {
|
|
return -1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static PyMethodDef module_methods[] = {
|
|
PYSQLITE_ADAPT_METHODDEF
|
|
PYSQLITE_COMPLETE_STATEMENT_METHODDEF
|
|
PYSQLITE_CONNECT_METHODDEF
|
|
PYSQLITE_ENABLE_CALLBACK_TRACE_METHODDEF
|
|
PYSQLITE_ENABLE_SHARED_CACHE_METHODDEF
|
|
PYSQLITE_REGISTER_ADAPTER_METHODDEF
|
|
PYSQLITE_REGISTER_CONVERTER_METHODDEF
|
|
{NULL, NULL}
|
|
};
|
|
|
|
static int add_integer_constants(PyObject *module) {
|
|
int ret = 0;
|
|
|
|
ret += PyModule_AddIntMacro(module, PARSE_DECLTYPES);
|
|
ret += PyModule_AddIntMacro(module, PARSE_COLNAMES);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_OK);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_DENY);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_IGNORE);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_CREATE_INDEX);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_CREATE_TABLE);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_CREATE_TEMP_INDEX);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_CREATE_TEMP_TABLE);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_CREATE_TEMP_TRIGGER);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_CREATE_TEMP_VIEW);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_CREATE_TRIGGER);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_CREATE_VIEW);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_DELETE);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_DROP_INDEX);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_DROP_TABLE);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_DROP_TEMP_INDEX);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_DROP_TEMP_TABLE);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_DROP_TEMP_TRIGGER);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_DROP_TEMP_VIEW);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_DROP_TRIGGER);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_DROP_VIEW);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_INSERT);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_PRAGMA);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_READ);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_SELECT);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_TRANSACTION);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_UPDATE);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_ATTACH);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_DETACH);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_ALTER_TABLE);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_REINDEX);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_ANALYZE);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_CREATE_VTABLE);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_DROP_VTABLE);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_FUNCTION);
|
|
ret += PyModule_AddIntMacro(module, SQLITE_SAVEPOINT);
|
|
#if SQLITE_VERSION_NUMBER >= 3008003
|
|
ret += PyModule_AddIntMacro(module, SQLITE_RECURSIVE);
|
|
#endif
|
|
ret += PyModule_AddIntMacro(module, SQLITE_DONE);
|
|
return ret;
|
|
}
|
|
|
|
static struct PyModuleDef _sqlite3module = {
|
|
PyModuleDef_HEAD_INIT,
|
|
"_sqlite3",
|
|
NULL,
|
|
-1,
|
|
module_methods,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
NULL
|
|
};
|
|
|
|
#define ADD_TYPE(module, type) \
|
|
do { \
|
|
if (PyModule_AddType(module, type) < 0) { \
|
|
goto error; \
|
|
} \
|
|
} while (0)
|
|
|
|
#define ADD_EXCEPTION(module, state, exc, base) \
|
|
do { \
|
|
state->exc = PyErr_NewException(MODULE_NAME "." #exc, base, NULL); \
|
|
if (state->exc == NULL) { \
|
|
goto error; \
|
|
} \
|
|
ADD_TYPE(module, (PyTypeObject *)state->exc); \
|
|
} while (0)
|
|
|
|
PyMODINIT_FUNC PyInit__sqlite3(void)
|
|
{
|
|
PyObject *module;
|
|
|
|
if (sqlite3_libversion_number() < 3007015) {
|
|
PyErr_SetString(PyExc_ImportError, MODULE_NAME ": SQLite 3.7.15 or higher required");
|
|
return NULL;
|
|
}
|
|
|
|
int rc = sqlite3_initialize();
|
|
if (rc != SQLITE_OK) {
|
|
PyErr_SetString(PyExc_ImportError, sqlite3_errstr(rc));
|
|
return NULL;
|
|
}
|
|
|
|
module = PyModule_Create(&_sqlite3module);
|
|
pysqlite_state *state = pysqlite_get_state(module);
|
|
|
|
if (!module ||
|
|
(pysqlite_row_setup_types(module) < 0) ||
|
|
(pysqlite_cursor_setup_types(module) < 0) ||
|
|
(pysqlite_connection_setup_types(module) < 0) ||
|
|
(pysqlite_statement_setup_types(module) < 0) ||
|
|
(pysqlite_prepare_protocol_setup_types(module) < 0)
|
|
) {
|
|
goto error;
|
|
}
|
|
|
|
ADD_TYPE(module, state->ConnectionType);
|
|
ADD_TYPE(module, state->CursorType);
|
|
ADD_TYPE(module, state->PrepareProtocolType);
|
|
ADD_TYPE(module, state->RowType);
|
|
|
|
/*** Create DB-API Exception hierarchy */
|
|
ADD_EXCEPTION(module, state, Error, PyExc_Exception);
|
|
ADD_EXCEPTION(module, state, Warning, PyExc_Exception);
|
|
|
|
/* Error subclasses */
|
|
ADD_EXCEPTION(module, state, InterfaceError, state->Error);
|
|
ADD_EXCEPTION(module, state, DatabaseError, state->Error);
|
|
|
|
/* DatabaseError subclasses */
|
|
ADD_EXCEPTION(module, state, InternalError, state->DatabaseError);
|
|
ADD_EXCEPTION(module, state, OperationalError, state->DatabaseError);
|
|
ADD_EXCEPTION(module, state, ProgrammingError, state->DatabaseError);
|
|
ADD_EXCEPTION(module, state, IntegrityError, state->DatabaseError);
|
|
ADD_EXCEPTION(module, state, DataError, state->DatabaseError);
|
|
ADD_EXCEPTION(module, state, NotSupportedError, state->DatabaseError);
|
|
|
|
/* Set integer constants */
|
|
if (add_integer_constants(module) < 0) {
|
|
goto error;
|
|
}
|
|
|
|
if (PyModule_AddStringConstant(module, "version", PYSQLITE_VERSION) < 0) {
|
|
goto error;
|
|
}
|
|
|
|
if (PyModule_AddStringConstant(module, "sqlite_version", sqlite3_libversion())) {
|
|
goto error;
|
|
}
|
|
|
|
/* initialize microprotocols layer */
|
|
if (pysqlite_microprotocols_init(module) < 0) {
|
|
goto error;
|
|
}
|
|
|
|
/* initialize the default converters */
|
|
if (converters_init(module) < 0) {
|
|
goto error;
|
|
}
|
|
|
|
if (load_functools_lru_cache(module) < 0) {
|
|
goto error;
|
|
}
|
|
|
|
return module;
|
|
|
|
error:
|
|
sqlite3_shutdown();
|
|
PyErr_SetString(PyExc_ImportError, MODULE_NAME ": init failed");
|
|
Py_XDECREF(module);
|
|
return NULL;
|
|
}
|