use core::{

ffi::{CStr, c_int, c_longlong, c_uint, c_void},

fmt::Debug,

ops::Deref,

ptr::{NonNull, null, null_mut},

};

use libsqlite3_sys::{

SQLITE_BLOB, SQLITE_DETERMINISTIC, SQLITE_FLOAT, SQLITE_INTEGER, SQLITE_NULL,

SQLITE_OPEN_CREATE, SQLITE_OPEN_READWRITE, SQLITE_OPEN_URI, SQLITE_TEXT, SQLITE_TRACE_STMT,

SQLITE_TRANSIENT, SQLITE_UTF8, sqlite3, sqlite3_aggregate_context, sqlite3_backup_finish,

sqlite3_backup_init, sqlite3_backup_pagecount, sqlite3_backup_remaining,

sqlite3_backup_step, sqlite3_bind_blob, sqlite3_bind_double, sqlite3_bind_int64,

sqlite3_bind_null, sqlite3_bind_parameter_count, sqlite3_bind_parameter_name,

sqlite3_bind_text, sqlite3_blob, sqlite3_blob_bytes, sqlite3_blob_close, sqlite3_blob_open,

sqlite3_blob_read, sqlite3_blob_write, sqlite3_busy_timeout, sqlite3_changes,

sqlite3_column_blob, sqlite3_column_bytes, sqlite3_column_count, sqlite3_column_decltype,

sqlite3_column_double, sqlite3_column_int64, sqlite3_column_name, sqlite3_column_text,

sqlite3_column_type, sqlite3_complete, sqlite3_context, sqlite3_context_db_handle,

sqlite3_create_collation_v2, sqlite3_create_function_v2, sqlite3_create_window_function,

sqlite3_data_count, sqlite3_db_handle, sqlite3_errcode, sqlite3_errmsg, sqlite3_exec,

sqlite3_expanded_sql, sqlite3_extended_errcode, sqlite3_finalize, sqlite3_get_autocommit,

sqlite3_interrupt, sqlite3_last_insert_rowid, sqlite3_libversion, sqlite3_limit,

sqlite3_open_v2, sqlite3_prepare_v2, sqlite3_progress_handler, sqlite3_reset,

sqlite3_result_blob, sqlite3_result_double, sqlite3_result_error,

sqlite3_result_error_nomem, sqlite3_result_error_toobig, sqlite3_result_int64,

sqlite3_result_null, sqlite3_result_text, sqlite3_set_authorizer, sqlite3_sleep,

sqlite3_step, sqlite3_stmt, sqlite3_stmt_busy, sqlite3_stmt_readonly, sqlite3_threadsafe,

sqlite3_total_changes, sqlite3_trace_v2, sqlite3_user_data, sqlite3_value,

sqlite3_value_blob, sqlite3_value_bytes, sqlite3_value_double, sqlite3_value_int64,

sqlite3_value_text, sqlite3_value_type,

};

use malachite_bigint::Sign;

use num_traits::ToPrimitive;

use rustpython_common::{

atomic::{Ordering, PyAtomic, Radium},

hash::PyHash,

lock::{PyMappedMutexGuard, PyMutex, PyMutexGuard},

static_cell,

};

use rustpython_vm::{

__exports::paste,

AsObject, Py, PyAtomicRef, PyObject, PyObjectRef, PyPayload, PyRef, PyResult,

TryFromBorrowedObject, VirtualMachine, atomic_func,

builtins::{

PyBaseException, PyBaseExceptionRef, PyByteArray, PyBytes, PyDict, PyDictRef, PyFloat,

PyInt, PyIntRef, PyModule, PySlice, PyStr, PyStrRef, PyTuple, PyTupleRef, PyType,

PyTypeRef, PyUtf8Str, PyUtf8StrRef,

},

convert::IntoObject,

function::{

ArgCallable, ArgIterable, FsPath, FuncArgs, OptionalArg, PyComparisonValue,

PySetterValue, TimeoutSeconds,

},

object::{Traverse, TraverseFn},

protocol::{

PyBuffer, PyIterReturn, PyMappingMethods, PyNumberMethods, PySequence,

PySequenceMethods,

},

sliceable::{SaturatedSliceIter, SliceableSequenceOp},

types::{

AsMapping, AsNumber, AsSequence, Callable, Comparable, Constructor, Hashable,

Initializer, IterNext, Iterable, PyComparisonOp, SelfIter,

},

utils::ToCString,

};

use std::thread::ThreadId;

macro_rules! exceptions {

($(($x:ident, $base:expr)),*) => {

paste::paste! {

static_cell! {

$(

static [<$x:snake:upper>]: PyTypeRef;

)*

}

$(

#[allow(dead_code)]

fn [<new_ $x:snake>](vm: &VirtualMachine, msg: String) -> PyBaseExceptionRef {

vm.new_exception_msg([<$x:snake _type>]().to_owned(), msg.into())

}

fn [<$x:snake _type>]() -> &'static Py<PyType> {

[<$x:snake:upper>].get().expect("exception type not initialize")

}

)*

fn setup_module_exceptions(module: &PyObject, vm: &VirtualMachine) {

$(

#[allow(clippy::redundant_closure_call)]

let exception = [<$x:snake:upper>].get_or_init(|| {

let base = $base(vm);

vm.ctx.new_exception_type("_sqlite3", stringify!($x), Some(vec![base.to_owned()]))

});

module.set_attr(stringify!($x), exception.clone().into_object(), vm).unwrap();

)*

}

}

};

}

exceptions!(

(Warning, |vm: &VirtualMachine| vm

.ctx

.exceptions

.exception_type),

(Error, |vm: &VirtualMachine| vm

.ctx

.exceptions

.exception_type),

(InterfaceError, |_| error_type()),

(DatabaseError, |_| error_type()),

(DataError, |_| database_error_type()),

(OperationalError, |_| database_error_type()),

(IntegrityError, |_| database_error_type()),

(InternalError, |_| database_error_type()),

(ProgrammingError, |_| database_error_type()),

(NotSupportedError, |_| database_error_type())

);

#[pyattr]

fn sqlite_version(vm: &VirtualMachine) -> String {

let s = unsafe { sqlite3_libversion() };

ptr_to_str(s, vm).unwrap().to_owned()

}

#[pyattr]

fn threadsafety(_: &VirtualMachine) -> c_int {

let mode = unsafe { sqlite3_threadsafe() };

match mode {

0 => 0,

1 => 3,

2 => 1,

_ => panic!("Unable to interpret SQLite threadsafety mode"),

}

}

#[pyattr(name = "_deprecated_version")]

const PYSQLITE_VERSION: &str = "2.6.0";

#[pyattr]

const PARSE_DECLTYPES: c_int = 1;

#[pyattr]

const PARSE_COLNAMES: c_int = 2;

#[pyattr]

const LEGACY_TRANSACTION_CONTROL: c_int = -1;

#[pyattr]

use libsqlite3_sys::{

SQLITE_ALTER_TABLE, SQLITE_ANALYZE, SQLITE_ATTACH, SQLITE_CREATE_INDEX,

SQLITE_CREATE_TABLE, SQLITE_CREATE_TEMP_INDEX, SQLITE_CREATE_TEMP_TABLE,

SQLITE_CREATE_TEMP_TRIGGER, SQLITE_CREATE_TEMP_VIEW, SQLITE_CREATE_TRIGGER,

SQLITE_CREATE_VIEW, SQLITE_CREATE_VTABLE, SQLITE_DELETE, SQLITE_DENY, SQLITE_DETACH,

SQLITE_DROP_INDEX, SQLITE_DROP_TABLE, SQLITE_DROP_TEMP_INDEX, SQLITE_DROP_TEMP_TABLE,

SQLITE_DROP_TEMP_TRIGGER, SQLITE_DROP_TEMP_VIEW, SQLITE_DROP_TRIGGER, SQLITE_DROP_VIEW,

SQLITE_DROP_VTABLE, SQLITE_FUNCTION, SQLITE_IGNORE, SQLITE_INSERT, SQLITE_LIMIT_ATTACHED,

SQLITE_LIMIT_COLUMN, SQLITE_LIMIT_COMPOUND_SELECT, SQLITE_LIMIT_EXPR_DEPTH,

SQLITE_LIMIT_FUNCTION_ARG, SQLITE_LIMIT_LENGTH, SQLITE_LIMIT_LIKE_PATTERN_LENGTH,

SQLITE_LIMIT_SQL_LENGTH, SQLITE_LIMIT_TRIGGER_DEPTH, SQLITE_LIMIT_VARIABLE_NUMBER,

SQLITE_LIMIT_VDBE_OP, SQLITE_LIMIT_WORKER_THREADS, SQLITE_PRAGMA, SQLITE_READ,

SQLITE_RECURSIVE, SQLITE_REINDEX, SQLITE_SAVEPOINT, SQLITE_SELECT, SQLITE_TRANSACTION,

SQLITE_UPDATE,

};

macro_rules! error_codes {

($($x:ident),*) => {

$(

#[allow(unused_imports)]

use libsqlite3_sys::$x;

)*

static ERROR_CODES: &[(&str, c_int)] = &[

$(

(stringify!($x), libsqlite3_sys::$x),

)*

];

};

}

error_codes!(

SQLITE_ABORT,

SQLITE_AUTH,

SQLITE_BUSY,

SQLITE_CANTOPEN,

SQLITE_CONSTRAINT,

SQLITE_CORRUPT,

SQLITE_DONE,

SQLITE_EMPTY,

SQLITE_ERROR,

SQLITE_FORMAT,

SQLITE_FULL,

SQLITE_INTERNAL,

SQLITE_INTERRUPT,

SQLITE_IOERR,

SQLITE_LOCKED,

SQLITE_MISMATCH,

SQLITE_MISUSE,

SQLITE_NOLFS,

SQLITE_NOMEM,

SQLITE_NOTADB,

SQLITE_NOTFOUND,

SQLITE_OK,

SQLITE_PERM,

SQLITE_PROTOCOL,

SQLITE_RANGE,

SQLITE_READONLY,

SQLITE_ROW,

SQLITE_SCHEMA,

SQLITE_TOOBIG,

SQLITE_NOTICE,

SQLITE_WARNING,

SQLITE_ABORT_ROLLBACK,

SQLITE_BUSY_RECOVERY,

SQLITE_CANTOPEN_FULLPATH,

SQLITE_CANTOPEN_ISDIR,

SQLITE_CANTOPEN_NOTEMPDIR,

SQLITE_CORRUPT_VTAB,

SQLITE_IOERR_ACCESS,

SQLITE_IOERR_BLOCKED,

SQLITE_IOERR_CHECKRESERVEDLOCK,

SQLITE_IOERR_CLOSE,

SQLITE_IOERR_DELETE,

SQLITE_IOERR_DELETE_NOENT,

SQLITE_IOERR_DIR_CLOSE,

SQLITE_IOERR_DIR_FSYNC,

SQLITE_IOERR_FSTAT,

SQLITE_IOERR_FSYNC,

SQLITE_IOERR_LOCK,

SQLITE_IOERR_NOMEM,

SQLITE_IOERR_RDLOCK,

SQLITE_IOERR_READ,

SQLITE_IOERR_SEEK,

SQLITE_IOERR_SHMLOCK,

SQLITE_IOERR_SHMMAP,

SQLITE_IOERR_SHMOPEN,

SQLITE_IOERR_SHMSIZE,

SQLITE_IOERR_SHORT_READ,

SQLITE_IOERR_TRUNCATE,

SQLITE_IOERR_UNLOCK,

SQLITE_IOERR_WRITE,

SQLITE_LOCKED_SHAREDCACHE,

SQLITE_READONLY_CANTLOCK,

SQLITE_READONLY_RECOVERY,

SQLITE_CONSTRAINT_CHECK,

SQLITE_CONSTRAINT_COMMITHOOK,

SQLITE_CONSTRAINT_FOREIGNKEY,

SQLITE_CONSTRAINT_FUNCTION,

SQLITE_CONSTRAINT_NOTNULL,

SQLITE_CONSTRAINT_PRIMARYKEY,

SQLITE_CONSTRAINT_TRIGGER,

SQLITE_CONSTRAINT_UNIQUE,

SQLITE_CONSTRAINT_VTAB,

SQLITE_READONLY_ROLLBACK,

SQLITE_IOERR_MMAP,

SQLITE_NOTICE_RECOVER_ROLLBACK,

SQLITE_NOTICE_RECOVER_WAL,

SQLITE_BUSY_SNAPSHOT,

SQLITE_IOERR_GETTEMPPATH,

SQLITE_WARNING_AUTOINDEX,

SQLITE_CANTOPEN_CONVPATH,

SQLITE_IOERR_CONVPATH,

SQLITE_CONSTRAINT_ROWID,

SQLITE_READONLY_DBMOVED,

SQLITE_AUTH_USER,

SQLITE_OK_LOAD_PERMANENTLY,

SQLITE_IOERR_VNODE,

SQLITE_IOERR_AUTH,

SQLITE_IOERR_BEGIN_ATOMIC,

SQLITE_IOERR_COMMIT_ATOMIC,

SQLITE_IOERR_ROLLBACK_ATOMIC,

SQLITE_ERROR_MISSING_COLLSEQ,

SQLITE_ERROR_RETRY,

SQLITE_READONLY_CANTINIT,

SQLITE_READONLY_DIRECTORY,

SQLITE_CORRUPT_SEQUENCE,

SQLITE_LOCKED_VTAB,

SQLITE_CANTOPEN_DIRTYWAL,

SQLITE_ERROR_SNAPSHOT,

SQLITE_CANTOPEN_SYMLINK,

SQLITE_CONSTRAINT_PINNED,

SQLITE_OK_SYMLINK,

SQLITE_BUSY_TIMEOUT,

SQLITE_CORRUPT_INDEX,

SQLITE_IOERR_DATA,

SQLITE_IOERR_CORRUPTFS

);

/// Autocommit mode setting for sqlite3 connections.

/// - Legacy (default): use isolation_level to control transactions

/// - Enabled: autocommit mode (no automatic transactions)

/// - Disabled: manual commit mode

#[derive(Clone, Copy, Debug, Default, PartialEq, Eq)]

enum AutocommitMode {

#[default]

Legacy,

Enabled,

Disabled,

}

impl TryFromBorrowedObject<'_> for AutocommitMode {

fn try_from_borrowed_object(vm: &VirtualMachine, obj: &PyObject) -> PyResult<Self> {

if obj.is(&vm.ctx.true_value) {

Ok(Self::Enabled)

} else if obj.is(&vm.ctx.false_value) {

Ok(Self::Disabled)

} else if let Ok(val) = obj.try_to_value::<c_int>(vm) {

if val == LEGACY_TRANSACTION_CONTROL {

Ok(Self::Legacy)

} else {

Err(vm.new_value_error(format!(

"autocommit must be True, False, or sqlite3.LEGACY_TRANSACTION_CONTROL, not {val}"

)))

}

} else {

Err(vm.new_type_error(format!(

"autocommit must be True, False, or sqlite3.LEGACY_TRANSACTION_CONTROL, not {}",

obj.class().name()

)))

}

}

}

#[derive(FromArgs)]

struct ConnectArgs {

#[pyarg(any)]

database: FsPath,

#[pyarg(any, default = TimeoutSeconds::new(5.0))]

timeout: TimeoutSeconds,

#[pyarg(any, default = 0)]

detect_types: c_int,

#[pyarg(any, default = Some(vm.ctx.empty_str.to_owned()))]

isolation_level: Option<PyStrRef>,

#[pyarg(any, default = true)]

check_same_thread: bool,

#[pyarg(any, default = Connection::class(&vm.ctx).to_owned())]

factory: PyTypeRef,

// TODO: cache statements

#[allow(dead_code)]

#[pyarg(any, default = 0)]

cached_statements: c_int,

#[pyarg(any, default = false)]

uri: bool,

#[pyarg(any, default)]

autocommit: AutocommitMode,

}

unsafe impl Traverse for ConnectArgs {

fn traverse(&self, tracer_fn: &mut TraverseFn<'_>) {

self.isolation_level.traverse(tracer_fn);

self.factory.traverse(tracer_fn);

}

}

#[derive(FromArgs)]

struct BackupArgs {

#[pyarg(any)]

target: PyRef<Connection>,

#[pyarg(named, default = -1)]

pages: c_int,

#[pyarg(named, optional)]

progress: Option<ArgCallable>,

#[pyarg(named, optional)]

name: Option<PyStrRef>,

#[pyarg(named, default = 0.250)]

sleep: f64,

}

unsafe impl Traverse for BackupArgs {

fn traverse(&self, tracer_fn: &mut TraverseFn<'_>) {

self.progress.traverse(tracer_fn);

self.name.traverse(tracer_fn);

}

}

#[derive(FromArgs)]

struct CreateFunctionArgs {

#[pyarg(any)]

name: PyStrRef,

#[pyarg(any)]

narg: c_int,

#[pyarg(any)]

func: PyObjectRef,

#[pyarg(named, default)]

deterministic: bool,

}

#[derive(FromArgs)]

struct CreateAggregateArgs {

#[pyarg(any)]

name: PyStrRef,

#[pyarg(positional)]

narg: c_int,

#[pyarg(positional)]

aggregate_class: PyObjectRef,

}

#[derive(FromArgs)]

struct BlobOpenArgs {

#[pyarg(positional)]

table: PyStrRef,

#[pyarg(positional)]

column: PyStrRef,

#[pyarg(positional)]

row: i64,

#[pyarg(named, default)]

readonly: bool,

#[pyarg(named, default = vm.ctx.new_str("main"))]

name: PyStrRef,

}

#[derive(FromArgs)]

struct CursorArgs {

#[pyarg(any, default)]

factory: OptionalArg<PyObjectRef>,

}

struct CallbackData {

obj: NonNull<PyObject>,

vm: *const VirtualMachine,

}

impl CallbackData {

fn new(obj: PyObjectRef, vm: &VirtualMachine) -> Option<Self> {

(!vm.is_none(&obj)).then_some(Self {

obj: obj.into_raw(),

vm,

})

}

fn retrieve(&self) -> (&PyObject, &VirtualMachine) {

unsafe { (self.obj.as_ref(), &*self.vm) }

}

unsafe extern "C" fn destructor(data: *mut c_void) {

drop(unsafe { Box::from_raw(data.cast::<Self>()) });

}

unsafe extern "C" fn func_callback(

context: *mut sqlite3_context,

argc: c_int,

argv: *mut *mut sqlite3_value,

) {

let context = SqliteContext::from(context);

let (func, vm) = unsafe { (*context.user_data::<Self>()).retrieve() };

let args = unsafe { core::slice::from_raw_parts(argv, argc as usize) };

let f = || -> PyResult<()> {

let db = context.db_handle();

let args = args

.iter()

.copied()

.map(|val| value_to_object(val, db, vm))

.collect::<PyResult<Vec<PyObjectRef>>>()?;

let val = func.call(args, vm)?;

context.result_from_object(&val, vm)

};

if let Err(exc) = f() {

context.result_exception(vm, exc, "user-defined function raised exception\0")

}

}

unsafe extern "C" fn step_callback(

context: *mut sqlite3_context,

argc: c_int,

argv: *mut *mut sqlite3_value,

) {

let context = SqliteContext::from(context);

let (cls, vm) = unsafe { (*context.user_data::<Self>()).retrieve() };

let args = unsafe { core::slice::from_raw_parts(argv, argc as usize) };

let instance = context.aggregate_context::<*const PyObject>();

if unsafe { (*instance).is_null() } {

match cls.call((), vm) {

Ok(obj) => unsafe { *instance = obj.into_raw().as_ptr() },

Err(exc) => {

return context.result_exception(

vm,

exc,

"user-defined aggregate's '__init__' method raised error\0",

);

}

}

}

let instance = unsafe { &**instance };

Self::call_method_with_args(context, instance, "step", args, vm);

}

unsafe extern "C" fn finalize_callback(context: *mut sqlite3_context) {

let context = SqliteContext::from(context);

let (_, vm) = unsafe { (*context.user_data::<Self>()).retrieve() };

let instance = context.aggregate_context::<*const PyObject>();

let Some(instance) = (unsafe { (*instance).as_ref() }) else {

return;

};

Self::callback_result_from_method(context, instance, "finalize", vm);

}

unsafe extern "C" fn collation_callback(

data: *mut c_void,

a_len: c_int,

a_ptr: *const c_void,

b_len: c_int,

b_ptr: *const c_void,

) -> c_int {

let (callable, vm) = unsafe { (*data.cast::<Self>()).retrieve() };

let f = || -> PyResult<c_int> {

let text1 = ptr_to_string(a_ptr.cast(), a_len, null_mut(), vm)?;

let text1 = vm.ctx.new_str(text1);

let text2 = ptr_to_string(b_ptr.cast(), b_len, null_mut(), vm)?;

let text2 = vm.ctx.new_str(text2);

let val = callable.call((text1, text2), vm)?;

let Some(val) = val.number().index(vm) else {

return Ok(0);

};

let val = match val?.as_bigint().sign() {

Sign::Plus => 1,

Sign::Minus => -1,

Sign::NoSign => 0,

};

Ok(val)

};

f().unwrap_or(0)

}

unsafe extern "C" fn value_callback(context: *mut sqlite3_context) {

let context = SqliteContext::from(context);

let (_, vm) = unsafe { (*context.user_data::<Self>()).retrieve() };

let instance = context.aggregate_context::<*const PyObject>();

let instance = unsafe { &**instance };

Self::callback_result_from_method(context, instance, "value", vm);

}

unsafe extern "C" fn inverse_callback(

context: *mut sqlite3_context,

argc: c_int,

argv: *mut *mut sqlite3_value,

) {

let context = SqliteContext::from(context);

let (_, vm) = unsafe { (*context.user_data::<Self>()).retrieve() };

let args = unsafe { core::slice::from_raw_parts(argv, argc as usize) };

let instance = context.aggregate_context::<*const PyObject>();

let instance = unsafe { &**instance };

Self::call_method_with_args(context, instance, "inverse", args, vm);

}

unsafe extern "C" fn authorizer_callback(

data: *mut c_void,

action: c_int,

arg1: *const libc::c_char,

arg2: *const libc::c_char,

db_name: *const libc::c_char,

access: *const libc::c_char,

) -> c_int {

let (callable, vm) = unsafe { (*data.cast::<Self>()).retrieve() };

let f = || -> PyResult<c_int> {

let arg1 = ptr_to_str(arg1, vm)?;

let arg2 = ptr_to_str(arg2, vm)?;

let db_name = ptr_to_str(db_name, vm)?;

let access = ptr_to_str(access, vm)?;

let val = callable.call((action, arg1, arg2, db_name, access), vm)?;

let Some(val) = val.downcast_ref::<PyInt>() else {

return Ok(SQLITE_DENY);

};

val.try_to_primitive::<c_int>(vm)

};

f().unwrap_or(SQLITE_DENY)

}

unsafe extern "C" fn trace_callback(

_typ: c_uint,

data: *mut c_void,

stmt: *mut c_void,

sql: *mut c_void,

) -> c_int {

let (callable, vm) = unsafe { (*data.cast::<Self>()).retrieve() };

let expanded = unsafe { sqlite3_expanded_sql(stmt.cast()) };

let f = || -> PyResult<()> {

let stmt = ptr_to_str(expanded, vm).or_else(|_| ptr_to_str(sql.cast(), vm))?;

callable.call((stmt,), vm)?;

Ok(())

};

let _ = f();

0

}

unsafe extern "C" fn progress_callback(data: *mut c_void) -> c_int {

let (callable, vm) = unsafe { (*data.cast::<Self>()).retrieve() };

if let Ok(val) = callable.call((), vm)

&& let Ok(val) = val.is_true(vm)

{

return val as c_int;

}

-1

}

fn callback_result_from_method(

context: SqliteContext,

instance: &PyObject,

name: &str,

vm: &VirtualMachine,

) {

let f = || -> PyResult<()> {

let val = vm.call_method(instance, name, ())?;

context.result_from_object(&val, vm)

};

if let Err(exc) = f() {

if exc.fast_isinstance(vm.ctx.exceptions.attribute_error) {

context.result_exception(

vm,

exc,

&format!("user-defined aggregate's '{name}' method not defined\0"),

)

} else {

context.result_exception(

vm,

exc,

&format!("user-defined aggregate's '{name}' method raised error\0"),

)

}

}

}

fn call_method_with_args(

context: SqliteContext,

instance: &PyObject,

name: &str,

args: &[*mut sqlite3_value],

vm: &VirtualMachine,

) {

let f = || -> PyResult<()> {

let db = context.db_handle();

let args = args

.iter()

.copied()

.map(|val| value_to_object(val, db, vm))

.collect::<PyResult<Vec<PyObjectRef>>>()?;

vm.call_method(instance, name, args).map(drop)

};

if let Err(exc) = f() {

if exc.fast_isinstance(vm.ctx.exceptions.attribute_error) {

context.result_exception(

vm,

exc,

&format!("user-defined aggregate's '{name}' method not defined\0"),

)

} else {

context.result_exception(

vm,

exc,

&format!("user-defined aggregate's '{name}' method raised error\0"),

)

}

}

}

}

impl Drop for CallbackData {

fn drop(&mut self) {

unsafe {

let _ = PyObjectRef::from_raw(self.obj);

};

}

}

#[pyfunction]

fn connect(args: ConnectArgs, vm: &VirtualMachine) -> PyResult {

let factory = args.factory.clone();

let conn = Connection::py_new(&factory, args, vm)?;

conn.into_ref_with_type(vm, factory).map(Into::into)

}

#[pyfunction]

fn complete_statement(statement: PyStrRef, vm: &VirtualMachine) -> PyResult<bool> {

let s = statement.to_cstring(vm)?;

let ret = unsafe { sqlite3_complete(s.as_ptr()) };

Ok(ret == 1)

}

#[pyfunction]

fn enable_callback_tracebacks(flag: bool) {

enable_traceback().store(flag, Ordering::Relaxed);

}

#[pyfunction]

fn register_adapter(typ: PyTypeRef, adapter: ArgCallable, vm: &VirtualMachine) -> PyResult<()> {

if typ.is(PyInt::class(&vm.ctx))

|| typ.is(PyFloat::class(&vm.ctx))

|| typ.is(PyStr::class(&vm.ctx))

|| typ.is(PyByteArray::class(&vm.ctx))

{

let _ = BASE_TYPE_ADAPTED.set(());

}

let protocol = PrepareProtocol::class(&vm.ctx).to_owned();

let key = vm.ctx.new_tuple(vec![typ.into(), protocol.into()]);

adapters().set_item(key.as_object(), adapter.into(), vm)

}

#[pyfunction]

fn register_converter(

typename: PyStrRef,

converter: ArgCallable,

vm: &VirtualMachine,

) -> PyResult<()> {

let name = typename.expect_str().to_uppercase();

converters().set_item(&name, converter.into(), vm)

}

fn _adapt<F>(obj: &PyObject, proto: PyTypeRef, alt: F, vm: &VirtualMachine) -> PyResult

where

F: FnOnce(&PyObject) -> PyResult,

{

let proto = proto.into_object();

let key = vm

.ctx

.new_tuple(vec![obj.class().to_owned().into(), proto.clone()]);

if let Some(adapter) = adapters().get_item_opt(key.as_object(), vm)? {

return adapter.call((obj,), vm);

}

if let Ok(adapter) = proto.get_attr("__adapt__", vm) {

match adapter.call((obj,), vm) {

Ok(val) => {

if !vm.is_none(&val) {

return Ok(val);

}

}

Err(exc) => {

if !exc.fast_isinstance(vm.ctx.exceptions.type_error) {

return Err(exc);

}

}

}

}

if let Ok(adapter) = obj.get_attr("__conform__", vm) {

match adapter.call((proto,), vm) {

Ok(val) => {

if !vm.is_none(&val) {

return Ok(val);

}

}

Err(exc) => {

if !exc.fast_isinstance(vm.ctx.exceptions.type_error) {

return Err(exc);

}

}

}

}

alt(obj)

}

#[pyfunction]

fn adapt(

obj: PyObjectRef,

proto: OptionalArg<Option<PyTypeRef>>,

alt: OptionalArg<PyObjectRef>,

vm: &VirtualMachine,

) -> PyResult {

if matches!(proto, OptionalArg::Present(None)) {

return if let OptionalArg::Present(alt) = alt {

Ok(alt)

} else {

Err(new_programming_error(vm, "can't adapt".to_owned()))

};

}

let proto = proto

.flatten()

.unwrap_or_else(|| PrepareProtocol::class(&vm.ctx).to_owned());

_adapt(

&obj,

proto,

|_| {

if let OptionalArg::Present(alt) = alt {

Ok(alt)

} else {

Err(new_programming_error(vm, "can't adapt".to_owned()))

}

},

vm,

)

}

fn need_adapt(obj: &PyObject, vm: &VirtualMachine) -> bool {

if BASE_TYPE_ADAPTED.get().is_some() {

true

} else {

let cls = obj.class();

!(cls.is(vm.ctx.types.int_type)

|| cls.is(vm.ctx.types.float_type)

|| cls.is(vm.ctx.types.str_type)

|| cls.is(vm.ctx.types.bytearray_type))

}

}

static_cell! {

static CONVERTERS: PyDictRef;

static ADAPTERS: PyDictRef;

static BASE_TYPE_ADAPTED: ();

static USER_FUNCTION_EXCEPTION: PyAtomicRef<Option<PyBaseException>>;

static ENABLE_TRACEBACK: PyAtomic<bool>;

}

fn converters() -> &'static Py<PyDict> {

CONVERTERS.get().expect("converters not initialize")

}

fn adapters() -> &'static Py<PyDict> {

ADAPTERS.get().expect("adapters not initialize")

}

fn user_function_exception() -> &'static PyAtomicRef<Option<PyBaseException>> {